AWS Systems Manager Change Calendar で AWS Lambda のスケジュール実行を制御しよう

こんにちは、マネージドサービス部テクニカルサポート課の坂口です。

AWS Systems Manager Change Calendar で AWS Lambda のスケジュール実行を制御する方法についてまとめました。

はじめに
構成図
Google カレンダーのエクスポート
AWS Systems Manager Change Calendar の作成
AWS Lambda 関数の作成
IAM ポリシーの作成
IAM ロールの作成
AWS Step Functions ステートマシンの作成
Amazon EventBridge スケジューラの作成
動作確認
まとめ

はじめに

AWS Lambdaをスケジュール実行させたい場合、Amazon EventBridge を使用すると思いますが、Amazon EventBridgeだけでは曜日の指定は出来るものの、祝日の指定が出来ないため困っていました。

今回は、AWS Systems Manager Change Calendar + Amazon EventBridge + AWS Step Functions を使用し、祝日を除いた平日に AWS Lambda を実行する方法について説明します。

構成図

Google カレンダーのエクスポート

AWS Systems Manager Change Calendar では、祝日の設定がなく手作業での追加が必要となるため、Google カレンダーの祝日情報のファイルを使用します。

Google カレンダーの設定で、日本の祝日を選択し「iCal 形式の公開 URL」をコピーします。
コピーした URL をブラウザにペーストすると、「basic.ics」というファイルが自動的にダウンロードされます。
ダウンロードされた「basic.ics」をテキストエディタで開き、7行目の X-WR-TIMEZONE:UTC を X-WR-TIMEZONE:Asia/Tokyo に変更し保存します。

AWS Systems Manager Change Calendar の作成

AWS Systems Manager Change Calendar とは、指定したアクション (Systems Manager Automation ランブックなど) が AWS アカウントで実行できるまたはできない日付と時刻の範囲（カレンダー）を設定できます。

docs.aws.amazon.com

AWS マネジメントコンソールで、AWS Systems Manager Change Calendar を開き、「カレンダーを作成」を押下します。
「カレンダー名」を入力後、「Import calendar」を押下し「basic.ics」を選択します。
「カレンダーを作成」を押下します。

AWS Lambda 関数の作成

今回は、関数作成時のデフォルト状態のまま使用します。

AWS マネジメントコンソールで AWS Lambda 関数を開き、「関数の作成」を押下します。
「関数名」を入力し、「関数の作成」を押下します。

IAM ポリシーの作成

後述する IAM ロールにアタッチするため、IAM ポリシーを事前に作成しておきます。

AWS マネジメントコンソールで IAM ポリシーを開き、「ポリシーを作成」を押下します。
以下のポリシーを自身の環境に修正し、IAM ポリシーを作成します。

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "ssm:GetCalendarState"
            ],
            "Resource": [
                "arn:aws:ssm:<AWS リージョン>:<AWS アカウント>:document/<カレンダー名>"
            ]
        }
    ]
}

IAM ロールの作成

AWS Step Functions ステートマシンで使用する IAM ロールを事前に作成しておきます。

AWS マネジメントコンソールで IAM ロールを開き、「ロールを作成」を押下します。
ユースケースのサービスに「Step Functions」を選択し、「次へ」を押下し続け IAM ロールを作成します。
作成した IAM ロールにて先ほど作成した IAM ポリシーをアタッチします。

AWS Step Functions ステートマシンの作成

AWS マネジメントコンソールで AWS Step Functions ステートマシンを開き、「ステートマシンの作成」を押下します。
「コードでワークフローを記述」を選択します。
以下のコードを自身の環境に修正しコピーします。

{
  "Comment": "",
  "StartAt": "GetCalendarState",
  "States": {
    "GetCalendarState": {
      "Type": "Task",
      "Parameters": {
        "CalendarNames": [
          "Holiday-Calendar"
        ]
      },
      "Resource": "arn:aws:states:::aws-sdk:ssm:getCalendarState",
      "Next": "Choice"
    },
    "Choice": {
      "Type": "Choice",
      "Choices": [
        {
          "Variable": "$.State",
          "StringEquals": "OPEN",
          "Next": "Lambda Invoke"
        }
      ],
      "Default": "Pass"
    },
    "Pass": {
      "Type": "Pass",
      "End": true
    },
    "Lambda Invoke": {
      "Type": "Task",
      "Resource": "arn:aws:states:::lambda:invoke",
      "OutputPath": "$.Payload",
      "Parameters": {
        "Payload.$": "$",
        "FunctionName": "arn:aws:lambda:<AWS リージョン>:<AWS アカウント>:function:<Lambda 関数名>:$LATEST"
      },
      "Retry": [
        {
          "ErrorEquals": [
            "Lambda.ServiceException",
            "Lambda.AWSLambdaException",
            "Lambda.SdkClientException",
            "Lambda.TooManyRequestsException"
          ],
          "IntervalSeconds": 2,
          "MaxAttempts": 6,
          "BackoffRate": 2
        }
      ],
      "End": true
    }
  }
}