fix review part6

spqrcor · Sep 20, 2024 · 55619cc · 55619cc
1 parent 29271dd
commit 55619cc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/workers/order.go b/internal/workers/order.go
@@ -29,6 +29,8 @@ func (w *OrderWorker) Run() {
 	for i := 1; i <= w.workerCount; i++ {
 		errorCh := w.worker()
 		select {
+		case <-w.ctx.Done():
+			return
 		case err := <-errorCh:
 			if err != nil && retryCount < w.retryWorkerCount {
 				w.worker()