Wait group for collecting goroutine information.
在go语言
waitGroup和errGroup都是用来控制goroutine的并发的方式，前者只能等待所有goroutine执行完成之后再执行Wait()函数后面的代码并且不能捕获运行中的错误，而后者能解决在goroutine运行出现的错误还能继续，但是只能捕获到第一次出错的goroutine的错误信息。有时候我们需要让多个协程在其中几个出错的时候还能正常运行其他的协程，并且还能捕获到出错协程的相关信息，前面2个waitGroup和errGroup都不能够满足我们的需求，所以打算自己动手实现一个collectGroup。
go get -u github.com/higker/collgroup- 能够支持
context - 能够获取错误信息
当然我们使用
errGroup加channel也可以实现，但是笔者想自己撸一个单独包。
我们在执行多个任务的时候，启动了多个协程，但是我们不能确定这些协程在运行的时候会不会出现问题，而出现了什么样的问题，怎么获取到error消息，现在我们通过collectGroup就可以实现了。
type task func() error
func TestCollGroup(t *testing.T) {
// 创建一个collectGroup
g := new(Group)
// 模拟多任务
tasks := []task{
func() error {
time.Sleep(4 * time.Second)
fmt.Println("task 1 done.")
return nil
},
func() error {
time.Sleep(2 * time.Second)
fmt.Println("task 2 done.")
return nil
},
func() error {
time.Sleep(3 * time.Second)
fmt.Println("task 3 done.")
return nil
},
// 出错任务
func() error {
time.Sleep(3 * time.Second)
return errors.New("task 4 running error")
},
func() error {
time.Sleep(3 * time.Second)
return errors.New("task 5 running error")
},
}
g.Errs = make(map[string]error, cap(tasks))
for i, t := range tasks {
g.Go(fmt.Sprintf("go-id-%s", cast.ToString(i)), t)
}
if g.Wait() {
fmt.Println("Get errors: ", g.Errs)
} else {
fmt.Println("run all task successfully!")
}
}output
=== RUN TestCollGroup
collect_group_test.go:34: task 2 done.
collect_group_test.go:39: task 3 done.
collect_group_test.go:29: task 1 done.
collect_group_test.go:57: Get errors: map[go-id-3:task 4 running error go-id-4:task 5 running error]
--- PASS: TestCollGroup (4.00s)
PASS
ok github.com/higker/collgroup 4.012s本案例是检测到一个错误就返回了，使用context完成
func TestWithContext(t *testing.T) {
// 创建一个errGroup
group, ctx := WithContext(context.Background())
// 模拟多任务
tasks := []task{
func() error {
time.Sleep(4 * time.Second)
t.Log("向订单表加入消息....")
return nil
},
func() error {
time.Sleep(2 * time.Second)
t.Log("更新库存消息....")
return nil
},
func() error {
time.Sleep(3 * time.Second)
t.Log("发送用户通知.....")
return nil
},
func() error {
time.Sleep(4 * time.Second)
return errors.New("用户扣款发送错误")
},
}
for i, t := range tasks {
group.Go(fmt.Sprintf("go-id-%s", cast.ToString(i)), t)
}
6A7B
// group.Wait()
// 监听任务出错了一个就返回
<-ctx.Done()
if len(group.Errs) > 0 {
t.Log("group exit...任务出，拿到错误消息回滚业务....")
t.Log("Get errors: ", group.Errs)
}
}output
go test -v
=== RUN TestCollGroup
collect_group_test.go:35: task 2 done.
collect_group_test.go:40: task 3 done.
collect_group_test.go:30: task 1 done.
collect_group_test.go:58: Get errors: map[go-id-3:task 4 running error go-id-4:task 5 running error]
--- PASS: TestCollGroup (4.00s)
=== RUN TestWithContext
collect_group_test.go:77: 更新库存消息....
collect_group_test.go:82: 发送用户通知.....
collect_group_test.go:72: 向订单表加入消息....
collect_group_test.go:98: group exit...任务出，拿到错误消息回滚业务....
collect_group_test.go:99: Get errors: map[go-id-3:用户扣款发送错误]
--- PASS: TestWithContext (4.01s)
PASS
ok github.com/higker/collgroup 8.013s