refactor: follower controller

kubewharf · Jul 21, 2023 · 739969d · 739969d
1 parent e56fb93
commit 739969d
Show file tree

Hide file tree

Showing 7 changed files with 271 additions and 237 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/cmd/controller-manager/app/controllermanager.go b/cmd/controller-manager/app/controllermanager.go
@@ -43,6 +43,7 @@ const (
 
 var knownControllers = map[string]controllermanager.StartControllerFunc{
 	FederateControllerName: startFederateController,
+	FollowerControllerName: startFollowerController,
 }
 
 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/follower"
 )
 
 func startFederateController(
@@ -51,3 +52,22 @@ func startFederateController(
 
 	return federateController, nil
 }
+
+func startFollowerController(ctx context.Context, controllerCtx *controllercontext.Context) (controllermanager.Controller, error) {
+	controller, err := follower.NewFollowerController(
+		controllerCtx.KubeClientset,
+		controllerCtx.FedClientset,
+		controllerCtx.FedInformerFactory.Core().V1alpha1().FederatedObjects(),
+		controllerCtx.FedInformerFactory.Core().V1alpha1().ClusterFederatedObjects(),
+		controllerCtx.Metrics,
+		klog.Background(),
+		controllerCtx.WorkerCount,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("error creating follower controller: %w", err)
+	}
+
+	go controller.Run(ctx)
+
+	return controller, nil
+}
diff --git a/pkg/controllers/follower/bidirectional_cache.go b/pkg/controllers/follower/bidirectional_cache.go
@@ -1,4 +1,3 @@
-//go:build exclude
 /*
 Copyright 2023 The KubeAdmiral Authors.