Golang Go语言中多个协程池如何优雅退出

#10 context 是上下文啊，用来把 waitgroup 传递下去的，让不管多下层都能用同一个 waitgroup 并且能通过 context 感知上层要主动关闭 比如超时，而上层也可以通过 waitgroup 来感知下层任务是否已经完成，不管你下层开多少个协程，你每一层都是知道数量的，添加进去就好了

协程池本身就应该自带控制协程数量的属性，否则协程池还不如直接 go 。你看我上面写的也是 pool.Go

和四楼的一个意思 单还是谢谢回复：）

先说结论：要在 main 中感知任务的完成状态可以通过 chan，控制数据流转也是用 chan，如果有超时机制或者每个任务属性之类数据需要用到 context 。

1 ）步骤大约是
main 函数中创建三个用于接受结果的 A B C chan，大小为 A 任务个数，和一个 notify chan 用于完成通知，再来一个通知主程序关闭的 doneChan

起一个 goruntine 去 for notify chan
go func() {
for a := range nChan {
if (ACount == BCount == CCount == 100) {
doneChan <- struct{}
}

}
}


起三个 goruntine 去 for 分别处理 A B C chan，里面写处理逻辑，大约是这样子
go func(nChan chan Notify) {
for a := range AChan {
// dosomething

if ok {
bChan <- weiboEntryID
} else {
nChan <- weiboEntryID
}
}
close(bChan)
}(nChan)

go func(chan Notify) {
for b := range BChan {
// dosomething

if ok {
cChan <- commentID
} else {
nChan <- commentID
}
}
close(cChan)
}(nChan)

go func(chan Notify) {
for c := range CChan {
// dosomething

if ok {
cChan <- commentUserID
} else {
nChan <- commentUserID
}
}
}(nChan)

这时候就 main 中可以往 AChan 里写数据了，写完直接 close AChan，然后直接用 doneChan 阻塞

for userID := range userIDs {
AChan <- userID
}
close(aChan)
<-doneChan


大致流程就是这样，需要注意的是需要正确关闭 ABC chan，就是在发送完成后关闭，nChan 用于任务信息回收还可以用于任务回放，另外为了保证每个任务都会返回，需要弄一个 timeout context 超时当作失败处理

2 ）第二种情况，由于任务个数是未知的，上面的 100 就不能用了，有两种方式可以解决，一是可以预先知道每次数量，用另外的一个 chan 再任务开始时候传给发送 doneChan 那个 goruntine，把 100 改成最后总数即可。二是未知任务数量，这时候只能再来一个 chan，在做完 ABC 后发送一个 aDone 、bDone 、cDone，给发送 doneChan 那个 goruntine，当确认三个都完成后，就可以发送 done 了。

------

code 硬敲的可能很多不对，变量和具体结构可酌情改变 :-)