南强小屋 Design By 杰米

Go 程序的性能优化及 pprof 的使用

程序的性能优化无非就是对程序占用资源的优化。对于服务器而言,最重要的两项资源莫过于 CPU 和内存。性能优化,就是在对于不影响程序数据处理能力的情况下,我们通常要求程序的 CPU 的内存占用尽量低。反过来说,也就是当程序 CPU 和内存占用不变的情况下,尽量地提高程序的数据处理能力或者说是吞吐量。

Go 的原生工具链中提供了非常多丰富的工具供开发者使用,其中包括 pprof。

对于 pprof 的使用要分成下面两部分来说。

Web 程序使用 pprof

先写一个简单的 Web 服务程序。程序在 9876 端口上接收请求。

package main

import (
  "bytes"
  "io/ioutil"
  "log"
  "math/rand"
  "net/http"

  _ "net/http/pprof"
)

func main() {
  http.HandleFunc("/test", handler)
  log.Fatal(http.ListenAndServe(":9876", nil))
}

func handler(w http.ResponseWriter, r *http.Request) {
  err := r.ParseForm()
  if nil != err {
    w.Write([]byte(err.Error()))
    return
  }
  doSomeThingOne(10000)
  buff := genSomeBytes()
  b, err := ioutil.ReadAll(buff)
  if nil != err {
    w.Write([]byte(err.Error()))
    return
  }
  w.Write(b)
}

func doSomeThingOne(times int) {
  for i := 0; i < times; i++ {
    for j := 0; j < times; j++ {

    }
  }
}

func genSomeBytes() *bytes.Buffer {
  var buff bytes.Buffer
  for i := 1; i < 20000; i++ {
    buff.Write([]byte{'0' + byte(rand.Intn(10))})
  }
  return &buff
}

可以看到我们只是简单地引入了 net/http/pprof ,并未显示地使用。

启动程序。

我们用 wrk 来简单地模拟请求。

wrk -c 400 -t 8 -d 3m http://localhost:9876/test

这时我们打开 http://localhost:9876/debug/pprof,会显示如下页面:

Go程序性能优化及pprof使用方法详解

用户可以点击相应的链接浏览内容。不过这不是我们重点讲述的,而且这些内容看起来并不直观。

我们打开链接 http://localhost:9876/debug/pprof/profile 稍后片刻,可以下载到文件 profile。

使用 Go 自带的 pprof 工具打开。go tool pprof test profile。(proof 后跟的 test 为程序编译的可执行文件)

输入 top 命令得到:

Go程序性能优化及pprof使用方法详解

可以看到 cpu 占用前 10 的函数,我们可以对此分析进行优化。

只是这样可能还不是很直观。

我们输入命令 web(需要事先安装 graphviz,macOS 下可以 brew install graphviz),会在浏览器中打开界面如下:

Go程序性能优化及pprof使用方法详解

可以看到 main.doSomeThingOne 占用了 92.46% 的 CPU 时间,需要对其进行优化。

Web 形式的 CPU 时间图对于优化已经完全够用,这边再介绍一下火焰图的生成。macOS 推荐使用 go-torch 工具。使用方法和 go tool pprof 相似。

go-torch test profile 会生成 torch.svg 文件。可以用浏览器打开,如图。

Go程序性能优化及pprof使用方法详解

刚才只是讲了 CPU 的占用分析文件的生成查看,其实内存快照的生成相似。http://localhost:9876/debug/pprof/heap,会下载得到 heap.gz 文件。

我们同样可以使用 go tool pprof test heap.gz,然后输入 top 或 web 命令查看相关内容。

Go程序性能优化及pprof使用方法详解

Go程序性能优化及pprof使用方法详解

通用程序使用 pprof

我们写的 Go 程序并非都是 Web 程序,这时候再使用上面的方法就不行了。

我们仍然可以使用 pprof 工具,但引入的位置为 runtime/pprof 。

这里贴出两个函数,作为示例:

// 生成 CPU 报告
func cpuProfile() {
  f, err := os.OpenFile("cpu.prof", os.O_RDWR|os.O_CREATE, 0644)
  if err != nil {
    log.Fatal(err)
  }
  defer f.Close()

  log.Println("CPU Profile started")
  pprof.StartCPUProfile(f)
  defer pprof.StopCPUProfile()

  time.Sleep(60 * time.Second)
  fmt.Println("CPU Profile stopped")
}

// 生成堆内存报告
func heapProfile() {
  f, err := os.OpenFile("heap.prof", os.O_RDWR|os.O_CREATE, 0644)
  if err != nil {
    log.Fatal(err)
  }
  defer f.Close()

  time.Sleep(30 * time.Second)

  pprof.WriteHeapProfile(f)
  fmt.Println("Heap Profile generated")
}

两个函数分别会生成 cpu.prof 和 heap.prof 文件。仍然可以使用 go tool pprof 工具进行分析,在此就不赘述。

Trace 报告

直接贴代码:

// 生成追踪报告
func traceProfile() {
  f, err := os.OpenFile("trace.out", os.O_RDWR|os.O_CREATE, 0644)
  if err != nil {
    log.Fatal(err)
  }
  defer f.Close()

  log.Println("Trace started")
  trace.Start(f)
  defer trace.Stop()

  time.Sleep(60 * time.Second)
  fmt.Println("Trace stopped")
}

使用工具 go tool trace 进行分析,会得到非常详细的追踪报告,供更深入的程序分析优化。由于报告内容比较复杂,且使用方法类似,就不继续了。读者可自行尝试。

贴张网上的图给大家大概看一下:

Go程序性能优化及pprof使用方法详解

参考:https://github.com/caibirdme/hand-to-hand-optimize-go

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

标签:
Go,pprof

南强小屋 Design By 杰米
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
南强小屋 Design By 杰米

评论“Go程序性能优化及pprof使用方法详解”

暂无Go程序性能优化及pprof使用方法详解的评论...

《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线

暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。

艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。

《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。