diff --git a/pool.go b/pool.go index eb52dee..63c86cc 100644 --- a/pool.go +++ b/pool.go @@ -777,9 +777,10 @@ func (pool *SimplePool) BatchedSubManyEose( select { case res <- ie: case <-ctx.Done(): + wg.Done() + return } } - wg.Done() }(df) }