前言
最近使用go開發后端服務，服務關閉需要保證golang/71577.html">channel中的數據都被讀取完，理由很簡單，在收到系統的中斷信號后，系統需要做收尾工作，保證channel的數據都要被處理掉，然后才可以關閉系統。但實現起來沒那么簡單，下面來一起看看詳細的介紹吧。
關于Go channel設計和規范的批評：
所以Golang 內建的 close 方法可以關閉 channel，如果往已經關閉的 channel 發送數據，則會報錯：panic: close of closed channel.
看如下代碼，在一段時間內，生產者可以不斷往 channel 寫入數據，消費者進行處理，一段時間后 channel 關閉了，這個時候如果還有數據往 channel 發送，程序就會報錯。
package main import ( "fmt" "sync" "time") func main() { jobs := make(chan int) var wg sync.WaitGroup go func() { time.Sleep(time.Second * 3) close(jobs) }() go func() { for i := 0; ; i++ { jobs <- i fmt.Println("produce:", i) } }() wg.Add(1) go func() { defer wg.Done() for i := range jobs { fmt.Println("consume:", i) } }() wg.Wait()}多運行幾次出錯的概率會比較大：
produce: 33334consume: 33334consume: 33335produce: 33335produce: 33336consume: 33336consume: 33337produce: 33337produce: 33338consume: 33338consume: 33339produce: 33339produce: 33340consume: 33340panic: send on closed channel goroutine 19 [running]:panic(0x49b660, 0xc042410bb0) C:/Go/src/runtime/panic.go:500 +0x1afmain.main.func2(0xc04203a180) C:/Users/tanteng/Go/src/examples/channel_close.go:18 +0x6bcreated by main.main C:/Users/tanteng/Go/src/examples/channel_close.go:21 +0xb8exit status 2
如何優雅關閉 channel
那么在往通道發數據前如何判斷通道是否關閉呢？
1._,ok := <- jobs
此時如果 channel 關閉，ok 值為 false，如果 channel 沒有關閉，則會漏掉一個 jobs
2.使用 select 方式
再創建一個 channel，叫做 timeout，如果超時往這個 channel 發送 true，在生產者發送數據給 jobs 的 channel，用 select 監聽 timeout，如果超時則關閉 jobs 的 channel.
完整代碼如下：
package main import ( "fmt" "sync" "time") func main() { jobs := make(chan int) timeout := make(chan bool) var wg sync.WaitGroup go func() { time.Sleep(time.Second * 3) timeout <- true }() go func() { for i := 0; ; i++ { select { case <-timeout: close(jobs) return default: jobs <- i fmt.Println("produce:", i) } } }() wg.Add(1) go func() { defer wg.Done() for i := range jobs { fmt.Println("consume:", i) } }() wg.Wait()}這樣就可以保證不會往已經關閉的 channel 中發送數據了。
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對VEVB武林網的支持。
新聞熱點
疑難解答
