基本的 cpu 和記憶體分析

在主程式中新增以下程式碼。

var cpuprofile = flag.String("cpuprofile", "", "write cpu profile `file`")
var memprofile = flag.String("memprofile", "", "write memory profile to `file`")

func main() {
    flag.Parse()
    if *cpuprofile != "" {
        f, err := os.Create(*cpuprofile)
        if err != nil {
            log.Fatal("could not create CPU profile: ", err)
        }
        if err := pprof.StartCPUProfile(f); err != nil {
            log.Fatal("could not start CPU profile: ", err)
        }
        defer pprof.StopCPUProfile()
    }
    ...
    if *memprofile != "" {
        f, err := os.Create(*memprofile)
        if err != nil {
            log.Fatal("could not create memory profile: ", err)
        }
        runtime.GC() // get up-to-date statistics
        if err := pprof.WriteHeapProfile(f); err != nil {
            log.Fatal("could not write memory profile: ", err)
        }
        f.Close()
    }
}

之後構建 go 程式，如果在主 go build main.go 中新增。使用程式碼 main.exe -cpuprofile cpu.prof -memprof mem.prof 中定義的標誌執行主程式。如果對測試用例進行了分析，則使用相同的標誌 go test -cpuprofile cpu.prof -memprofile mem.prof 執行測試