Merge pull request ipfs/go-bitswap#504 from ipfs/fix/task-worker-coun…

…t-race fix: hold the task worker lock when starting task workers This commit was moved from ipfs/go-bitswap@d48cee0
Jorropo · Jun 24, 2021 · 8c62bb5 · 8c62bb5
2 parents d496037 + 0f4401d
commit 8c62bb5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bitswap/internal/decision/engine.go b/bitswap/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)