Skip to content

Commit

Permalink
Add support for custom KubeClient (flyteorg#154)
Browse files Browse the repository at this point in the history
* Add support for custom KubeClient

Signed-off-by: Filipe Regadas <filiperegadas@gmail.com>

* review

Signed-off-by: Filipe Regadas <filiperegadas@gmail.com>
  • Loading branch information
regadas authored Mar 17, 2021
1 parent 93ed208 commit cef76a8
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions flyteplugins/go/tasks/pluginmachinery/k8s/plugin.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit cef76a8

Please sign in to comment.