ECS: Add UI Logs for Listener Rules in TRAFFIC_ROUTING and ROLLBACK

Signed-off-by: “niladrix719” <[email protected]>

Author: niladrix719

pkg/app/piped/executor/ecs/ecs.go

@@ -467,10 +467,17 @@ func routing(ctx context.Context, in *executor.Input, platformProviderName strin
 		return false
 	}
 
-	if err := client.ModifyListeners(ctx, currListenerArns, routingTrafficCfg); err != nil {
+	modifiedRules, err := client.ModifyListeners(ctx, currListenerArns, routingTrafficCfg);
+	if err != nil {
 		in.LogPersister.Errorf("Failed to routing traffic to PRIMARY/CANARY variants: %v", err)
 		return false
 	}
+	
+	if len(modifiedRules) > 0 {
+		in.LogPersister.Infof("Successfully modified ELB listener rules: %s", strings.Join(modifiedRules, ", "))
+	} else {
+		in.LogPersister.Info("No ELB listener rules were modified")
+	}
 
 	return true
 }

pkg/app/piped/executor/ecs/rollback.go

@@ -17,6 +17,7 @@ package ecs
 import (
 	"context"
 	"errors"
+	"strings"
 
 	"github.com/aws/aws-sdk-go-v2/service/ecs/types"
 
@@ -158,10 +159,17 @@ func rollback(ctx context.Context, in *executor.Input, platformProviderName stri
 			return false
 		}
 
-		if err := client.ModifyListeners(ctx, currListenerArns, routingTrafficCfg); err != nil {
+		modifiedRules, err := client.ModifyListeners(ctx, currListenerArns, routingTrafficCfg);
+		if err != nil {
 			in.LogPersister.Errorf("Failed to routing traffic to PRIMARY/CANARY variants: %v", err)
 			return false
 		}
+		
+		if len(modifiedRules) > 0 {
+			in.LogPersister.Infof("Successfully modified ELB listener rules: %s", strings.Join(modifiedRules, ", "))
+		} else {
+			in.LogPersister.Info("No ELB listener rules were modified")
+		}
 	}
 
 	// Delete previous ACTIVE taskSets

pkg/app/piped/platformprovider/ecs/client.go

@@ -472,17 +472,19 @@ func (c *client) getLoadBalancerArn(ctx context.Context, targetGroupArn string)
 	return output.TargetGroups[0].LoadBalancerArns[0], nil
 }
 
-func (c *client) ModifyListeners(ctx context.Context, listenerArns []string, routingTrafficCfg RoutingTrafficConfig) error {
+func (c *client) ModifyListeners(ctx context.Context, listenerArns []string, routingTrafficCfg RoutingTrafficConfig) ([]string, error) {
 	if len(routingTrafficCfg) != 2 {
-		return fmt.Errorf("invalid listener configuration: requires 2 target groups")
+		return nil, fmt.Errorf("invalid listener configuration: requires 2 target groups")
 	}
 
+	modifiedRuleArns := make([]string, 0)
+
 	for _, listenerArn := range listenerArns {
 		describeRulesOutput, err := c.elbClient.DescribeRules(ctx, &elasticloadbalancingv2.DescribeRulesInput{
 			ListenerArn: aws.String(listenerArn),
 		})
 		if err != nil {
-			return fmt.Errorf("failed to describe rules of listener %s: %w", listenerArn, err)
+			return nil, fmt.Errorf("failed to describe rules of listener %s: %w", listenerArn, err)
 		}
 
 		for _, rule := range describeRulesOutput.Rules {
@@ -519,20 +521,22 @@ func (c *client) ModifyListeners(ctx context.Context, listenerArns []string, rou
 					DefaultActions: modifiedActions,
 				})
 				if err != nil {
-					return fmt.Errorf("failed to modify default rule %s: %w", *rule.RuleArn, err)
-				}
+					return nil, fmt.Errorf("failed to modify default rule %s: %w", *rule.RuleArn, err)
+				} 
+				modifiedRuleArns = append(modifiedRuleArns, fmt.Sprintf("default rule of listener %s", listenerArn))
 			} else {
 				_, err := c.elbClient.ModifyRule(ctx, &elasticloadbalancingv2.ModifyRuleInput{
 					RuleArn: rule.RuleArn,
 					Actions: modifiedActions,
 				})
 				if err != nil {
-					return fmt.Errorf("failed to modify rule %s: %w", *rule.RuleArn, err)
+					return nil, fmt.Errorf("failed to modify rule %s: %w", *rule.RuleArn, err)
 				}
+				modifiedRuleArns = append(modifiedRuleArns, *rule.RuleArn)
 			}
 		}
 	}
-	return nil
+	return modifiedRuleArns, nil
 }
 
 func (c *client) TagResource(ctx context.Context, resourceArn string, tags []types.Tag) error {

pkg/app/piped/platformprovider/ecs/ecs.go

@@ -64,7 +64,7 @@ type ELB interface {
 	GetListenerArns(ctx context.Context, targetGroup types.LoadBalancer) ([]string, error)
 	// ModifyListeners modifies the actions of type ActionTypeEnumForward to perform routing traffic
 	// to the given target groups. Other actions won't be modified.
-	ModifyListeners(ctx context.Context, listenerArns []string, routingTrafficCfg RoutingTrafficConfig) error
+	ModifyListeners(ctx context.Context, listenerArns []string, routingTrafficCfg RoutingTrafficConfig) ([]string, error)
 }
 
 // Registry holds a pool of aws client wrappers.