Lambda에서 인수를 지정하여 AWS Batch를 시작하는 방법

개요

준비

Lambda 준비

import json
import boto3

def lambda_handler(event, context):

    client = boto3.client('batch')

    JOB_NAME = 'pandas-envtest'
    JOB_QUEUE = "arn:aws:batch:ap-northeast-1:XXXXXXXXXX:job-queue/first-run-job-queue"
    JOB_DEFINITION = "arn:aws:batch:ap-northeast-1:XXXXXXXXXX:job-definition/pandas-envtest:1"

    response = client.submit_job(
        jobName = JOB_NAME,
        jobQueue = JOB_QUEUE,
        jobDefinition = JOB_DEFINITION,
    containerOverrides={
        'command': [
            'test','abc','ckd'
        ],
        'environment': [
            {
                'name': 'DYNAMODB_REGION',
                'value': 'test'
            }
        ]
    }
    )
    print(response)

    return {
        'statusCode': 200,
        'body': json.dumps('run')
    }

인수로 지정한 내용과 환경 변수가 출력되고 있습니다.