fix: hold the task worker lock when starting task workers

Otherwise, we could try to shutdown at the same time and race.
ipfs · Jun 24, 2021 · 24c356f · 24c356f
1 parent eac35a4
commit 24c356f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/decision/engine.go b/internal/decision/engine.go
@@ -235,6 +235,9 @@ func (e *Engine) StartWorkers(ctx context.Context, px process.Process) {
 	e.bsm.start(px)
 	e.startScoreLedger(px)
 
+	e.taskWorkerLock.Lock()
+	defer e.taskWorkerLock.Unlock()
+
 	for i := 0; i < e.taskWorkerCount; i++ {
 		px.Go(func(px process.Process) {
 			e.taskWorker(ctx)