go 關閉channel分析

2019-11-10 19:43:20

字體：大中小

來源：轉載

供稿：網友

背景

最近使用go開發后端服務，服務關閉需要保證channel中的數據都被讀取完，理由很簡單，在收到系統的中斷信號后，系統需要做收尾工作，保證channel的數據都要被處理掉，然后才可以關閉系統。

后面我會給出方案，見示例代碼，但在解決這個問題之前我們先了解下close channel的一些特性。

channel

關閉channelch := make(chan bool) close(ch) close(ch) // 這樣會panic的，channel不能close兩次向已經關閉的channel寫數據ch := make(chan string) close(ch) ch <- "good" // 會panic的從已經關閉的channel讀取數據需要分兩種情況：無緩沖channel或者緩沖channel已經讀取完畢緩沖channel未讀取完畢，可以繼續讀取channel中的剩余的數據//無緩沖channelch := make(chan string) close(ch) i := <- ch // 不會panic, i讀取到的值是空 "", 如果channel是bool的，那么讀取到的是false 判斷channel是否關閉i, ok := <- ch if ok { PRintln(i) } else { println("channel closed") }

方案

我直接上示例代碼

package mainimport ( "fmt" "os" "os/signal" "sync" "syscall" "time" )func main() { var wg sync.WaitGroup ch := make(chan int, 100) chSend := make(chan int) chConsume := make(chan int) sc := make(chan os.Signal, 1) signal.Notify(sc, os.Kill, os.Interrupt, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT) go func(ch, quit chan int) { defer func() { if err := recover(); err != nil { fmt.Println("send to ch panic.===", err) } }() i := 0 for { select { case ch <- i: fmt.Println("send", i) time.Sleep(time.Second) i++ case <-quit: fmt.Println("send quit.") return } } }(ch, chSend) go func(ch, quit chan int) { wg.Add(1) for { select { case i, ok := <-ch: if ok { fmt.Println("read1", i) time.Sleep(time.Second * 2) } else { fmt.Println("close ch1.") } case <-quit: for { select { case i, ok := <-ch: if ok { fmt.Println("read2", i) time.Sleep(time.Second * 2) } else { fmt.Println("close ch2.") goto L } } } L: fmt.Println("consume quit.") wg.Done() return } } }(ch, chConsume) <-sc close(ch) fmt.Println("close ch ") close(chSend) close(chConsume) wg.Wait()}

輸出結果： send 0 read1 0 send 1 send 2 read1 1 send 3 send 4 read1 2 send 5 close ch send quit. read1 3 read2 4 read2 5 close ch2. consume quit.

說明收到中斷信號后，會關閉帶緩沖的channel ch、無緩沖的chSend、chConsume.從打印的日志可以看出 close ch之后，send的goroutine就結束了（可能打印send quit，也可能打印send to ch panic，可以多執行幾次就會發現這種情況，原因就是select case有多個case滿足條件會隨機執行一個case），此時還可以從ch繼續讀取channel中的數據（打印了read1 3 read2 4 read2 5），后面就打印了close ch2 說明ok是false，此時才知道ch已經關閉。通過這個特性可以很優雅的關閉服務。

如果服務被強行kill掉或者機器異常等情況，channel中的未讀取數據還是會丟失，系統設計需要允許這種情況的發生