在 Go 中，可以使用 runtime/pprof 包来生成 CPU 和内存的火焰图。下面是一个简单的示例：
package main
import (
"log"
"os"
"runtime/pprof"
)
func main() {
// 创建 CPU Profile
f, err := os.Create("cpu.pprof")
if err != nil {
log.Fatal(err)
}
defer f.Close()
err = pprof.StartCPUProfile(f)
if err != nil {
log.Fatal(err)
}
defer pprof.StopCPUProfile()
// 运行需要分析的代码
myFunc()
// 创建 Memory Profile
f, err = os.Create("mem.pprof")
if err != nil {
log.Fatal(err)
}
defer f.Close()
err = pprof.WriteHeapProfile(f)
if err != nil {
log.Fatal(err)
}
}
func myFunc() {
// 这里写需要分析的代码
}
运行上述程序后，会生成 cpu.pprof 和 mem.pprof 两个文件。接下来可以使用 go tool pprof 工具来生成火焰图。例如：
go tool pprof -http=:8080 cpu.pprof
此时可以在浏览器中访问 localhost:8080 查看 CPU 的火焰图。
注意，在生成火焰图之前需要确保代码已经运行足够长的时间，并且达到了稳定状态。
