检测 goroutine 泄漏

比如常见的异步生产消费模型，下面这段代码你能快速定位泄漏问题吗？如果你不停的访问这个 3000 端口，网站会内存泄漏。

package main

import (
	"fmt"
	"net/http"
)

func main() {
	_ = http.ListenAndServe(":3000", http.HandlerFunc(handle))
}

func handle(_ http.ResponseWriter, _ *http.Request) {
	c := make(chan int)
	go produce(c)
	go consume(c)
}

func produce(c chan int) {
	for i := range "..." {
		c <- i
	}
}

func consume(c chan int) {
	for i := range c {
		fmt.Println(i)
	}
}

我们可以用 gotrace 来自动定位泄漏点，添加一个 main_test.go 文件：

package main

import (
	"testing"

	"github.com/ysmood/gotrace"
)

func TestHandle(t *testing.T) {
	gotrace.CheckLeak(t, 0)

	handle(nil, nil)
}

运行 GODEBUG="tracebackancestors=10" go test，等待 3 秒后就可以看到打印信息如下：

--- FAIL: TestHandle (3.00s)
    main_test.go:10: leaking goroutines: goroutine 6 [chan receive]:
        play.consume(0x0)
        	/Users/ys/repos/play/default/main.go:25 +0x74

可以看到第 25 行的 range 卡住了，原因是 chan receive，它在等待 c 的新消息，然而 produce 函数已经退出不会再往 c 里发布消息了。

修正的方法很多，比如把 produce 修改为：

func produce(c chan int) {
	for i := range "..." {
		c <- i
	}
	close(c)
}

再运行 go test 测试就立马成功结束了。

想了解更多 gotrace 用法可以移步 https://github.com/ysmood/gotrace