Merge pull request #436 from GreatLazyMan/virtualcluster_join_controller

Virtualcluster join controller
kosmos-io · Apr 3, 2024 · 4df5b15 · 4df5b15
2 parents d4946d8 + 7e35c9d
commit 4df5b15
Show file tree

Hide file tree

Showing 4 changed files with 2,727 additions and 73 deletions.
diff --git a/cmd/clustertree-operator/app/options/options.go b/cmd/clustertree-operator/app/options/options.go
@@ -9,8 +9,9 @@ import (
 )
 
 type Options struct {
-	LeaderElection    componentbaseconfig.LeaderElectionConfiguration
-	KubernetesOptions KubernetesOptions
+	LeaderElection             componentbaseconfig.LeaderElectionConfiguration
+	KubernetesOptions          KubernetesOptions
+	AllowNodeOwnbyMulticluster bool
 }
 
 type KubernetesOptions struct {
@@ -37,10 +38,11 @@ func (o *Options) AddFlags(flags *pflag.FlagSet) {
 	}
 
 	flags.BoolVar(&o.LeaderElection.LeaderElect, "leader-elect", true, "Start a leader election client and gain leadership before executing the main loop. Enable this when running replicated components for high availability.")
-	flags.StringVar(&o.LeaderElection.ResourceName, "leader-elect-resource-name", "network-manager", "The name of resource object that is used for locking during leader election.")
+	flags.StringVar(&o.LeaderElection.ResourceName, "leader-elect-resource-name", "clustertree-operator", "The name of resource object that is used for locking during leader election.")
 	flags.StringVar(&o.LeaderElection.ResourceNamespace, "leader-elect-resource-namespace", utils.DefaultNamespace, "The namespace of resource object that is used for locking during leader election.")
 	flags.Float32Var(&o.KubernetesOptions.QPS, "kube-qps", 40.0, "QPS to use while talking with kube-apiserver.")
 	flags.IntVar(&o.KubernetesOptions.Burst, "kube-burst", 60, "Burst to use while talking with kube-apiserver.")
 	flags.StringVar(&o.KubernetesOptions.KubeConfig, "kubeconfig", "", "Path for kubernetes kubeconfig file, if left blank, will use in cluster way.")
 	flags.StringVar(&o.KubernetesOptions.Master, "master", "", "Used to generate kubeconfig for downloading, if not specified, will use host in kubeconfig.")
+	flags.BoolVar(&o.AllowNodeOwnbyMulticluster, "multiowner", false, "Allow node own by multicluster or not.")
 }
diff --git a/cmd/clustertree-operator/app/tree_operator.go b/cmd/clustertree-operator/app/tree_operator.go
@@ -83,9 +83,10 @@ func run(ctx context.Context, opts *options.Options) error {
 	}
 
 	VirtualClusterJoinController := controller.VirtualClusterJoinController{
-		Client:         mgr.GetClient(),
-		EventRecorder:  mgr.GetEventRecorderFor(constants.JoinControllerName),
-		KubeconfigPath: opts.KubernetesOptions.KubeConfig,
+		Client:                     mgr.GetClient(),
+		EventRecorder:              mgr.GetEventRecorderFor(constants.JoinControllerName),
+		KubeconfigPath:             opts.KubernetesOptions.KubeConfig,
+		AllowNodeOwnbyMulticluster: opts.AllowNodeOwnbyMulticluster,
 	}
 	if err = VirtualClusterJoinController.SetupWithManager(mgr); err != nil {
 		return fmt.Errorf("error starting %s: %v", constants.JoinControllerName, err)