Fix subpool tests

alitto · Oct 20, 2024 · bea3bad · bea3bad
1 parent a293ca8
commit bea3bad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/dispatcher/dispatcher.go b/internal/dispatcher/dispatcher.go
@@ -121,7 +121,7 @@ func (d *Dispatcher[T]) run(ctx context.Context) {
 				batch = batch[:0]
 			}
 
-			if !ok {
+			if !ok || d.closed.Load() {
 				// Channel was closed, read all remaining elements and exit
 				return
 			}