fix: release semaphore at the proper time

1 files changed, 4 insertions, 3 deletions

diff --git a/classic.go b/classic.go
index bb22b0b..0b2bfa8 100644
--- a/classic.go
+++ b/classic.go
@@ -40,12 +40,13 @@ loop:
 
 				numPendingSends++
 				wg.Go(func() {
+					sema <- struct{}{}
+					defer func() { <-sema }()
+
 					select {
 					case <-ctx.Done():
 						return
-					case sema <- struct{}{}:
-						defer func() { <-sema }()
-						worklist <- getBatch(u)
+					case worklist <- getBatch(u):
 					}
 				})
 			}