Merge pull request #504 from XiShanYongYe-Chang/derived-service

Derive service from ServiceImport

cmd/controller-manager/app/controllermanager.go

@@ -260,4 +260,12 @@ func setupControllers(mgr controllerruntime.Manager, opts *options.Options, stop
 	if err := endpointSliceController.SetupWithManager(mgr); err != nil {
 		klog.Fatalf("Failed to setup EndpointSlice controller: %v", err)
 	}
+
+	serviceImportController := &mcs.ServiceImportController{
+		Client:        mgr.GetClient(),
+		EventRecorder: mgr.GetEventRecorderFor(mcs.ServiceImportControllerName),
+	}
+	if err := serviceImportController.SetupWithManager(mgr); err != nil {
+		klog.Fatalf("Failed to setup ServiceImport controller: %v", err)
+	}
 }

pkg/controllers/mcs/service_import_controller.go

@@ -0,0 +1,136 @@
+package mcs
+
+import (
+	"context"
+
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"k8s.io/client-go/tools/record"
+	"k8s.io/klog/v2"
+	controllerruntime "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	mcsv1alpha1 "sigs.k8s.io/mcs-api/pkg/apis/v1alpha1"
+
+	"github.com/karmada-io/karmada/pkg/util/names"
+)
+
+// ServiceImportControllerName is the controller name that will be used when reporting events.
+const ServiceImportControllerName = "service-import-controller"
+
+// ServiceImportController is to sync derived service from ServiceImport.
+type ServiceImportController struct {
+	client.Client
+	EventRecorder record.EventRecorder
+}
+
+// Reconcile performs a full reconciliation for the object referred to by the Request.
+func (c *ServiceImportController) Reconcile(ctx context.Context, req controllerruntime.Request) (controllerruntime.Result, error) {
+	klog.V(4).Infof("Reconciling ServiceImport %s.", req.NamespacedName.String())
+
+	svcImport := &mcsv1alpha1.ServiceImport{}
+	if err := c.Client.Get(context.TODO(), req.NamespacedName, svcImport); err != nil {
+		if errors.IsNotFound(err) {
+			return controllerruntime.Result{}, nil
+		}
+
+		return controllerruntime.Result{Requeue: true}, err
+	}
+
+	if !svcImport.DeletionTimestamp.IsZero() {
+		return controllerruntime.Result{}, nil
+	}
+
+	return c.deriveServiceFromServiceImport(svcImport)
+}
+
+// SetupWithManager creates a controller and register to controller manager.
+func (c *ServiceImportController) SetupWithManager(mgr controllerruntime.Manager) error {
+	return controllerruntime.NewControllerManagedBy(mgr).For(&mcsv1alpha1.ServiceImport{}).Complete(c)
+}
+
+func (c *ServiceImportController) deriveServiceFromServiceImport(svcImport *mcsv1alpha1.ServiceImport) (controllerruntime.Result, error) {
+	newDerivedService := &corev1.Service{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: svcImport.Namespace,
+			Name:      names.GenerateDerivedServiceName(svcImport.Name),
+			OwnerReferences: []metav1.OwnerReference{
+				*metav1.NewControllerRef(svcImport, svcImport.GroupVersionKind()),
+			},
+		},
+		Spec: corev1.ServiceSpec{
+			Type:  corev1.ServiceTypeClusterIP,
+			Ports: servicePorts(svcImport),
+		},
+	}
+
+	oldDerivedService := &corev1.Service{}
+	err := c.Client.Get(context.TODO(), types.NamespacedName{
+		Name:      names.GenerateDerivedServiceName(svcImport.Name),
+		Namespace: svcImport.Namespace,
+	}, oldDerivedService)
+	if err != nil {
+		if errors.IsNotFound(err) {
+			if err = c.Client.Create(context.TODO(), newDerivedService); err != nil {
+				klog.Errorf("Create derived service(%s/%s) failed, Error: %v", newDerivedService.Namespace, newDerivedService.Name, err)
+				return controllerruntime.Result{Requeue: true}, err
+			}
+
+			return controllerruntime.Result{}, nil
+		}
+
+		return controllerruntime.Result{Requeue: true}, err
+	}
+
+	// retain necessary fields with old service
+	retainServiceFields(oldDerivedService, newDerivedService)
+
+	err = c.Client.Update(context.TODO(), newDerivedService)
+	if err != nil {
+		klog.Errorf("Update derived service(%s/%s) failed, Error: %v", newDerivedService.Namespace, newDerivedService.Name, err)
+		return controllerruntime.Result{Requeue: true}, err
+	}
+
+	return c.updateServiceStatus(svcImport, newDerivedService)
+}
+
+// updateServiceStatus update loadbalanacer status with provided clustersetIPs
+func (c *ServiceImportController) updateServiceStatus(svcImport *mcsv1alpha1.ServiceImport, derivedService *corev1.Service) (controllerruntime.Result, error) {
+	ingress := make([]corev1.LoadBalancerIngress, 0)
+	for _, ip := range svcImport.Spec.IPs {
+		ingress = append(ingress, corev1.LoadBalancerIngress{
+			IP: ip,
+		})
+	}
+	derivedService.Status = corev1.ServiceStatus{
+		LoadBalancer: corev1.LoadBalancerStatus{
+			Ingress: ingress,
+		},
+	}
+
+	if err := c.Client.Status().Update(context.TODO(), derivedService); err != nil {
+		klog.Errorf("Update derived service(%s/%s) status failed, Error: %v", derivedService.Namespace, derivedService.Name, err)
+		return controllerruntime.Result{Requeue: true}, err
+	}
+
+	return controllerruntime.Result{}, nil
+}
+
+func servicePorts(svcImport *mcsv1alpha1.ServiceImport) []corev1.ServicePort {
+	ports := make([]corev1.ServicePort, len(svcImport.Spec.Ports))
+	for i, p := range svcImport.Spec.Ports {
+		ports[i] = corev1.ServicePort{
+			Name:        p.Name,
+			Protocol:    p.Protocol,
+			Port:        p.Port,
+			AppProtocol: p.AppProtocol,
+		}
+	}
+	return ports
+}
+
+func retainServiceFields(oldSvc, newSvc *corev1.Service) {
+	newSvc.Spec.ClusterIP = oldSvc.Spec.ClusterIP
+	newSvc.ResourceVersion = oldSvc.ResourceVersion
+}