目录

一 基准测试

基准测试:用于测试一段程序的运行性能及CPU消耗

性能测试函数以 Benchmark 为名称前缀,同样保存在 *_test.go 文件里。

示例:

func Benchmark_Hello(b *testing.B){
    
    // 开始测试性能相关代码
    b.ResetTimer()
    for i := 0; i < b.N; i++ {
        //  测试代码
    }

    // 结束性能测试
    b.StopTimer()
}

测试:

go test -v -bench=Hello          # -bench=.表示运行所有基准测试。win下参数为:-bench="-"

常用参数:

-benchmem             # 显示性能具体的开销情况
-benchtime=5s        # 自定义测试时间为5秒

二 性能监控

利用go的 runtime/pprof 包可以生成prof文件,用来查看go代码的运行情况,示例代码:

package main

import (
    "fmt"
    "os"
    "runtime/pprof"
)

func slowFunc(){

    str := "hello world "
    for i := 0; i < 5; i++ {
        str += str
    }
}

func main() {

    // 创建输出文件
    f, err := os.Create("cpu.prof")
    if err != nil {
        fmt.Println("create cpu.prof err :", err)
        return
    }
    // 获取系统信息
    if err := pprof.StartCPUProfile(f); err != nil {
        fmt.Println("start cpu.prof err :", err)
        return
    }
    defer pprof.StopCPUProfile()

    // 业务代码
    slowFunc()

    // 获取内存相关信息
    f1, err := os.Create("mem.prof")
    defer f1.Close()
    if err != nil {
        fmt.Println("create mem.prof err :", err)
        return
    }
    // runtime.GC()            // 是否获取最新的数据信息
    if err := pprof.WriteHeapProfile(f1); err != nil {
        fmt.Println("write cpu.prof err :", err)
        return
    }


    // 获取协程相关信息
    f2, err := os.Create("goroutine.prof")
    defer f2.Close()
    if err != nil {
        fmt.Println("create goroutine.prof err :", err)
        return
    }
    if gProf := pprof.Lookup("goroutine"); gProf != nil {
        fmt.Println("write goroutine.prof err :", err)
        return
    } else {
        gProf.WriteTo(f2, 0)
    }

    return
}

生成prof文件:

# 生成程序的二进制文件
go build -o program main.go                // 此时会按照代码中的要求生成多份prof文件

# 查看prof文件
go tool pprof program cpu.prof

贴士:

  • 导入 "_ "net/http/pprof"包还可以实现以网页形式展示prof文件内容!

  • 程序执行前加上环境变量可以查看GC日志,如:GODEBUG=gctrace=1 go run main.go