Adding ProvisionedConcurrency Support for Alias (#97)

Issue #, if available:

**Summary**
Adding support to configure `ProvisionedConcurrency` for alias in ACK.

**Description**
This PR adds support to configure `ProvisionedConcurrency` for Lambda function's alias using ACK. To implement this functionality I added `ProvisionedConcurrencyConfig` as an inline property to alias resource. Thus the user can directly set configurations for Provisioned Concurrency while creating alias. The user can edit the following code to set Provisioned Concurrency along with alias.

```
apiVersion: lambda.services.k8s.aws/v1alpha1
kind: Alias
metadata:
  name: $ALIAS_NAME
  annotations:
    services.k8s.aws/region: $AWS_REGION
spec:
  name: $ALIAS_NAME
  functionName: $FUNCTION_NAME
  functionVersion: $FUNCTION_VERSION
  description: alias created by ACK lambda-controller e2e tests
  provisionedConcurrencyConfig:
    provisionedConcurrentExecutions: $PROVISIONED_CONCURRENT_EXECUTIONS
```
This PR includes both implementation code and e2e tests.

**Acknowledgement**
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

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-24T18:19:28Z"
   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 {