add gateway public service controller

pkg/yurtmanager/controller/controller.go

@@ -30,6 +30,7 @@ import (
 	"github.com/openyurtio/openyurt/pkg/yurtmanager/controller/raven/dns"
 	"github.com/openyurtio/openyurt/pkg/yurtmanager/controller/raven/gatewayinternalservice"
 	"github.com/openyurtio/openyurt/pkg/yurtmanager/controller/raven/gatewaypickup"
+	"github.com/openyurtio/openyurt/pkg/yurtmanager/controller/raven/gatewaypublicservice"
 	"github.com/openyurtio/openyurt/pkg/yurtmanager/controller/servicetopology"
 	servicetopologyendpoints "github.com/openyurtio/openyurt/pkg/yurtmanager/controller/servicetopology/endpoints"
 	servicetopologyendpointslice "github.com/openyurtio/openyurt/pkg/yurtmanager/controller/servicetopology/endpointslice"
@@ -68,6 +69,7 @@ func init() {
 	controllerAddFuncs[raven.GatewayPickupControllerName] = []AddControllerFn{gatewaypickup.Add}
 	controllerAddFuncs[raven.GatewayDNSControllerName] = []AddControllerFn{dns.Add}
 	controllerAddFuncs[raven.GatewayInternalServiceController] = []AddControllerFn{gatewayinternalservice.Add}
+	controllerAddFuncs[raven.GatewayPublicServiceController] = []AddControllerFn{gatewaypublicservice.Add}
 }
 
 // If you want to add additional RBAC, enter it here !!! @kadisi

pkg/yurtmanager/controller/raven/common.go

@@ -24,5 +24,6 @@ const (
 	ControllerName                   = "gateway"
 	GatewayPickupControllerName      = "raven-gateway-pickup"
 	GatewayInternalServiceController = "raven-gateway-internal-service"
+	GatewayPublicServiceController   = "raven-gateway-public-service"
 	GatewayDNSControllerName         = "raven-dns"
 )

pkg/yurtmanager/controller/raven/gatewayinternalservice/gateway_internal_service_enqueue_handlers.go

@@ -30,10 +30,18 @@ import (
 
 type EnqueueRequestForGatewayEvent struct{}
 
+func logGatewayError() {
+	klog.Error(Format("fail to assert runtime Object to v1beta1.Gateway"))
+}
+
+func logConfigmapError() {
+	klog.Error(Format("fail to assert runtime Object to v1.Configmap"))
+}
+
 func (h *EnqueueRequestForGatewayEvent) Create(e event.CreateEvent, q workqueue.RateLimitingInterface) {
 	gw, ok := e.Object.(*ravenv1beta1.Gateway)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to v1alpha1.Gateway"))
+		logGatewayError()
 		return
 	}
 	if gw.Spec.ExposeType == "" {
@@ -46,12 +54,12 @@ func (h *EnqueueRequestForGatewayEvent) Create(e event.CreateEvent, q workqueue.
 func (h *EnqueueRequestForGatewayEvent) Update(e event.UpdateEvent, q workqueue.RateLimitingInterface) {
 	newGw, ok := e.ObjectNew.(*ravenv1beta1.Gateway)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to v1alpha1.Gateway"))
+		logGatewayError()
 		return
 	}
 	oldGw, ok := e.ObjectOld.(*ravenv1beta1.Gateway)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to v1alpha1.Gateway"))
+		logGatewayError()
 		return
 	}
 	if oldGw.Spec.ExposeType == "" && newGw.Spec.ExposeType == "" {
@@ -64,7 +72,7 @@ func (h *EnqueueRequestForGatewayEvent) Update(e event.UpdateEvent, q workqueue.
 func (h *EnqueueRequestForGatewayEvent) Delete(e event.DeleteEvent, q workqueue.RateLimitingInterface) {
 	gw, ok := e.Object.(*ravenv1beta1.Gateway)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to v1alpha1.Gateway"))
+		logGatewayError()
 		return
 	}
 	if gw.Spec.ExposeType == "" {
@@ -83,7 +91,7 @@ type EnqueueRequestForConfigEvent struct{}
 func (h *EnqueueRequestForConfigEvent) Create(e event.CreateEvent, q workqueue.RateLimitingInterface) {
 	cm, ok := e.Object.(*corev1.ConfigMap)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to corev1.Configmap"))
+		logConfigmapError()
 		return
 	}
 	if cm.Data == nil {
@@ -108,12 +116,12 @@ func (h *EnqueueRequestForConfigEvent) Create(e event.CreateEvent, q workqueue.R
 func (h *EnqueueRequestForConfigEvent) Update(e event.UpdateEvent, q workqueue.RateLimitingInterface) {
 	newCm, ok := e.ObjectNew.(*corev1.ConfigMap)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to corev1.Configmap"))
+		logConfigmapError()
 		return
 	}
 	oldCm, ok := e.ObjectOld.(*corev1.ConfigMap)
 	if !ok {
-		klog.Error(Format("fail to assert runtime Object to corev1.Configmap"))
+		logConfigmapError()
 		return
 	}
 	_, newInsecurePort, newErr := net.SplitHostPort(newCm.Data[utils.ProxyServerInsecurePortKey])

pkg/yurtmanager/controller/raven/gatewaypickup/gateway_pickup_controller.go

@@ -21,6 +21,7 @@ import (
 	"fmt"
 	"reflect"
 	"sort"
+	"strconv"
 	"strings"
 	"time"
 
@@ -180,6 +181,7 @@ func (r *ReconcileGateway) Reconcile(ctx context.Context, req reconcile.Request)
 	activeEp := r.electActiveEndpoint(nodeList, &gw)
 	r.recordEndpointEvent(&gw, gw.Status.ActiveEndpoints, activeEp)
 	gw.Status.ActiveEndpoints = activeEp
+	r.configEndpoints(ctx, &gw)
 	// 2. get nodeInfo list of nodes managed by the Gateway
 	var nodes []ravenv1beta1.NodeInfo
 	for _, v := range nodeList.Items {
@@ -363,3 +365,20 @@ func getActiveEndpointsInfo(eps []*ravenv1beta1.Endpoint) (map[string][]string,
 	}
 	return infos, len(infos[ActiveEndpointsName])
 }
+
+func (r *ReconcileGateway) configEndpoints(ctx context.Context, gw *ravenv1beta1.Gateway) {
+	enableProxy, enableTunnel := utils.CheckServer(ctx, r.Client)
+	for idx, val := range gw.Status.ActiveEndpoints {
+		if gw.Status.ActiveEndpoints[idx].Config == nil {
+			gw.Status.ActiveEndpoints[idx].Config = make(map[string]string)
+		}
+		switch val.Type {
+		case ravenv1beta1.Proxy:
+			gw.Status.ActiveEndpoints[idx].Config[utils.RavenEnableProxy] = strconv.FormatBool(enableProxy)
+		case ravenv1beta1.Tunnel:
+			gw.Status.ActiveEndpoints[idx].Config[utils.RavenEnableTunnel] = strconv.FormatBool(enableTunnel)
+		default:
+		}
+	}
+	return
+}