go使用context包避免goroutine泄露問題

標籤：contex   fun   div   情況   ext   通知   sele   done   特性   

go是帶記憶體自動回收的特性，因此記憶體一般不會泄漏。但是Goroutine確存在泄漏的情況，同時泄漏的Goroutine引用的記憶體同樣無法被回收。

下面的程式中後台Goroutine向管道輸入自然數序列，main函數中輸出序列。但是當break跳出for迴圈的時候，後台Goroutine就處於無法被回收的狀態了。

func main() {    ch := func() <-chan int {        ch := make(chan int)        go func() {            for i := 0; ; i++ {                ch <- i            }        } ()        return ch    }()    for v := range ch {        fmt.Println(v)        if v == 5 {            break        }    }}

我們可以通過context包來避免這個問題：

func main() {    ctx, cancel := context.WithCancel(context.Background())    ch := func(ctx context.Context) <-chan int {        ch := make(chan int)        go func() {            for i := 0; ; i++ {                select {                case <- ctx.Done():                    return                case ch <- i:                }            }        } ()        return ch    }(ctx)    for v := range ch {        fmt.Println(v)        if v == 5 {            cancel()            break        }    }}

當main函數在break跳出迴圈時，通過調用cancel()來通知後台Goroutine退出，這樣就避免了Goroutine的泄漏。

 

go使用context包避免goroutine泄露問題