feat: gateway light support multiple backends

Signed-off-by: nabil salah <nabil.salah203@gmail.com>

Author: Nabil-Salah

pkg/gateway/gateway.go

@@ -558,10 +558,6 @@ func (g *gatewayModule) SetNamedProxy(wlID string, config zos.GatewayNameProxy)
 	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute)
 	defer cancel()
 
-	// if len(config.Backends) != 1 {
-	// 	return "", fmt.Errorf("only one backend is supported got '%d'", len(config.Backends))
-	// }
-
 	twinID, _, _, err := gridtypes.WorkloadID(wlID).Parts()
 	if err != nil {
 		return "", errors.Wrap(err, "invalid workload id")
@@ -600,10 +596,6 @@ func (g *gatewayModule) SetFQDNProxy(wlID string, config zos.GatewayFQDNProxy) e
 	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute)
 	defer cancel()
 
-	if len(config.Backends) != 1 {
-		return fmt.Errorf("only one backend is supported got '%d'", len(config.Backends))
-	}
-
 	cfg, err := g.ensureGateway(ctx, false)
 	if err != nil {
 		return err
@@ -661,21 +653,21 @@ func (g *gatewayModule) setupRouting(ctx context.Context, wlID string, fqdn stri
 		return errors.Wrap(err, "failed to get user network")
 	}
 	ns := net.Namespace(ctx, netID)
-	
+
 	processedBackends := make([]zos.Backend, 0, len(config.Backends))
-    for _, backend := range config.Backends {
+	for _, backend := range config.Backends {
 		processedBackend, err := g.nncEnsure(wlID, ns, backend)
-        if err != nil {
-            return errors.Wrap(err, "failed to ensure local gateway")
-        }
+		if err != nil {
+			return errors.Wrap(err, "failed to ensure local gateway")
+		}
+
+		if !config.TLSPassthrough {
+			// Format for non-TLS passthrough
+			processedBackend = zos.Backend(fmt.Sprintf("http://%s", processedBackend))
+		}
+
+		processedBackends = append(processedBackends, processedBackend)
 
-        if !config.TLSPassthrough {
-            // Format for non-TLS passthrough
-            processedBackend = zos.Backend(fmt.Sprintf("http://%s", processedBackend))
-        }
-        
-        processedBackends = append(processedBackends, processedBackend)
-    
 	}
 
 	config.Backends = processedBackends
@@ -696,12 +688,12 @@ func (g *gatewayModule) setupRoutingGeneric(wlID string, fqdn string, tlsConfig
 	servers := make([]Server, 0, len(config.Backends))
 	for _, backend := range config.Backends {
 		var server Server
-        if config.TLSPassthrough {
-            server = Server{Address: string(backend)}
-        } else {
-            server = Server{Url: string(backend)}
-        }
-        servers = append(servers, server)
+		if config.TLSPassthrough {
+			server = Server{Address: string(backend)}
+		} else {
+			server = Server{Url: string(backend)}
+		}
+		servers = append(servers, server)
 	}
 
 	route := fmt.Sprintf("%s-route", wlID)

pkg/gateway_light/gateway.go

@@ -569,10 +569,6 @@ func (g *gatewayModule) SetNamedProxy(wlID string, config zos.GatewayNameProxy)
 	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute)
 	defer cancel()
 
-	if len(config.Backends) != 1 {
-		return "", fmt.Errorf("only one backend is supported got '%d'", len(config.Backends))
-	}
-
 	twinID, _, _, err := gridtypes.WorkloadID(wlID).Parts()
 	if err != nil {
 		return "", errors.Wrap(err, "invalid workload id")
@@ -611,10 +607,6 @@ func (g *gatewayModule) SetFQDNProxy(wlID string, config zos.GatewayFQDNProxy) e
 	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute)
 	defer cancel()
 
-	if len(config.Backends) != 1 {
-		return fmt.Errorf("only one backend is supported got '%d'", len(config.Backends))
-	}
-
 	domain, err := g.ensureGateway(ctx, false)
 	if err != nil {
 		return err
@@ -642,10 +634,10 @@ func (g *gatewayModule) setupRouting(ctx context.Context, wlID string, fqdn stri
 	g.domainLock.Lock()
 	defer g.domainLock.Unlock()
 
-	backend := config.Backends[0]
-
-	if err := backend.Valid(config.TLSPassthrough); err != nil {
-		return errors.Wrapf(err, "failed to validate backend '%s'", backend)
+	for _, backend := range config.Backends {
+		if err := backend.Valid(config.TLSPassthrough); err != nil {
+			return errors.Wrapf(err, "failed to validate backend '%s'", backend)
+		}
 	}
 
 	if _, ok := g.getReservedDomain(fqdn); ok {
@@ -689,7 +681,6 @@ func (g *gatewayModule) setupRouting(ctx context.Context, wlID string, fqdn stri
 }
 
 func (g *gatewayModule) setupRoutingGeneric(wlID string, fqdn string, tlsConfig TlsConfig, config zos.GatewayBase) error {
-	backend := config.Backends[0]
 	var rule string
 	if config.TLSPassthrough {
 		rule = fmt.Sprintf("HostSNI(`%s`)", fqdn)
@@ -700,11 +691,15 @@ func (g *gatewayModule) setupRoutingGeneric(wlID string, fqdn string, tlsConfig
 		rule = fmt.Sprintf("Host(`%s`)", fqdn)
 	}
 
-	var server Server
-	if config.TLSPassthrough {
-		server = Server{Address: string(backend)}
-	} else {
-		server = Server{Url: string(backend)}
+	var servers []Server
+	for _, backend := range config.Backends {
+		var server Server
+		if config.TLSPassthrough {
+			server = Server{Address: string(backend)}
+		} else {
+			server = Server{Url: string(backend)}
+		}
+		servers = append(servers, server)
 	}
 
 	route := fmt.Sprintf("%s-route", wlID)
@@ -721,7 +716,7 @@ func (g *gatewayModule) setupRoutingGeneric(wlID string, fqdn string, tlsConfig
 		Services: map[string]Service{
 			wlID: {
 				LoadBalancer: LoadBalancer{
-					Servers: []Server{server},
+					Servers: servers,
 				},
 			},
 		},

pkg/gridtypes/zos/gw.go

@@ -105,10 +105,6 @@ func (g GatewayBase) Valid(getter gridtypes.WorkloadGetter) error {
 		return fmt.Errorf("backends list can not be empty")
 	}
 
-	// if len(g.Backends) != 1 {
-	// 	return fmt.Errorf("only one backend is supported")
-	// }
-
 	for _, backend := range g.Backends {
 		if err := backend.Valid(g.TLSPassthrough); err != nil {
 			return errors.Wrapf(err, "failed to validate backend '%s'", backend)