Merge pull request kubewharf#304 from Poor12/add-mcs-api

feat: support mcs

.gitattributes

@@ -1,3 +1,4 @@
 zz_generated* linguist-generated=true
 pkg/client linguist-generated=true
-config/crds/*.yaml linguist-generated=true
+config/crds/bases/core_kubeadmiral_io/*.yaml linguist-generated=true
+config/crds/bases/multicluster_x-k8s_io/*.yaml linguist-generated=true

cmd/controller-manager/app/controllermanager.go

@@ -47,6 +47,8 @@ const (
 	AutoMigrationControllerName            = "auto-migration"
 	StatusAggregatorControllerName         = "status-aggregator"
 	FederatedHPAControllerName             = "federated-hpa"
+	ServiceExportControllerName            = "serviceexport"
+	ServiceImportControllerName            = "serviceimport"
 )
 
 var knownControllers = map[string]controllermanager.StartControllerFunc{
@@ -62,6 +64,8 @@ var knownControllers = map[string]controllermanager.StartControllerFunc{
 	AutoMigrationControllerName:            startAutoMigrationController,
 	StatusAggregatorControllerName:         startStatusAggregatorController,
 	FederatedHPAControllerName:             startFederatedHPAController,
+	ServiceExportControllerName:            startServiceExportController,
+	ServiceImportControllerName:            startServiceImportController,
 }
 
 var controllersDisabledByDefault = sets.New[string]()

cmd/controller-manager/app/core.go

@@ -29,6 +29,7 @@ import (
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/federatedcluster"
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/federatedhpa"
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/follower"
+	"github.com/kubewharf/kubeadmiral/pkg/controllers/mcs"
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/nsautoprop"
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/override"
 	"github.com/kubewharf/kubeadmiral/pkg/controllers/policyrc"
@@ -344,3 +345,46 @@ func startFederatedHPAController(
 
 	return federatedHPAController, nil
 }
+
+func startServiceExportController(
+	ctx context.Context,
+	controllerCtx *controllercontext.Context,
+) (controllermanager.Controller, error) {
+	serviceExportController, err := mcs.NewServiceExportController(
+		controllerCtx.KubeClientset,
+		controllerCtx.KubeInformerFactory.Discovery().V1beta1().EndpointSlices(),
+		controllerCtx.FederatedInformerManager,
+		klog.Background(),
+		controllerCtx.Metrics,
+		controllerCtx.WorkerCount,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("error creating serviceexport controller: %w", err)
+	}
+
+	go serviceExportController.Run(ctx)
+
+	return serviceExportController, nil
+}
+
+func startServiceImportController(
+	ctx context.Context,
+	controllerCtx *controllercontext.Context,
+) (controllermanager.Controller, error) {
+	serviceImportController, err := mcs.NewServiceImportController(
+		controllerCtx.KubeClientset,
+		controllerCtx.KubeInformerFactory.Discovery().V1beta1().EndpointSlices(),
+		controllerCtx.FedClientset,
+		controllerCtx.FedInformerFactory.Core().V1alpha1().FederatedObjects(),
+		controllerCtx.Metrics,
+		klog.Background(),
+		controllerCtx.WorkerCount,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("error creating serviceimport controller: %w", err)
+	}
+
+	go serviceImportController.Run(ctx)
+
+	return serviceImportController, nil
+}

config/sample/host/01-ftc.yaml

@@ -422,3 +422,45 @@ spec:
     fields:
       - metadata.creationTimestamp
       - status
+---
+apiVersion: core.kubeadmiral.io/v1alpha1
+kind: FederatedTypeConfig
+metadata:
+  name: serviceimports.multicluster.x-k8s.io
+spec:
+  sourceType:
+    kind: ServiceImport
+    pluralName: serviceimports
+    scope: Namespaced
+    version: v1alpha1
+    group: multicluster.x-k8s.io
+  controllers:
+    - - kubeadmiral.io/global-scheduler
+    - - kubeadmiral.io/overridepolicy-controller
+---
+apiVersion: core.kubeadmiral.io/v1alpha1
+kind: FederatedTypeConfig
+metadata:
+  name: serviceexports.multicluster.x-k8s.io
+spec:
+  sourceType:
+    kind: ServiceExport
+    pluralName: serviceexports
+    scope: Namespaced
+    version: v1alpha1
+    group: multicluster.x-k8s.io
+  controllers:
+    - - kubeadmiral.io/global-scheduler
+    - - kubeadmiral.io/overridepolicy-controller
+---
+apiVersion: core.kubeadmiral.io/v1alpha1
+kind: FederatedTypeConfig
+metadata:
+  name: endpointslices.discovery.k8s.io
+spec:
+  sourceType:
+    kind: EndpointSlice
+    pluralName: endpointslices
+    scope: Namespaced
+    version: v1beta1
+    group: discovery.k8s.io

go.mod

@@ -30,6 +30,7 @@ require (
 	sigs.k8s.io/controller-runtime v0.14.1
 	sigs.k8s.io/custom-metrics-apiserver v1.27.0
 	sigs.k8s.io/kind v0.17.0
+	sigs.k8s.io/mcs-api v0.1.0
 )
 
 require (