fix: executor run block when executor.stop() (#475)

go-co-op · May 8, 2023 · 2156260 · 2156260
1 parent 2936a47
commit 2156260
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/executor.go b/executor.go
@@ -194,10 +194,16 @@ func (e *executor) run() {
 					switch e.limitMode {
 					case RescheduleMode:
 						if e.limitModeRunningJobs.Load() < int64(e.limitModeMaxRunningJobs) {
-							e.limitModeQueue <- f
+							select {
+							case e.limitModeQueue <- f:
+							case <-e.ctx.Done():
+							}
 						}
 					case WaitMode:
-						e.limitModeQueue <- f
+						select {
+						case e.limitModeQueue <- f:
+						case <-e.ctx.Done():
+						}
 					}
 					return
 				}