파이썬으로 AWS Step Functions 개발하기

20 Sep 2023

python

AWS Step Functions는 분산 응용 프로그램의 워크플로우를 쉽게 구축, 실행 및 조정할 수 있는 서비스입니다. 해당 서비스를 사용하면 각 단계를 단순한 JSON 기반 상태 머신으로 정의할 수 있으며, 복잡한 작업 흐름을 구축할 수 있습니다. 이번 포스트에서는 파이썬을 사용하여 AWS Step Functions를 개발하는 방법을 살펴보겠습니다.

1. AWS Step Functions 및 Boto3 설치

첫 번째 단계로, AWS CLI를 사용하여 AWS Step Functions와 Boto3를 설치해야 합니다. 다음 명령어를 사용하여 설치할 수 있습니다:

pip install awscli boto3

2. AWS IAM 역할 생성

AWS IAM(Identity and Access Management) 콘솔을 사용하여 AWS Step Functions에 대한 적절한 역할을 생성해야 합니다. 이 역할은 AWS 리소스에 대한 액세스 권한을 설정하고 AWS Step Functions의 실행을 위한 권한을 부여합니다.

3. AWS Step Functions 정의

다음은 AWS Step Functions를 정의하는 파이썬 코드의 예입니다:

import boto3
import json

def create_state_machine(sfn_client):
    definition = {
      "Comment": "Example State Machine",
      "StartAt": "Hello World",
      "States": {
        "Hello World": {
          "Type": "Pass",
          "Result": "Hello, World!",
          "End": True
        }
      }
    }
    
    response = sfn_client.create_state_machine(
        name='HelloWorldStateMachine',
        definition=json.dumps(definition),
        roleArn='arn:aws:iam::123456789012:role/StepFunctions-HelloWorld'
    )
    
    return response['stateMachineArn']

# AWS Step Functions 클라이언트 생성
sfn_client = boto3.client('stepfunctions')

# 상태 기계 생성
state_machine_arn = create_state_machine(sfn_client)

print(f'Created State Machine: {state_machine_arn}')

위의 코드에서는 “Hello World”라는 단계를 생성하는 간단한 상태 머신을 정의합니다. 상태 머신은 JSON 형식으로 작성되며, 각 단계는 Type, Result, End 등의 속성을 가지고 있습니다.

4. AWS Step Functions 실행

상태 기계를 성공적으로 생성하면, 해당 기계를 실행할 수 있습니다. 다음은 기계를 실행하는 예제 코드입니다:

import boto3

def start_execution(sfn_client, state_machine_arn):
    response = sfn_client.start_execution(
        stateMachineArn=state_machine_arn
    )
    
    return response['executionArn']

# AWS Step Functions 클라이언트 생성
sfn_client = boto3.client('stepfunctions')

# 상태 기계 실행
execution_arn = start_execution(sfn_client, state_machine_arn)

print(f'Started Execution: {execution_arn}')

위의 코드에서는 이전 단계에서 생성한 상태 머신을 실행합니다. 실행이 성공하면 실행 ARN을 반환합니다.

이제 위의 예제 코드를 사용하여 파이썬으로 AWS Step Functions를 개발할 수 있습니다. AWS Step Functions를 사용하면 다양한 작업 흐름을 구축하고 복잡한 응용 프로그램을 효율적으로 관리할 수 있습니다.

#AWS #StepFunctions