From 19f57ab5020a6c8a4b7810d64181557bfc7f6990 Mon Sep 17 00:00:00 2001
From: ChrisLiu <chrisliu1995@163.com>
Date: Mon, 16 Dec 2024 15:03:06 +0800
Subject: [PATCH] support same port of tcp&udp

Signed-off-by: ChrisLiu <chrisliu1995@163.com>
---
 cloudprovider/alibabacloud/multi_nlbs.go | 29 +++++++++++++++++++-----
 1 file changed, 23 insertions(+), 6 deletions(-)

diff --git a/cloudprovider/alibabacloud/multi_nlbs.go b/cloudprovider/alibabacloud/multi_nlbs.go
index 38ffdb8..d0a007a 100644
--- a/cloudprovider/alibabacloud/multi_nlbs.go
+++ b/cloudprovider/alibabacloud/multi_nlbs.go
@@ -50,6 +50,8 @@ const (
 
 	// service label defined by OKG
 	ServiceBelongNetworkTypeKey = "game.kruise.io/network-type"
+
+	ProtocolTCPUDP corev1.Protocol = "TCPUDP"
 )
 
 type MultiNlbsPlugin struct {
@@ -352,12 +354,27 @@ func (m *MultiNlbsPlugin) consSvc(conf *multiNLBsConfig, pod *corev1.Pod, lbName
 
 	svcPorts := make([]corev1.ServicePort, 0)
 	for i := 0; i < len(podLbsPorts.ports); i++ {
-		svcPorts = append(svcPorts, corev1.ServicePort{
-			Name:       strconv.Itoa(podLbsPorts.targetPort[i]) + "-" + strings.ToLower(string(podLbsPorts.protocols[i])),
-			Port:       podLbsPorts.ports[i],
-			TargetPort: intstr.FromInt(podLbsPorts.targetPort[i]),
-			Protocol:   podLbsPorts.protocols[i],
-		})
+		if podLbsPorts.protocols[i] == ProtocolTCPUDP {
+			svcPorts = append(svcPorts, corev1.ServicePort{
+				Name:       strconv.Itoa(podLbsPorts.targetPort[i]) + "-" + strings.ToLower(string(corev1.ProtocolTCP)),
+				Port:       podLbsPorts.ports[i],
+				TargetPort: intstr.FromInt(podLbsPorts.targetPort[i]),
+				Protocol:   corev1.ProtocolTCP,
+			})
+			svcPorts = append(svcPorts, corev1.ServicePort{
+				Name:       strconv.Itoa(podLbsPorts.targetPort[i]) + "-" + strings.ToLower(string(corev1.ProtocolUDP)),
+				Port:       podLbsPorts.ports[i],
+				TargetPort: intstr.FromInt(podLbsPorts.targetPort[i]),
+				Protocol:   corev1.ProtocolUDP,
+			})
+		} else {
+			svcPorts = append(svcPorts, corev1.ServicePort{
+				Name:       strconv.Itoa(podLbsPorts.targetPort[i]) + "-" + strings.ToLower(string(podLbsPorts.protocols[i])),
+				Port:       podLbsPorts.ports[i],
+				TargetPort: intstr.FromInt(podLbsPorts.targetPort[i]),
+				Protocol:   podLbsPorts.protocols[i],
+			})
+		}
 	}
 
 	loadBalancerClass := "alibabacloud.com/nlb"