fix race on error member of TaskRunner

authzed · Mar 14, 2024 · 869723c · 869723c
1 parent 33ae2e5
commit 869723c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/taskrunner/taskrunner.go b/internal/taskrunner/taskrunner.go
@@ -160,5 +160,8 @@ func (tr *TaskRunner) emptyForCancel() {
 // or the parent context to have been canceled.
 func (tr *TaskRunner) Wait() error {
 	tr.wg.Wait()
+
+	tr.lock.Lock()
+	defer tr.lock.Unlock()
 	return tr.err
 }