channel 与 并行化

创建channel时,可以制定chanel的缓存:

c := make(chan int)   这种情况下缓存为1

c := make(chan int, num) 创建缓存为num的channel

为了灵活控制程序中协程的个数,可以通过创建一个带缓存的channel,在启动一个新的goroutine时,往channel中写入一个数据,然后在协程完成的时候读出来,实现控制协程个数等同于channel缓存大小的效果。

func RunClean() {
   cfg := initConfig()
   //初始化一个带缓存的chanel
   workerChan := make(chan int, cfg.WorkerNum)
   namespaces, err := QueryNamespaces()
   if err != nil {
      log.Error("Get namespaces error: ", err)
      return
   }
   step := cfg.Step
   last, err := QueryLastItem()
   if err != nil {
      log.Error("Get last item error:", err)
      return
   }
 
   for i := 0; i < last; i += step {
        //往channel中写入一个数字
        workerChan <- 1
        go Clean(int64(i), int64(i + Step))
   }
      }
   }
}
 
func Clean(beginId, endId int64) {
   cfg := &CleanTaskConfig{
      BeginId:   beginId,
      EndId:     endId,
   }
   if t, err := NewCleanTask(cfg); err == nil {
      t.CleanItems()
   }
   //当一个协程完成的时候,从chanel中读一个数字
   <-workerChan
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值