reschedule limit mode was broken - fixing the inc/dec (#572)

go-co-op · Sep 21, 2023 · 7a102da · 7a102da
1 parent 269c4d1
commit 7a102da
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/executor.go b/executor.go
@@ -154,6 +154,11 @@ func (e *executor) start() {
 }
 
 func (e *executor) runJob(f jobFunction) {
+	defer func() {
+		if e.limitMode == RescheduleMode && e.limitModeMaxRunningJobs > 0 {
+			e.limitModeRunningJobs.Add(-1)
+		}
+	}()
 	switch f.runConfig.mode {
 	case defaultMode:
 		lockKey := f.jobName
@@ -244,6 +249,7 @@ func (e *executor) run() {
 						if e.limitModeRunningJobs.Load() < int64(e.limitModeMaxRunningJobs) {
 							select {
 							case e.limitModeQueue <- f:
+								e.limitModeRunningJobs.Inc()
 							case <-e.ctx.Done():
 							}
 						}