go 關(guān)閉channel分析

2019-11-10 19:43:20

字體：大中小

供稿：網(wǎng)友

背景

最近使用go開發(fā)后端服務(wù)，服務(wù)關(guān)閉需要保證channel中的數(shù)據(jù)都被讀取完，理由很簡單，在收到系統(tǒng)的中斷信號(hào)后，系統(tǒng)需要做收尾工作，保證channel的數(shù)據(jù)都要被處理掉，然后才可以關(guān)閉系統(tǒng)。

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

channel

關(guān)閉channelch := make(chan bool) close(ch) close(ch) // 這樣會(huì)panic的，channel不能close兩次向已經(jīng)關(guān)閉的channel寫數(shù)據(jù)ch := make(chan string) close(ch) ch <- "good" // 會(huì)panic的從已經(jīng)關(guān)閉的channel讀取數(shù)據(jù) 需要分兩種情況：無緩沖channel或者緩沖channel已經(jīng)讀取完畢緩沖channel未讀取完畢，可以繼續(xù)讀取channel中的剩余的數(shù)據(jù)//無緩沖channelch := make(chan string) close(ch) i := <- ch // 不會(huì)panic, i讀取到的值是空 "", 如果channel是bool的，那么讀取到的是false 判斷channel是否關(guān)閉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()}

輸出結(jié)果： 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.

說明收到中斷信號(hào)后，會(huì)關(guān)閉帶緩沖的channel ch、無緩沖的chSend、chConsume.從打印的日志可以看出 close ch之后，send的goroutine就結(jié)束了（可能打印send quit，也可能打印send to ch panic，可以多執(zhí)行幾次就會(huì)發(fā)現(xiàn)這種情況，原因就是select case有多個(gè)case滿足條件會(huì)隨機(jī)執(zhí)行一個(gè)case），此時(shí)還可以從ch繼續(xù)讀取channel中的數(shù)據(jù)（打印了read1 3 read2 4 read2 5），后面就打印了close ch2 說明ok是false，此時(shí)才知道ch已經(jīng)關(guān)閉。通過這個(gè)特性可以很優(yōu)雅的關(guān)閉服務(wù)。

如果服務(wù)被強(qiáng)行kill掉或者機(jī)器異常等情況，channel中的未讀取數(shù)據(jù)還是會(huì)丟失，系統(tǒng)設(shè)計(jì)需要允許這種情況的發(fā)生