add composable PluginCleanupPolicy

flyteorg · Aug 23, 2021 · 20d76da · 20d76da
1 parent 892f35e
commit 20d76da
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/go/tasks/pluginmachinery/k8s/mocks/plugin_cleanup_policy.go b/go/tasks/pluginmachinery/k8s/mocks/plugin_cleanup_policy.go
diff --git a/go/tasks/pluginmachinery/k8s/plugin.go b/go/tasks/pluginmachinery/k8s/plugin.go
@@ -91,3 +91,8 @@ type Plugin interface {
 	// Properties desired by the plugin
 	GetProperties() PluginProperties
 }
+
+// Defines functions to clean up the resource object after the task has been completed or cancelled.
+type PluginCleanupPolicy interface {
+	OnAbort(ctx context.Context, kubeClient client.Client, resource client.Object) error
+}