Tags support for elbv2-controller resources

generator.yaml

@@ -125,6 +125,12 @@ resources:
         code: customPreCompare(delta, a, b)
       sdk_read_many_post_set_output:
         template_path: hooks/load_balancer/sdk_read_many_post_set_output.go.tpl
+      sdk_create_post_set_output:
+        template_path: hooks/load_balancer/sdk_create_post_set_output.go.tpl
+      sdk_update_pre_build_request:
+        template_path: hooks/load_balancer/sdk_update_pre_build_request.go.tpl
+      sdk_read_many_post_build_request:
+        template_path: hooks/load_balancer/sdk_update_post_build_request.go.tpl
   Listener:
     fields:
       DefaultActions.TargetGroupARN:
@@ -162,6 +168,10 @@ resources:
     hooks:
       sdk_read_many_post_build_request:
         template_path: hooks/listener/sdk_read_many_post_build_request.go.tpl
+      sdk_create_post_set_output:
+        template_path: hooks/listener/sdk_create_post_set_output.go.tpl
+      sdk_update_pre_build_request:
+        template_path: hooks/target_group/sdk_update_pre_build_request.go.tpl
   TargetGroup:
     fields:
       Name:
@@ -234,3 +244,5 @@ resources:
         template_path: hooks/rule/sdk_update_pre_build_request.go.tpl
       sdk_read_many_post_set_output:
         template_path: hooks/rule/sdk_read_many_post_set_output.go.tpl
+      sdk_create_post_build_request:
+        template_path: hooks/rule/sdk_create_post_build_request.go.tpl

pkg/resource/listener/hooks.go

@@ -1,5 +1,18 @@
 package listener
 
+import (
+	"context"
+	"time"
+
+	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
+	"github.com/aws-controllers-k8s/elbv2-controller/pkg/resource/tags"
+	svcapitypes "github.com/aws-controllers-k8s/elbv2-controller/apis/v1alpha1"
+)
+
+var (
+	RequeueAfterUpdateDuration = 5 * time.Second
+)
+
 // customCheckRequiredFieldsMissingMethod returns true if there are any fields
 // for the ReadOne Input shape that are required but not present in the
 // resource's Spec or Status.
@@ -8,3 +21,21 @@ func (rm *resourceManager) customCheckRequiredFieldsMissingMethod(
 ) bool {
 	return r.Identifiers().ARN() == nil
 }
+
+func (rm *resourceManager) getTags(
+	ctx context.Context,
+	resourceARN  string,
+) ([]*svcapitypes.Tag, error) {
+	return tags.GetResourceTags(ctx, rm.sdkapi, rm.metrics, resourceARN )
+}
+
+func (rm *resourceManager) updateTags(
+	ctx context.Context,
+	desired *resource,
+	latest *resource,
+) (err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.describeTargets")
+	defer func() { exit(err) }()
+	return tags.SyncRecourseTags(ctx, rm.sdkapi, rm.metrics, string(*desired.ko.Status.ACKResourceMetadata.ARN), desired.ko.Spec.Tags, latest.ko.Spec.Tags)
+}

pkg/resource/load_balancer/hooks.go

@@ -15,13 +15,19 @@ package load_balancer
 
 import (
 	"context"
+	"time"
 
 	svcapitypes "github.com/aws-controllers-k8s/elbv2-controller/apis/v1alpha1"
 	ackcompare "github.com/aws-controllers-k8s/runtime/pkg/compare"
 	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
 	"github.com/aws/aws-sdk-go-v2/aws"
 	svcsdk "github.com/aws/aws-sdk-go-v2/service/elasticloadbalancingv2"
 	svcsdktypes "github.com/aws/aws-sdk-go-v2/service/elasticloadbalancingv2/types"
+	"github.com/aws-controllers-k8s/elbv2-controller/pkg/resource/tags"
+)
+
+var(
+	RequeueAfterUpdateDuration = 5 * time.Second
 )
 
 // setResourceAdditionalFields will describe the fields that are not return by the
@@ -126,7 +132,11 @@ func (rm *resourceManager) customUpdateLoadBalancer(
 		}
 	}
 	// Leaving room for tag updates...
-	// if delta.DifferentAt("Spec.Tags") {...}
+	if delta.DifferentAt("Spec.Tags") {
+		if err := rm.updateLoadBalancerTags(ctx, desired, latest); err != nil {
+			return nil, err
+		}
+	}
 
 	return desired, nil
 }
@@ -169,3 +179,58 @@ func (rm *resourceManager) updateLoadBalancerAttributes(
 	}
 	return nil
 }
+
+func (rm *resourceManager) getTags(
+	ctx context.Context,
+	resourceARN  string,
+) ([]*svcapitypes.Tag, error) {
+	return tags.GetResourceTags(ctx, rm.sdkapi, rm.metrics, resourceARN )
+}
+
+func (rm *resourceManager) updateTags(
+	ctx context.Context,
+	desired *resource,
+	latest *resource,
+) (err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.describeTargets")
+	defer func() { exit(err) }()
+	return tags.SyncRecourseTags(ctx, rm.sdkapi, rm.metrics, string(*desired.ko.Status.ACKResourceMetadata.ARN), desired.ko.Spec.Tags, latest.ko.Spec.Tags)
+}
+
+// updateLoadBalancerTags updates the tags of the load balancer.
+func (rm *resourceManager) updateLoadBalancerTags(
+	ctx context.Context,
+	desired *resource,
+	latest *resource,
+) error {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.updateLoadBalancerTags")
+	var err error
+	defer func() { exit(err) }()
+
+	currentTags, err := rm.getTags(
+		ctx,
+		string(*desired.ko.Status.ACKResourceMetadata.ARN),
+	)
+	if err != nil {
+		return err
+	}
+
+	desiredTags := []*svcapitypes.Tag{}
+	for _, tag := range desired.ko.Spec.Tags {
+		desiredTags = append(desiredTags, &svcapitypes.Tag{
+			Key:   tag.Key,
+			Value: tag.Value,
+		})
+	}
+
+	return tags.SyncRecourseTags(
+		ctx,
+		rm.sdkapi,
+		rm.metrics,
+		string(*desired.ko.Status.ACKResourceMetadata.ARN),
+		currentTags,
+		desiredTags,
+	)
+}

pkg/resource/rule/hooks.go

@@ -17,16 +17,20 @@ import (
 	"context"
 	"errors"
 	"strconv"
+	"time"
 
 	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
 	"github.com/aws/aws-sdk-go-v2/aws"
 	svcsdk "github.com/aws/aws-sdk-go-v2/service/elasticloadbalancingv2"
 	svcsdktypes "github.com/aws/aws-sdk-go-v2/service/elasticloadbalancingv2/types"
+	svcapitypes "github.com/aws-controllers-k8s/elbv2-controller/apis/v1alpha1"
+	"github.com/aws-controllers-k8s/elbv2-controller/pkg/resource/tags"
 )
 
 var (
 	// ErrInvalidPriority is an error that is returned when the priority value is invalid.
 	ErrInvalidPriority = errors.New("invalid priority value")
+	RequeueAfterUpdateDuration = 5 * time.Second
 )
 
 // setRulePriority sets the priority of the rule.
@@ -81,3 +85,21 @@ func int32OrNil(val *int64) *int32 {
 	}
 	return nil
 }
+
+func (rm *resourceManager) getTags(
+	ctx context.Context,
+	resourceARN  string,
+) ([]*svcapitypes.Tag, error) {
+	return tags.GetResourceTags(ctx, rm.sdkapi, rm.metrics, resourceARN )
+}
+
+func (rm *resourceManager) updateTags(
+	ctx context.Context,
+	desired *resource,
+	latest *resource,
+) (err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.describeTargets")
+	defer func() { exit(err) }()
+	return tags.SyncRecourseTags(ctx, rm.sdkapi, rm.metrics, string(*desired.ko.Status.ACKResourceMetadata.ARN), desired.ko.Spec.Tags, latest.ko.Spec.Tags)
+}