[golang]go語言的channel學習
阿新 • • 發佈:2021-07-13
使用者故事:
有一個訊息生產者會生產10000條訊息,需要多個消費者來消費,當所有的訊息消費完畢,系統自動退出。
在golang裡面就可以使用chan通道的方法來解決這個問題。
建立一個生產者協程,多個消費者協程,使用有快取通道。
讀取chan時,設定超時條件,超過這個時間就認為消費完畢。
package main import ( "fmt" "sync" "time" ) // 建立sync.WaitGroup var wg = new(sync.WaitGroup) // 生產者 func producer(p chan<- int, count int) { for i := 0; i < count; i++ { p <- i // fmt.Println("send:", i) } wg.Done() } // 消費者 func consumer(c <-chan int, id int) { for { select { case v, ok := <-c: fmt.Printf("[%d]recv:[%d] [%v]\n", id, v, ok) // time.Sleep(100 * time.Millisecond) case err := <-time.After(1 * time.Second): fmt.Printf("[%d]recv timeout:[%v]\n", id, err) wg.Done() } } } func main() { ch := make(chan int, 100) wg.Add(1) go producer(ch, 100000) for i := 0; i < 50; i++ { wg.Add(1) go consumer(ch, i+1) } wg.Wait() }