sync/waitgroup: initial sync.WaitGroup wrapper with context support #319
@ -57,8 +57,9 @@ func (g *WaitGroup) WaitContext(ctx context.Context) {
|
|||||||
|
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
|
g.drain.Store(true)
|
||||||
g.wg.Add(-int(g.c.Load()))
|
g.wg.Add(-int(g.c.Load()))
|
||||||
<-done
|
g.drain.Store(false)
|
||||||
return
|
return
|
||||||
case <-done:
|
case <-done:
|
||||||
return
|
return
|
||||||
|
Loading…
Reference in New Issue
Block a user