More than 1 year has passed since last update.

stepfunctionsでのDynamoDBを使用した多重実行制御

Last updated at 2022-05-13Posted at 2022-05-13

はじめに

AWSのEventBridgeを使用して定期実行などを行う際、最低１回の実行が保証されている（参照）。
しかし、処理によっては多重実行が望まれないものもある。
そこで今回は、DynamoDBを用いて、stepfunctionsの多重実行を制御する方法を考えた。

実装

今回は、日次で１回のみ行う処理をイメージした。ステートマシンの構造は以下の通りである。

DynamoDBに実行日を表す値(yyyymmdd)を入れる
上の処理が正常に動作したらメインの処理が走る、エラーが起こったらステートマシンを終了する

DynamoDBのテーブル作成

今回はtestというテーブル名で、パーティションキーをyyyymmddとするテーブルを作成した。

ステートマシンの作成

以下のようなイメージのステートマシンを作成した。
なお、今回は、多重実行が制御できれば良いので、メインの処理はmainという名のpassステートで代用してある。

{
  "Comment": "A description of my state machine",
  "StartAt": "PutItem",
  "States": {
    "PutItem": {
      "Type": "Task",
      "Resource": "arn:aws:states:::dynamodb:putItem",
      "Parameters": {
        "TableName": "test",
        "ConditionExpression": "attribute_not_exists(yyyymmdd)",
        "Item": {
          "yyyymmdd": {
            "S.$": "$.yyyymmdd"
          }
        }
      },
      "ResultPath": null,
      "Catch": [
        {
          "ErrorEquals": [
            "DynamoDB.ConditionalCheckFailedException"
          ],
          "Next": "Already put"
        }
      ],
      "Next": "main"
    },
    "Already put": {
      "Type": "Pass",
      "End": true
    },
    "main": {
      "Type": "Pass",
      "End": true
    }
  }
}

パラメータ説明

putItem
Request Parameterに関するドキュメント

"parameters"内の"ConditionExpression"を"attribute_not_exsits(yyyymmdd)"とすることで、指定したyyyymmddが既に存在しているかどうかを判定する（参照）。
既にDynamoDBに値が存在している場合、"DynamoDB.ConditionalCheckFailedException"が返却されるので、これを"Catch"で受け取っている。