Fix misuse in struct kobject.ServiceConfig.Port (#1423)

2021-08-29 21:37:22 +08:00
parent 0671410009
commit 2f081075a8
9 changed files with 80 additions and 92 deletions
@@ -156,20 +156,15 @@ func loadEnvVars(service Service) ([]kobject.EnvVar, error) {
 func loadPorts(service Service) ([]kobject.Ports, error) {
 	ports := []kobject.Ports{}
 	for _, port := range service.Ports {
-		var p api.Protocol
-		switch port.Protocol {
-		default:
-			p = api.ProtocolTCP
-		case "TCP":
-			p = api.ProtocolTCP
-		case "UDP":
-			p = api.ProtocolUDP
-		}
-		ports = append(ports, kobject.Ports{
+		komposePorts := kobject.Ports{
 			HostPort:      int32(port.Port),
 			ContainerPort: int32(port.Port),
-			Protocol:      p,
-		})
+			Protocol:      port.Protocol,
+		}
+		if protocol := api.Protocol(port.Protocol); protocol != api.ProtocolTCP && protocol != api.ProtocolUDP {
+			komposePorts.Protocol = string(api.ProtocolTCP)
+		}
+		ports = append(ports, komposePorts)
 	}
 	return ports, nil
 }