用户故事:
有一个消息生产者会生产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()
}