Go 超时控制导致内存泄露

这是一个很经典的面试题，例如现在面试官让你写一个超时控制函数，你是否有注意到内存泄露？

package main

import (
    "context"
    "fmt"
    "time"
)

func job() error {
    ctx, _ := context.WithTimeout(context.Background(), time.Second)
    done := make(chan struct{})
    go func() {
        //耗时操作
        time.Sleep(time.Millisecond * 1200)
        done <- struct{}{}
    }()
    select {
    case <-ctx.Done():
        return fmt.Errorf("time out!")
    case <-done:
        return nil
    }
}

func main() {
    fmt.Println(job())
}

如果你的代码是这样写的，那么恭喜你，内存泄露了！

我们修改一下 main 函数：

func main() {
    for i := 0; i <= 50; i++ {
        go func() {
            job()
        }()
    }
    for {
        time.Sleep(time.Second * 2)
        fmt.Println(runtime.NumGoroutine())
    }

}

执行一下再看看，效果很神奇。

它会一直打印 52，证明我们上面开的 50 个协程一直没有被释放。

问题其实就出在我画的圈里面，因为每一个程序都超时了，那么就租到了 ←ctx.Done() 这个 case 中去，Job 就算是完成了，但是我们的 ←done 就无法取值了， done ← strcut{}{} 就会被阻塞在这里了。这个协程就永远结束不了。

解决办法就是给 done 一个长度：

done := make(chan struct{}, 1)

只要协程能够往下走，job 中的协程就会被 gc。