114 lines
2.2 KiB
Go
114 lines
2.2 KiB
Go
## 一 基准测试
|
||
|
||
> 基准测试：用于测试一段程序的运行性能及CPU消耗
|
||
|
||
性能测试函数以 Benchmark 为名称前缀，同样保存在 `*_test.go` 文件里。
|
||
|
||
示例：
|
||
```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代码的运行情况，示例代码：
|
||
```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` |