[K10-2154] Add tomb cancel onDelete ActionSet (#5348)

* Add tomb cancel onDelete ActionSet * Address review comments * Address review comments * Refactor variable * Refactor based on commits * Remove log msg
kanisterio · Apr 8, 2019 · e8ee60e · e8ee60e
1 parent 4872990
commit e8ee60e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/pkg/controller/controller.go b/pkg/controller/controller.go
@@ -238,7 +238,18 @@ func (c *Controller) onUpdateBlueprint(oldBP, newBP *crv1alpha1.Blueprint) error
 }
 
 func (c *Controller) onDeleteActionSet(as *crv1alpha1.ActionSet) error {
-	log.Infof("Deleted ActionSet %s", as.GetName())
+	asName := as.GetName()
+	log.Infof("Deleted ActionSet %s", asName)
+	v, ok := c.actionSetTombMap.Load(asName)
+	if !ok {
+		return nil
+	}
+	t, castOk := v.(*tomb.Tomb)
+	if !castOk {
+		return nil
+	}
+	t.Kill(nil) // TODO: @Deepika Give reason for ActionSet kill
+	c.actionSetTombMap.Delete(asName)
 	return nil
 }