はじめに

こんにちは。技術4課の河野です。今回は、AWS Step Functions(以下「SFN」) で AWS Lambda(以下「Lambda」) がタイムアウトした場合に、再試行する方法について紹介します。 AWS公式ドキュメントにも、事前にエラー処理をステートマシンに定義することがベストプラティクスと記載されています。もちろん、タイムアウト以外にも想定されるエラーはきちんと定義しておきましょう。

今回はServerlss Framework というサーバレスアプリケーションの構築管理ができる無料のオープンソースを使用して構築及びデプロイを行います。インストール方法などの各種準備については割愛します。詳細については、弊社過去記事を参考にしていただければ幸いです。

手順

Lambda 作成

まずは、１０秒間待機させる単純な関数を作成します。

import time


def handler(event=None, context=None):

    # 10秒待機
    time.sleep(10)
    return {}

予め Lambda のタイムアウト時間を５秒に設定することで、タイムアウトエラーを発生させます。

SFN 作成

次に、ステートマシンを定義します。最初に示したワークフローを作成するには、serverless.yml を下記のように編集します。

service: stepfunctions
provider:
  name: aws
  runtime: python3.8
  stage: dev
  region: ap-northeast-1

plugins:
  - serverless
  - serverless-step-functions

package:
  include:
     handler.py
  exclude:
    - node_modules/**
    - .serverless/**

functions:
  timeoutSample:
    handler: handler.handler
    name: ${self:provider.stage}-timeout-sample
    timeout: 5

stepFunctions:
  stateMachines:
    myStepFUnctions:
      name: myStepfunctions
      definition:
        StartAt: executeFunction
        States:
          executeFunction:
            Type: Task
            Resource:
              Fn::GetAtt: [timeoutSample,Arn]
            TimeoutSeconds: 60
            ResultPath: '$'
            Retry:
            - ErrorEquals:
              - 'Lambda.Unknown'
              IntervalSeconds: 1
              MaxAttempts: 2
            End: True

再試行は、上記のRetryフィールドで定義しています。設定している各パラメータについて解説します。

ErrorEquals

キャッチするエラーを指定します。ステートマシンの実行結果と指定したエラーが一致した場合に再試行されます。メモリ不足や関数タイムアウトなどの Lambda で処理されないエラーは、Lambda.Unknown、 States.ALL または、States.TaskFailed で一致できます。その他の Lambda に関するエラーはAWS Lambda Developer Guideを参考にしてキャッチするエラーを指定することができます。