Add support for custom KubeClient (flyteorg#154)

* Add support for custom KubeClient Signed-off-by: Filipe Regadas <[email protected]> * review Signed-off-by: Filipe Regadas <[email protected]> Signed-off-by: Chao-Han Tsai <[email protected]>
milton0825 · May 25, 2021 · c4e6f53 · c4e6f53
1 parent 30316ac
commit c4e6f53
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/go/tasks/pluginmachinery/k8s/plugin.go b/go/tasks/pluginmachinery/k8s/plugin.go
@@ -32,6 +32,14 @@ type PluginEntry struct {
 	// support the same task type. This must be a subset of RegisteredTaskTypes and at most one default per task type
 	// is supported.
 	DefaultForTaskTypes []pluginsCore.TaskType
+	// Returns a new KubeClient to be used instead of the internal controller-runtime client.
+	CustomKubeClient func(ctx context.Context) (pluginsCore.KubeClient, error)
+	// Boolean that indicates if kubernetes resources that this plugin is responsible for should include OwnerReferences.
+	// Ingoring is only useful if resources will be created in a remote cluster.
+	OverrideInjectOwnerReferences *bool
+	// Boolean flag that indicates if a finalizer should be injected.
+	// This will override the `inject-finalizer` set under k8s config.
+	OverrideInjectFinalizer *bool
 }
 
 // Special context passed in to plugins when checking task phase