update handler to recieve status update for gateways

Author: salonichf5

cmd/gateway/commands_test.go

@@ -63,6 +63,8 @@ func TestCommonFlagsValidation(t *testing.T) {
 			args: []string{
 				"--gateway-ctlr-name=gateway.nginx.org/nginx-gateway",
 				"--gatewayclass=nginx",
+				"--gateway=nginx-gateway/nginx",
+				"--config=nginx-gateway-config",
 			},
 			wantErr: false,
 		},

internal/mode/static/handler.go

@@ -294,64 +294,89 @@ func (h *eventHandlerImpl) waitForStatusUpdates(ctx context.Context) {
 			return
 		}
 
-		// TODO(sberman): once we support multiple Gateways, we'll have to get
-		// the correct Graph for the Deployment contained in the update message
+		// get the deploymentName from the item
+		deploymentName := item.Deployment
+		if deploymentName == (types.NamespacedName{}) {
+			continue
+		}
+
+		fmt.Println("Processing status update for deployment:", deploymentName)
+
 		gr := h.cfg.processor.GetLatestGraph()
 		if gr == nil {
 			continue
 		}
 
+		gw := gr.Gateways[deploymentName]
+
+		// if gateway is nil,, we update the gateway class status
+		if gw == nil {
+			h.updateGatewayClassStatus(ctx, gr)
+			continue
+		}
+
 		var nginxReloadRes graph.NginxReloadResult
-		for _, gw := range gr.Gateways {
-			switch {
-			case item.Error != nil:
-				h.cfg.logger.Error(item.Error, "Failed to update NGINX configuration")
-				nginxReloadRes.Error = item.Error
-			case gw != nil:
-				h.cfg.logger.Info("NGINX configuration was successfully updated")
-			}
-			gr.LatestReloadResult = nginxReloadRes
-
-			switch item.UpdateType {
-			case status.UpdateAll:
-				h.updateStatuses(ctx, gr, gw)
-			case status.UpdateGateway:
-				gwAddresses, err := getGatewayAddresses(
-					ctx,
-					h.cfg.k8sClient,
+		switch {
+		case item.Error != nil:
+			h.cfg.logger.Error(item.Error, "Failed to update NGINX configuration")
+			nginxReloadRes.Error = item.Error
+		case gw != nil:
+			h.cfg.logger.Info("NGINX configuration was successfully updated")
+		}
+		gr.LatestReloadResult[deploymentName] = nginxReloadRes
+
+		switch item.UpdateType {
+		case status.UpdateAll:
+			h.updateStatuses(ctx, gr, gw)
+		case status.UpdateGateway:
+			fmt.Println("update gateways status", gw)
+			gwAddresses, err := getGatewayAddresses(
+				ctx,
+				h.cfg.k8sClient,
+				item.GatewayService,
+				gw,
+				h.cfg.gatewayClassName,
+			)
+			if err != nil {
+				msg := "error getting Gateway Service IP address"
+				h.cfg.logger.Error(err, msg)
+				h.cfg.eventRecorder.Eventf(
 					item.GatewayService,
-					gw,
-					h.cfg.gatewayClassName,
+					v1.EventTypeWarning,
+					"GetServiceIPFailed",
+					msg+": %s",
+					err.Error(),
 				)
-				if err != nil {
-					msg := "error getting Gateway Service IP address"
-					h.cfg.logger.Error(err, msg)
-					h.cfg.eventRecorder.Eventf(
-						item.GatewayService,
-						v1.EventTypeWarning,
-						"GetServiceIPFailed",
-						msg+": %s",
-						err.Error(),
-					)
-					continue
-				}
-
-				transitionTime := metav1.Now()
-
-				gatewayStatuses := status.PrepareGatewayRequests(
-					gw,
-					transitionTime,
-					gwAddresses,
-					gr.LatestReloadResult,
-				)
-				h.cfg.statusUpdater.UpdateGroup(ctx, groupGateways, gatewayStatuses...)
-			default:
-				panic(fmt.Sprintf("unknown event type %T", item.UpdateType))
+				continue
 			}
+
+			transitionTime := metav1.Now()
+
+			gatewayStatuses := status.PrepareGatewayRequests(
+				gw,
+				transitionTime,
+				gwAddresses,
+				gr.LatestReloadResult[deploymentName],
+			)
+			h.cfg.statusUpdater.UpdateGroup(ctx, groupGateways, gatewayStatuses...)
+		default:
+			panic(fmt.Sprintf("unknown event type %T", item.UpdateType))
 		}
 	}
 }
 
+func (h *eventHandlerImpl) updateGatewayClassStatus(ctx context.Context, gr *graph.Graph) {
+	transitionTime := metav1.Now()
+
+	gcReqs := status.PrepareGatewayClassRequests(gr.GatewayClass, gr.IgnoredGatewayClasses, transitionTime)
+
+	reqs := make([]frameworkStatus.UpdateRequest, 0, len(gcReqs))
+	reqs = append(reqs, gcReqs...)
+
+	fmt.Println("Processing status update for GatewayClass", gr.GatewayClass)
+	h.cfg.statusUpdater.UpdateGroup(ctx, groupAllExceptGateways, reqs...)
+}
+
 func (h *eventHandlerImpl) updateStatuses(ctx context.Context, gr *graph.Graph, gw *graph.Gateway) {
 	gwAddresses, err := getGatewayAddresses(ctx, h.cfg.k8sClient, nil, gw, h.cfg.gatewayClassName)
 	if err != nil {
@@ -373,7 +398,7 @@ func (h *eventHandlerImpl) updateStatuses(ctx context.Context, gr *graph.Graph,
 		gr.L4Routes,
 		gr.Routes,
 		transitionTime,
-		gr.LatestReloadResult,
+		gr.LatestReloadResult[gw.DeploymentName],
 		h.cfg.gatewayCtlrName,
 	)
 
@@ -404,7 +429,7 @@ func (h *eventHandlerImpl) updateStatuses(ctx context.Context, gr *graph.Graph,
 		gw,
 		transitionTime,
 		gwAddresses,
-		gr.LatestReloadResult,
+		gr.LatestReloadResult[gw.DeploymentName],
 	)
 	h.cfg.statusUpdater.UpdateGroup(ctx, groupGateways, gwReqs...)
 }

internal/mode/static/handler_test.go

@@ -381,7 +381,7 @@ var _ = Describe("eventHandler", func() {
 			}).Should(Equal(2))
 
 		gr := handler.cfg.processor.GetLatestGraph()
-		Expect(gr.LatestReloadResult.Error.Error()).To(Equal("status error"))
+		Expect(gr.LatestReloadResult[types.NamespacedName{}].Error.Error()).To(Equal("status error"))
 	})
 
 	It("should update Gateway status when receiving a queue event", func() {

internal/mode/static/state/graph/backend_refs.go

@@ -49,7 +49,7 @@ func addBackendRefsToRouteRules(
 	gws map[types.NamespacedName]*Gateway,
 ) {
 	for _, gw := range gws {
-		if gw == nil {
+		if gw == nil || gw == (&Gateway{}) {
 			continue
 		}
 		for _, r := range routes {