feat(backend): Add CLI flags to support Kubernetes native API implementation (#11907)

Signed-off-by: VaniHaripriya <[email protected]>

Author: VaniHaripriya

backend/src/apiserver/client_manager/client_manager.go

@@ -117,6 +117,7 @@ type ClientManager struct {
 // Options to pass to Client Manager initialization
 type Options struct {
 	UsePipelineKubernetesStorage bool
+	GlobalKubernetesWebhookMode  bool
 	Context                      context.Context
 	WaitGroup                    *sync.WaitGroup
 }
@@ -210,7 +211,7 @@ func (c *ClientManager) init(options *Options) error {
 
 	var pipelineStoreForRef storage.PipelineStoreInterface
 
-	if options.UsePipelineKubernetesStorage || common.IsOnlyKubernetesWebhookMode() {
+	if options.UsePipelineKubernetesStorage || options.GlobalKubernetesWebhookMode {
 		glog.Info("Initializing controller client...")
 		restConfig, err := util.GetKubernetesConfig()
 		if err != nil {
@@ -219,7 +220,7 @@ func (c *ClientManager) init(options *Options) error {
 
 		var cacheConfig map[string]cache.Config
 
-		if !common.IsMultiUserMode() && common.GetPodNamespace() != "" && !common.IsOnlyKubernetesWebhookMode() {
+		if !common.IsMultiUserMode() && common.GetPodNamespace() != "" && !options.GlobalKubernetesWebhookMode {
 			cacheConfig = map[string]cache.Config{common.GetPodNamespace(): {}}
 		}
 
@@ -259,7 +260,7 @@ func (c *ClientManager) init(options *Options) error {
 
 		c.controllerClient = controllerClient
 		c.controllerClientNoCache = controllerClientNoCache
-		if common.IsOnlyKubernetesWebhookMode() {
+		if options.GlobalKubernetesWebhookMode {
 			return nil
 		}
 

backend/src/apiserver/common/config.go

@@ -32,7 +32,6 @@ const (
 	KubeflowUserIDPrefix                    string = "KUBEFLOW_USERID_PREFIX"
 	UpdatePipelineVersionByDefault          string = "AUTO_UPDATE_PIPELINE_DEFAULT_VERSION"
 	TokenReviewAudience                     string = "TOKEN_REVIEW_AUDIENCE"
-	GlobalKubernetesWebhookMode             string = "GLOBAL_KUBERNETES_WEBHOOK_MODE"
 )
 
 func IsPipelineVersionUpdatedByDefault() bool {
@@ -128,7 +127,3 @@ func GetKubeflowUserIDPrefix() string {
 func GetTokenReviewAudience() string {
 	return GetStringConfigWithDefault(TokenReviewAudience, DefaultTokenReviewAudience)
 }
-
-func IsOnlyKubernetesWebhookMode() bool {
-	return GetBoolConfigWithDefault(GlobalKubernetesWebhookMode, false)
-}

backend/src/apiserver/main.go

@@ -70,6 +70,8 @@ var (
 	sampleConfigPath              = flag.String("sampleconfig", "", "Path to samples")
 	collectMetricsFlag            = flag.Bool("collectMetricsFlag", true, "Whether to collect Prometheus metrics in API server.")
 	usePipelinesKubernetesStorage = flag.Bool("pipelinesStoreKubernetes", false, "Store and run pipeline versions in Kubernetes")
+	disableWebhook                = flag.Bool("disableWebhook", false, "Set this if pipelinesStoreKubernetes is on but using a global webhook in a separate pod")
+	globalKubernetesWebhookMode   = flag.Bool("globalKubernetesWebhookMode", false, "Set this to run exclusively in Kubernetes Webhook mode")
 )
 
 type RegisterHttpHandlerFromEndpoint func(ctx context.Context, mux *runtime.ServeMux, endpoint string, opts []grpc.DialOption) error
@@ -93,6 +95,7 @@ func main() {
 
 	options := &cm.Options{
 		UsePipelineKubernetesStorage: *usePipelinesKubernetesStorage,
+		GlobalKubernetesWebhookMode:  *globalKubernetesWebhookMode,
 		Context:                      backgroundCtx,
 		WaitGroup:                    &wg,
 	}
@@ -122,9 +125,13 @@ func main() {
 	}
 
 	defer clientManager.Close()
-	webhookOnlyMode := common.IsOnlyKubernetesWebhookMode()
+	webhookOnlyMode := *globalKubernetesWebhookMode
+
+	if (*usePipelinesKubernetesStorage && !*disableWebhook) || webhookOnlyMode {
+		if *disableWebhook && webhookOnlyMode {
+			glog.Fatalf("Invalid configuration: globalKubernetesWebhookMode is enabled but the webhook is disabled")
+		}
 
-	if *usePipelinesKubernetesStorage || webhookOnlyMode {
 		wg.Add(1)
 		webhookServer, err := startWebhook(
 			clientManager.ControllerClient(true), clientManager.ControllerClient(false), &wg,
@@ -147,6 +154,7 @@ func main() {
 			wg.Wait()
 			return
 		}
+
 	}
 
 	resourceManager := resource.NewResourceManager(