golang 生产消费模式,如何保证不丢数据呢?

chazex · · 632 次点击 · 开始浏览    置顶

如下是一个简单的生产-消费的代码,当我们程序要退出,或者重启的时候,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群:701969077

632 次点击  
加入收藏 微博
8 回复  |  直到 2021-03-10 01:13:51
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传