constructor()

in autopilot/mlops/timeseries/aws-automl-ts-cdk/lib/construct/lambda.ts [25:76]
42 lines of code
1 McCabe index (conditional complexity)

    constructor(scope: Construct, id: string, props: LambdaConstructProps) {
        super(scope, id);
        
        const resourceBucketArn = props.resourceBucket.bucketArn;
        
        // Define the policy statement allows Full Access to specified S3 bucket
        const s3BucketFullAccessPolicy = new iam.PolicyStatement({
          actions: ['s3:*'],
          resources: [resourceBucketArn, `${resourceBucketArn}/*`],
        });
        
        // IAM Role
        this.role = new iam.Role(this, `${props.lambdaName}-Lambda-Role`, {
           assumedBy: new iam.ServicePrincipal('lambda.amazonaws.com'),
           roleName: `${props.lambdaName}-Lambda-Role`,
           managedPolicies: [
                {managedPolicyArn: 'arn:aws:iam::aws:policy/CloudWatchFullAccess'},
                {managedPolicyArn: 'arn:aws:iam::aws:policy/AmazonSageMakerFullAccess'}
            ],
            inlinePolicies: {
                's3BucketReadOnly': new iam.PolicyDocument({
                    statements: [s3BucketFullAccessPolicy]
                })
            }
        });
        
        // Lambda Function
        this.lambda = new lambda.Function(this, `${props.lambdaName}-Lambda-Function`, {
            code: lambda.Code.fromAsset(props.lambdaCodePath),
            handler: 'index.handler',
            functionName: props.lambdaName,
            runtime: lambda.Runtime.PYTHON_3_11,
            timeout: props.timeout,
            role: this.role,
            environment: props.environment
        });
        
        // Define StepFunction task for this Lambda
        this.task = new sfn_tasks.LambdaInvoke(this, `${props.taskName} Lambda Task`, {
            lambdaFunction: this.lambda,
            integrationPattern: sfn.IntegrationPattern.REQUEST_RESPONSE,
            resultPath: sfn.JsonPath.stringAt('$'),
            outputPath: sfn.JsonPath.stringAt('$.Payload')
        });
        
        this.task.addRetry({
            backoffRate: 1.0,
            errors: ['ResourcePending'],
            interval: cdk.Duration.seconds(30),
            maxAttempts: 600
        });
    }