fix: synchronize Group's err

This commit is contained in:
Andrey 'kondor' Sichevoy
2022-10-13 13:50:32 +02:00
committed by Andrew Sichevoi
parent c7a74b9be5
commit 7a514efd01
+14 -3
View File
@@ -34,7 +34,11 @@ type TaskGroupWithContext struct {
TaskGroup
ctx context.Context
cancel context.CancelFunc
errOnce sync.Once
errSync struct {
once sync.Once
guard sync.RWMutex
}
err error
}
@@ -57,8 +61,11 @@ func (g *TaskGroupWithContext) Submit(task func() error) {
// don't actually ignore errors
err := task()
if err != nil {
g.errOnce.Do(func() {
g.errSync.once.Do(func() {
g.errSync.guard.Lock()
g.err = err
g.errSync.guard.Unlock()
if g.cancel != nil {
g.cancel()
}
@@ -86,5 +93,9 @@ func (g *TaskGroupWithContext) Wait() error {
case <-g.ctx.Done():
}
return g.err
g.errSync.guard.RLock()
err := g.err
g.errSync.guard.RUnlock()
return err
}