Adding ProvisionedConcurrency Support for Alias

Author: Vandita2020

apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,13 +1,13 @@
 ack_generate_info:
-  build_date: "2023-06-28T18:32:11Z"
+  build_date: "2023-07-12T16:21:57Z"
   build_hash: e9b68590da73ce9143ba1e4361cebdc1d876c81e
   go_version: go1.19
   version: v0.26.1-7-ge9b6859
-api_directory_checksum: 5f162746e8495943dae5e96f48f4a3ab887b5be5
+api_directory_checksum: e8290d4489213bc48136215f33af8cae03f4e2a9
 api_version: v1alpha1
 aws_sdk_go_version: v1.44.181
 generator_config_info:
-  file_checksum: ed4abfc994c2c47465801d301604c584bd743d41
+  file_checksum: 0982d74c1a222e29260dbf276136ba99fcb856e2
   original_file_name: generator.yaml
 last_modification:
   reason: API generation

apis/v1alpha1/alias.go

@@ -106,6 +106,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,16 @@ spec:
               name:
                 description: The name of the alias.
                 type: string
+              provisionedConcurrencyConfig:
+                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

@@ -106,6 +106,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/zz_generated.deepcopy.go

@@ -104,6 +104,16 @@ spec:
               name:
                 description: The name of the alias.
                 type: string
+              provisionedConcurrencyConfig:
+                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.

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

@@ -70,6 +70,37 @@ func (rm *resourceManager) syncEventInvokeConfig(
 	return r, nil
 }
 
+func (rm *resourceManager) updateProvisionedConcurrency(
+	ctx context.Context,
+	desired *resource,
+) error {
+	var err error
+	rlog := ackrtlog.FromContext(ctx)
+	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)
+		}
+	}
+
+	_, err = rm.sdkapi.PutProvisionedConcurrencyConfigWithContext(ctx, input)
+	rm.metrics.RecordAPICall("UPDATE", "UpdateProvisionedConcurrency", err)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 func (rm *resourceManager) setResourceAdditionalFields(
 	ctx context.Context,
 	ko *svcapitypes.Alias,
@@ -78,6 +109,21 @@ func (rm *resourceManager) setResourceAdditionalFields(
 	exit := rlog.Trace("rm.setResourceAdditionalFields")
 	defer exit(err)
 
+	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 {
+		ko.Spec.ProvisionedConcurrencyConfig = nil
+	} else {
+		ko.Spec.ProvisionedConcurrencyConfig.ProvisionedConcurrentExecutions = getProvisionedConcurrencyConfigOutput.RequestedProvisionedConcurrentExecutions
+	}
+
 	var getFunctionEventInvokeConfigOutput *svcsdk.GetFunctionEventInvokeConfigOutput
 	getFunctionEventInvokeConfigOutput, err = rm.sdkapi.GetFunctionEventInvokeConfigWithContext(
 		ctx,

generator.yaml

@@ -3,4 +3,11 @@ if ko.Spec.FunctionEventInvokeConfig != nil {
    if err != nil{
       return nil, err
    }
+}
+
+if ko.Spec.ProvisionedConcurrencyConfig != nil {
+   err = rm.updateProvisionedConcurrency(ctx,desired)
+   if err != nil{
+      return nil, err
+   }
 }

helm/crds/lambda.services.k8s.aws_aliases.yaml

@@ -6,4 +6,10 @@ if delta.DifferentAt("Spec.FunctionEventInvokeConfig"){
 }
 if !delta.DifferentExcept("Spec.FunctionEventInvokeConfig"){
     return desired, nil
+}
+if delta.DifferentAt("Spec.ProvisionedConcurrencyConfig.ProvisionedConcurrentExecutions"){
+    err = rm.updateProvisionedConcurrency(ctx, desired)
+    if err != nil {
+        return nil, err
+    }
 }

pkg/resource/alias/delta.go

@@ -0,0 +1,13 @@
+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

pkg/resource/alias/hooks.go

@@ -160,4 +160,15 @@ def get_function_event_invoke_config_alias(self, function_name:str, qualifier:st
             return None
 
     def function_event_invoke_config_exists(self, function_name: str) -> bool:
-        return self.get_function_event_invoke_config(function_name) is not None
+        return self.get_function_event_invoke_config(function_name) is not None
+    
+    def get_provisioned_concurrency_config(self, function_name: str, qualifier:str) -> dict:
+        try: 
+            resp = self.lambda_client.get_provisioned_concurrency_config(
+                FunctionName = function_name,
+                Qualifier = qualifier
+            )
+            return resp
+        except Exception as e:
+            logging.debug(e)
+            return None