Avoid race in subsequent changes to the same controller

sysdiglabs · Dec 19, 2022 · 19927c1 · 19927c1
1 parent 83e45df
commit 19927c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/clusterloader2/pkg/measurement/common/wait_for_controlled_pods.go b/clusterloader2/pkg/measurement/common/wait_for_controlled_pods.go
@@ -615,10 +615,10 @@ func (w *waitForControlledPodsRunningMeasurement) waitForRuntimeObject(obj runti
 	o := newObjectChecker(key)
 	o.lock.Lock()
 	defer o.lock.Unlock()
+	ctx, cancel := context.WithCancel(ctx)
+	o.cancel = cancel
 	w.handlingGroup.Start(func() {
-		ctx, cancel := context.WithCancel(ctx)
 		defer cancel()
-		o.SetCancel(cancel)
 		if operationTimeout != time.Duration(0) {
 			ctx, cancel = context.WithTimeout(ctx, operationTimeout)
 			defer cancel()