用go的goroutine和channel实现一个简单的"生产、消费"(带有超时控制)小例子

直接上代码

1、没有超时的时候

package main
import (
 "fmt"
 "time"
)
func main() {
 //存放生产的channel
 jobChan := make(chan int, 100)
 //通知是否完成所有任务
 endChan := make(chan bool)
 go production(jobChan)
 go worker(jobChan, endChan)
 select {
 case <-endChan:
 fmt.Println("消费完成........................")
 return
 case <-time.After(time.Second * 20):
 fmt.Println("超时...........................")
 return
 }
}
//消费
func worker(jobChan <-chan int, endChan chan bool) {
 for job := range jobChan {
 fmt.Println("消费:", job)
 }
 //消费结束,通知endChan
 endChan <- true
}
//生产
func production(jobChan chan<- int) {
 for i := 1; i <= 10; i++ {
 fmt.Println("生产:", i)
 jobChan <- i
 }
 //关闭channel防止消费阻塞
 close(jobChan)
}

执行结果 :
这里写图片描述
可见生产和消费异步并发执行。

2、当超时时 (timeout设置小点,并增加消费处理时间)

package main
import (
 "fmt"
 "time"
)
func main() {
 //存放生产的channel
 jobChan := make(chan int, 100)
 //通知是否完成所有任务
 endChan := make(chan bool)
 go production(jobChan)
 go worker(jobChan, endChan)
 select {
 case <-endChan:
 fmt.Println("消费完成........................")
 return
 case <-time.After(time.Second * 5):
 fmt.Println("超时...........................")
 return
 }
}
//消费
func worker(jobChan <-chan int, endChan chan bool) {
 for job := range jobChan {
 fmt.Println("消费:", job)
 //增加消费处理时间
 time.Sleep(1 * time.Second)
 }
 //消费结束,通知endChan
 endChan <- true
}
//生产
func production(jobChan chan<- int) {
 for i := 1; i <= 10; i++ {
 fmt.Println("生产:", i)
 jobChan <- i
 }
 //关闭channel防止消费阻塞
 close(jobChan)
}

执行结果
这里写图片描述
5秒之后超时(此时队列里尚有未处理完的数据)。

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传