refactor: override controller

kubewharf · Jul 26, 2023 · 7286f96 · 7286f96
1 parent ea75f7c
commit 7286f96
Show file tree

Hide file tree

Showing 8 changed files with 316 additions and 297 deletions.
diff --git a/cmd/controller-manager/app/controllermanager.go b/cmd/controller-manager/app/controllermanager.go
@@ -39,10 +39,12 @@ const (
 	FederateControllerName         = "federate"
 	MonitorControllerName          = "monitor"
 	FollowerControllerName         = "follower"
+	OverrideControllerName         = "overridepolicy"
 )
 
 var knownControllers = map[string]controllermanager.StartControllerFunc{
 	FederateControllerName: startFederateController,
+	OverrideControllerName: startOverridePolicyController,
 }
 
 var controllersDisabledByDefault = sets.New(MonitorControllerName)

diff --git a/cmd/controller-manager/app/core.go b/cmd/controller-manager/app/core.go
@@ -25,6 +25,7 @@ import (
 	"github.com/kubewharf/kubeadmiral/pkg/controllermanager"
 	controllercontext "github.com/kubewharf/kubeadmiral/pkg/controllers/context"
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/federate"
+	"github.com/kubewharf/kubeadmiral/pkg/controllers/override"
 )
 
 func startFederateController(
@@ -51,3 +52,22 @@ func startFederateController(
 
 	return federateController, nil
 }
+
+func startOverridePolicyController(ctx context.Context, controllerCtx *controllercontext.Context) (controllermanager.Controller, error) {
+	overrideController, err := override.NewOverridePolicyController(
+		controllerCtx.KubeClientset,
+		controllerCtx.FedClientset,
+		controllerCtx.FedInformerFactory,
+		controllerCtx.InformerManager,
+		controllerCtx.Metrics,
+		klog.Background(),
+		controllerCtx.WorkerCount,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("error creating override controller: %w", err)
+	}
+
+	go overrideController.Run(ctx)
+
+	return overrideController, nil
+}