Adding ProvisionedConcurrency Support for Alias

Author: Vandita2020

apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,13 +1,13 @@
 ack_generate_info:
-  build_date: "2023-07-17T23:54:05Z"
+  build_date: "2023-07-24T17:09:21Z"
   build_hash: e9b68590da73ce9143ba1e4361cebdc1d876c81e
-  go_version: go1.20.5
+  go_version: go1.19
   version: v0.26.1-7-ge9b6859
-api_directory_checksum: 5d5c7aea8863c47e7303cc870aad4250267d93d2
+api_directory_checksum: 8f80588f678cbcff979aa67cf82e547315fdb2b8
 api_version: v1alpha1
 aws_sdk_go_version: v1.44.181
 generator_config_info:
-  file_checksum: a3a0fb7f87067244de99c0b37aaaf917b026f03d
+  file_checksum: 8b0c7bd625c74c9a471db45bbf8697069d35b218
   original_file_name: generator.yaml
 last_modification:
   reason: API generation

apis/v1alpha1/alias.go

@@ -108,6 +108,10 @@ resources:
         from:
           operation: PutFunctionEventInvokeConfig
           path: .
+      ProvisionedConcurrencyConfig:
+        from:
+          operation: PutProvisionedConcurrencyConfig
+          path: .
     hooks:
       sdk_update_pre_build_request:
         template_path: hooks/alias/sdk_update_pre_build_request.go.tpl

apis/v1alpha1/generator.yaml

@@ -104,6 +104,19 @@ spec:
               name:
                 description: The name of the alias.
                 type: string
+              provisionedConcurrencyConfig:
+                description: "Configures provisioned concurrency to a function's alias
+                  \n - ProvisionedConcurrentExecutions The amount of provisioned concurrency
+                  to allocate for the version or alias. Minimum value of 1 is required"
+                properties:
+                  functionName:
+                    type: string
+                  provisionedConcurrentExecutions:
+                    format: int64
+                    type: integer
+                  qualifier:
+                    type: string
+                type: object
               routingConfig:
                 description: The routing configuration (https://docs.aws.amazon.com/lambda/latest/dg/configuration-aliases.html#configuring-alias-routing)
                   of the alias.

apis/v1alpha1/types.go

@@ -39,3 +39,11 @@ resources:
           
           - MaximumRetryAttempts
               The maximum number of times to retry when the function returns an error.
+      
+      ProvisionedConcurrencyConfig:
+        prepend: |
+          Configures provisioned concurrency to a function's alias
+
+          - ProvisionedConcurrentExecutions
+              The amount of provisioned concurrency to allocate for the version or alias.
+              Minimum value of 1 is required

apis/v1alpha1/zz_generated.deepcopy.go

@@ -108,6 +108,10 @@ resources:
         from:
           operation: PutFunctionEventInvokeConfig
           path: .
+      ProvisionedConcurrencyConfig:
+        from:
+          operation: PutProvisionedConcurrencyConfig
+          path: .
     hooks:
       sdk_update_pre_build_request:
         template_path: hooks/alias/sdk_update_pre_build_request.go.tpl

config/crd/bases/lambda.services.k8s.aws_aliases.yaml

@@ -104,6 +104,19 @@ spec:
               name:
                 description: The name of the alias.
                 type: string
+              provisionedConcurrencyConfig:
+                description: "Configures provisioned concurrency to a function's alias
+                  \n - ProvisionedConcurrentExecutions The amount of provisioned concurrency
+                  to allocate for the version or alias. Minimum value of 1 is required"
+                properties:
+                  functionName:
+                    type: string
+                  provisionedConcurrentExecutions:
+                    format: int64
+                    type: integer
+                  qualifier:
+                    type: string
+                type: object
               routingConfig:
                 description: The routing configuration (https://docs.aws.amazon.com/lambda/latest/dg/configuration-aliases.html#configuring-alias-routing)
                   of the alias.

documentation.yaml

@@ -16,6 +16,7 @@ package alias
 import (
 	"context"
 
+	ackerr "github.com/aws-controllers-k8s/runtime/pkg/errors"
 	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
 	"github.com/aws/aws-sdk-go/aws"
 	svcsdk "github.com/aws/aws-sdk-go/service/lambda"
@@ -70,14 +71,71 @@ func (rm *resourceManager) syncEventInvokeConfig(
 	return r, nil
 }
 
-func (rm *resourceManager) setResourceAdditionalFields(
+func (rm *resourceManager) updateProvisionedConcurrency(
 	ctx context.Context,
-	ko *svcapitypes.Alias,
-) (err error) {
+	desired *resource,
+) error {
+	var err error
 	rlog := ackrtlog.FromContext(ctx)
-	exit := rlog.Trace("rm.setResourceAdditionalFields")
+	exit := rlog.Trace("rm.updateProvisionedConcurrency")
 	defer exit(err)
 
+	dspec := desired.ko.Spec
+	input := &svcsdk.PutProvisionedConcurrencyConfigInput{
+		FunctionName: aws.String(*dspec.FunctionName),
+		Qualifier:    aws.String(*dspec.Name),
+	}
+
+	if desired.ko.Spec.ProvisionedConcurrencyConfig != nil {
+		if desired.ko.Spec.ProvisionedConcurrencyConfig.ProvisionedConcurrentExecutions != nil {
+			input.ProvisionedConcurrentExecutions = aws.Int64(*desired.ko.Spec.ProvisionedConcurrencyConfig.ProvisionedConcurrentExecutions)
+		} else {
+			input.ProvisionedConcurrentExecutions = aws.Int64(0)
+		}
+	} else {
+		input.ProvisionedConcurrentExecutions = aws.Int64(0)
+	}
+
+	_, err = rm.sdkapi.PutProvisionedConcurrencyConfigWithContext(ctx, input)
+	rm.metrics.RecordAPICall("UPDATE", "UpdateProvisionedConcurrency", err)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (rm *resourceManager) getProvisionedConcurrencyConfig(
+	ctx context.Context,
+	ko *svcapitypes.Alias,
+) (err error) {
+
+	var getProvisionedConcurrencyConfigOutput *svcsdk.GetProvisionedConcurrencyConfigOutput
+	getProvisionedConcurrencyConfigOutput, err = rm.sdkapi.GetProvisionedConcurrencyConfigWithContext(
+		ctx,
+		&svcsdk.GetProvisionedConcurrencyConfigInput{
+			FunctionName: ko.Spec.FunctionName,
+			Qualifier:    ko.Spec.Name,
+		},
+	)
+	rm.metrics.RecordAPICall("GET", "GetProvisionedConcurrencyConfig", err)
+
+	if err != nil {
+		if awserr, ok := ackerr.AWSError(err); ok && (awserr.Code() == "ProvisionedConcurrencyConfigNotFoundException" || awserr.Code() == "ResourceNotFoundException") {
+			ko.Spec.ProvisionedConcurrencyConfig = nil
+		} else {
+			return err
+		}
+	} else {
+		ko.Spec.ProvisionedConcurrencyConfig.ProvisionedConcurrentExecutions = getProvisionedConcurrencyConfigOutput.RequestedProvisionedConcurrentExecutions
+	}
+
+	return nil
+}
+
+func (rm *resourceManager) getFunctionEventInvokeConfig(
+	ctx context.Context,
+	ko *svcapitypes.Alias,
+) (err error) {
 	var getFunctionEventInvokeConfigOutput *svcsdk.GetFunctionEventInvokeConfigOutput
 	getFunctionEventInvokeConfigOutput, err = rm.sdkapi.GetFunctionEventInvokeConfigWithContext(
 		ctx,
@@ -88,8 +146,13 @@ func (rm *resourceManager) setResourceAdditionalFields(
 	)
 
 	rm.metrics.RecordAPICall("GET", "GetFunctionEventInvokeConfig", err)
+
 	if err != nil {
-		ko.Spec.FunctionEventInvokeConfig = nil
+		if awserr, ok := ackerr.AWSError(err); ok && (awserr.Code() == "EventInvokeConfigNotFoundException" || awserr.Code() == "ResourceNotFoundException") {
+			ko.Spec.FunctionEventInvokeConfig = nil
+		} else {
+			return err
+		}
 	} else {
 		if getFunctionEventInvokeConfigOutput.DestinationConfig != nil {
 			if getFunctionEventInvokeConfigOutput.DestinationConfig.OnFailure != nil {
@@ -116,5 +179,27 @@ func (rm *resourceManager) setResourceAdditionalFields(
 			ko.Spec.FunctionEventInvokeConfig.MaximumRetryAttempts = nil
 		}
 	}
+
+	return nil
+}
+
+func (rm *resourceManager) setResourceAdditionalFields(
+	ctx context.Context,
+	ko *svcapitypes.Alias,
+) (err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.setResourceAdditionalFields")
+	defer exit(err)
+
+	eic_err := rm.getFunctionEventInvokeConfig(ctx, ko)
+	if eic_err != nil {
+		return eic_err
+	}
+
+	pc_err := rm.getProvisionedConcurrencyConfig(ctx, ko)
+	if pc_err != nil {
+		return pc_err
+	}
+
 	return nil
 }

generator.yaml

@@ -623,7 +623,11 @@ func (rm *resourceManager) setResourceAdditionalFields(
 	)
 	rm.metrics.RecordAPICall("GET", "GetFunctionEventInvokeConfig", err)
 	if err != nil {
-		ko.Spec.FunctionEventInvokeConfig = nil
+		if awserr, ok := ackerr.AWSError(err); ok && (awserr.Code() == "EventInvokeConfigNotFoundException" || awserr.Code() == "ResourceNotFoundException") {
+			ko.Spec.FunctionEventInvokeConfig = nil
+		} else {
+			return err
+		}
 	} else {
 		if getFunctionEventInvokeConfigOutput.DestinationConfig != nil {
 			if getFunctionEventInvokeConfigOutput.DestinationConfig.OnFailure != nil {