Go语言pprof实战指南：排查内存泄露，优化程序性能

见解分享

2022-12-12 03:17:01

用 pprof 轻松修复 Go 中的内存泄露问题

什么是内存泄露？

内存泄露是一种可怕的计算机错误，会导致程序随着时间的推移占用越来越多的内存，最终导致程序崩溃。这就像在购买杂货时忘记结账，导致你的购物车不断填满，直到你无法再推着它。

pprof：你的内存泄露修复帮手

pprof 是 Go 语言的救星，它可以帮助你轻松识别和修复内存泄露问题。它就像一个超级英雄，拥有 X 射线视觉，可以深入程序内部，找出导致内存问题的罪魁祸首。

实战演示：炸弹程序

为了说明 pprof 的强大功能，我们创建一个会泄露内存的“炸弹程序”：

package main

import (
    "fmt"
    "runtime"
    "time"
)

func main() {
    go func() {
        // 分配 100MB 内存
        data := make([]byte, 100*1024*1024)
        // 阻止协程退出，保持内存泄露
        for {
            time.Sleep(time.Second)
        }
    }()

    // 等待 10 秒，让泄露发生
    time.Sleep(10 * time.Second)

    // 打印内存使用情况
    fmt.Println("内存使用情况：", runtime.MemStats{})
}

使用 pprof 排查泄露

现在，让我们用 pprof 来拯救局面：

运行炸弹程序：go run pprof-example.go
运行 pprof：go tool pprof -http=:8080 ./pprof-example
打开浏览器，访问 http://localhost:8080/debug/pprof/goroutine?debug=2
在“Blocking”下，你会看到一个被阻塞的协程，这就是导致泄漏的罪魁祸首。
点击协程的“Stack”链接，查看导致泄漏的代码。

找到泄漏源

在我们的炸弹程序中，我们可以看到泄漏来自分配 100MB 内存的代码：make([]byte, 100*1024*1024)。就像一个贪婪的孩子不停地往购物车里装东西一样，这个代码不断地给程序分配内存，却没有释放它。

修复泄漏

解决方法很简单：将内存分配移到协程之外，这样当协程结束时，内存就会被释放。修改后的代码如下：

package main

import (
    "fmt"
    "runtime"
    "time"
)

func main() {
    // 在主线程中分配 100MB 内存
    data := make([]byte, 100*1024*1024)

    go func() {
        // 使用 100MB 内存
        for {
            time.Sleep(time.Second)
        }
    }()

    // 等待 10 秒，让内存分配发生
    time.Sleep(10 * time.Second)

    // 打印内存使用情况
    fmt.Println("内存使用情况：", runtime.MemStats{})
}