分享
这是一个创建于 的主题,其中的信息可能已经有所发展或是发生改变。
如下是一个简单的生产-消费的代码,当我们程序要退出,或者重启的时候,channel里面的数据会丢失掉。
大家在使用生产消费者的时候,是怎么处理的呢?
```
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
"time"
)
var oriTasks chan int
var taskStage21 chan int
var taskStage22 chan int
var endTasks chan int
func init() {
oriTasks = make(chan int , 1000)
taskStage21 = make(chan int , 1000)
taskStage22 = make(chan int , 1000)
endTasks = make(chan int , 1000)
}
func main() {
// 模拟任务来源,一般从网络获取
go func() {
i := 0
for {
oriTasks <- i
i++
time.Sleep(10*time.Second)
}
}()
go consumerStage1()
go consumerStage21()
go consumerStage22()
go consumerStageEnd()
quit := make(chan os.Signal)
signal.Notify(quit, syscall.SIGINT, syscall.SIGKILL, syscall.SIGTERM)
<-quit
}
func consumerStage1() {
for {
select {
case tasks := <-oriTasks:
if tasks %2 ==0 {
// do something ....
taskStage22 <- tasks
} else {
// do other something ...
taskStage22 <- tasks
}
}
}
}
func consumerStage21() {
for {
select {
case tasks := <-oriTasks:
// stage2-2 biz
endTasks <- tasks * 2
}
}
}
func consumerStage22() {
for {
select {
case tasks := <-oriTasks:
// stage2-3 biz
endTasks <- tasks * 4
}
}
}
func consumerStageEnd() {
for {
select {
case tasks := <-endTasks:
// end task biz
r := tasks +1
fmt.Println(r)
}
}
}
```
有疑问加站长微信联系(非本文作者)
入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889
关注微信2040 次点击
添加一条新回复
(您需要 后才能回复 没有账号 ?)
- 请尽量让自己的回复能够对别人有帮助
- 支持 Markdown 格式, **粗体**、~~删除线~~、
`单行代码` - 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
- 图片支持拖拽、截图粘贴等方式上传