Step Functions で EC2 + RDS 自動停止・起動ハンズオン

📋 概要

このハンズオンでは、AWS Step Functions を使って EC2 と RDS の起動・停止に 順序と状態確認を加えた自動化ワークフローを構築します。「EC2 が running になってから RDS を起動する」「RDS が stopped になってから EC2 を停止する」という依存関係を Step Functions のステートマシンで実現します。EventBridge Scheduler で毎朝・毎夜に自動実行します。

🏗️ 構成図

起動フロー（朝 8 時）

停止フロー（夜 22 時）

作成する Lambda 関数一覧

✅ 前提条件

• 停止・起動できる EC2 インスタンス（例: handson-webap-1a）のインスタンス ID
• 停止・起動できる RDS DB インスタンス（例: handson-db）の DB 識別子
• リージョン: ap-northeast-1（東京）

🔧 ステップ 1: Lambda 関数の作成（6 関数）

まず IAM ポリシーとロールを作成してから、6 つの Lambda 関数を作成します。

1-0. 共通 IAM ロールの作成

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Effect": "Allow",
      "Action": [
        "ec2:StartInstances", "ec2:StopInstances", "ec2:DescribeInstances",
        "rds:StartDBInstance", "rds:StopDBInstance", "rds:DescribeDBInstances"
      ],
      "Resource": "*"
    },
    {
      "Effect": "Allow",
      "Action": ["logs:CreateLogGroup","logs:CreateLogStream","logs:PutLogEvents"],
      "Resource": "arn:aws:logs:*:*:*"
    }
  ]
}

次に IAM ロールを作成します。

1-1. handson-sfn-start-ec2

import boto3, os

def lambda_handler(event, context):
    instance_id = os.environ['EC2_INSTANCE_ID']
    boto3.client('ec2').start_instances(InstanceIds=[instance_id])
    print(f'EC2 {instance_id}: 起動リクエスト送信')
    return {**event, 'ec2_instance_id': instance_id}

1-2. handson-sfn-check-ec2

import boto3, os

def lambda_handler(event, context):
    instance_id = event.get('ec2_instance_id', os.environ['EC2_INSTANCE_ID'])
    resp  = boto3.client('ec2').describe_instances(InstanceIds=[instance_id])
    state = resp['Reservations'][0]['Instances'][0]['State']['Name']
    print(f'EC2 {instance_id}: {state}')
    return {**event, 'ec2_state': state}

1-3. handson-sfn-start-rds

import boto3, os

def lambda_handler(event, context):
    db_id = os.environ['RDS_IDENTIFIER']
    boto3.client('rds').start_db_instance(DBInstanceIdentifier=db_id)
    print(f'RDS {db_id}: 起動リクエスト送信')
    return {**event, 'rds_identifier': db_id}

1-4. handson-sfn-stop-rds

import boto3, os

def lambda_handler(event, context):
    db_id = os.environ['RDS_IDENTIFIER']
    boto3.client('rds').stop_db_instance(DBInstanceIdentifier=db_id)
    print(f'RDS {db_id}: 停止リクエスト送信')
    return {**event, 'rds_identifier': db_id}

1-5. handson-sfn-check-rds

import boto3, os

def lambda_handler(event, context):
    db_id  = event.get('rds_identifier', os.environ['RDS_IDENTIFIER'])
    resp   = boto3.client('rds').describe_db_instances(DBInstanceIdentifier=db_id)
    status = resp['DBInstances'][0]['DBInstanceStatus']
    print(f'RDS {db_id}: {status}')
    return {**event, 'rds_status': status}

1-6. handson-sfn-stop-ec2

import boto3, os

def lambda_handler(event, context):
    instance_id = event.get('ec2_instance_id', os.environ['EC2_INSTANCE_ID'])
    boto3.client('ec2').stop_instances(InstanceIds=[instance_id])
    print(f'EC2 {instance_id}: 停止リクエスト送信')
    return {**event, 'ec2_state': 'stopping'}

🔧 ステップ 2: Step Functions 実行ロールの作成

Step Functions が Lambda を呼び出すための IAM ロールを作成します。

2-1. IAM ポリシーの作成

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Effect": "Allow",
      "Action": "lambda:InvokeFunction",
      "Resource": "arn:aws:lambda:ap-northeast-1:*:function:handson-sfn-*"
    },
    {
      "Effect": "Allow",
      "Action": ["logs:CreateLogDelivery","logs:PutLogEvents","logs:CreateLogGroup"],
      "Resource": "*"
    }
  ]
}

2-2. IAM ロールの作成

🔧 ステップ 3: 起動ステートマシンの作成

3-1. Step Functions コンソールを開く

3-2. ASL（Amazon States Language）の入力

以下の JSON をそのまま貼り付けます。Lambda 関数の ARN は 自動取得のため手動置換不要です（Resource に関数名を指定する方法を使います）。

{
  "Comment": "EC2 running 確認後に RDS を起動するワークフロー",
  "StartAt": "StartEC2",
  "States": {
    "StartEC2": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:ap-northeast-1:<ACCOUNT_ID>:function:handson-sfn-start-ec2",
      "Next": "WaitEC2Boot",
      "ResultPath": "$.start_ec2_result"
    },
    "WaitEC2Boot": {
      "Type": "Wait",
      "Seconds": 15,
      "Next": "CheckEC2Status"
    },
    "CheckEC2Status": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:ap-northeast-1:<ACCOUNT_ID>:function:handson-sfn-check-ec2",
      "Next": "IsEC2Running",
      "ResultPath": "$"
    },
    "IsEC2Running": {
      "Type": "Choice",
      "Choices": [
        {
          "Variable": "$.ec2_state",
          "StringEquals": "running",
          "Next": "StartRDS"
        }
      ],
      "Default": "WaitEC2Boot"
    },
    "StartRDS": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:ap-northeast-1:<ACCOUNT_ID>:function:handson-sfn-start-rds",
      "End": true
    }
  }
}

「作成」をクリックします。

🔧 ステップ 4: 停止ステートマシンの作成

起動と同じ手順でステートマシンを作成します。

{
  "Comment": "RDS stopped 確認後に EC2 を停止するワークフロー",
  "StartAt": "StopRDS",
  "States": {
    "StopRDS": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:ap-northeast-1:<ACCOUNT_ID>:function:handson-sfn-stop-rds",
      "Next": "WaitRDSStop",
      "ResultPath": "$.stop_rds_result"
    },
    "WaitRDSStop": {
      "Type": "Wait",
      "Seconds": 30,
      "Next": "CheckRDSStatus"
    },
    "CheckRDSStatus": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:ap-northeast-1:<ACCOUNT_ID>:function:handson-sfn-check-rds",
      "Next": "IsRDSStopped",
      "ResultPath": "$"
    },
    "IsRDSStopped": {
      "Type": "Choice",
      "Choices": [
        {
          "Variable": "$.rds_status",
          "StringEquals": "stopped",
          "Next": "StopEC2"
        }
      ],
      "Default": "WaitRDSStop"
    },
    "StopEC2": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:ap-northeast-1:<ACCOUNT_ID>:function:handson-sfn-stop-ec2",
      "End": true
    }
  }
}

🔧 ステップ 5: EventBridge Scheduler の設定

Step Functions を定時実行するための Scheduler を 2 つ作成します。まず Scheduler 用の IAM ロールが必要です。

5-1. Scheduler 用 IAM ロールの作成

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Effect": "Allow",
      "Action": "states:StartExecution",
      "Resource": [
        "arn:aws:states:ap-northeast-1:*:stateMachine:handson-start-sfn",
        "arn:aws:states:ap-northeast-1:*:stateMachine:handson-stop-sfn"
      ]
    }
  ]
}

IAM ロールを作成します。

5-2. 起動スケジューラーの作成

5-3. 停止スケジューラーの作成

🔧 ステップ 6: 動作確認

6-1. 停止ステートマシンを手動実行

6-2. 実行フロー図の確認

6-3. 入出力の確認

6-4. 起動ステートマシンを手動実行

EC2 と RDS が停止したことを確認してから、handson-start-sfn を同様に手動実行します。「EC2 が running になってから RDS が起動リクエストを受け取る」フローを確認してください。

6-5. 実行履歴の確認

🧹 クリーンアップ

削除手順（逆順で実施）

1. EventBridge Scheduler の削除
   EventBridge → Scheduler → handson-sfn-start-schedule と handson-sfn-stop-schedule を削除
2. Step Functions ステートマシンの削除
   Step Functions → handson-start-sfn と handson-stop-sfn を選択 → 「削除」
3. Lambda 関数の削除（6 関数）
   Lambda → handson-sfn-start-ec2, check-ec2, start-rds, stop-rds, check-rds, stop-ec2 を削除
4. IAM ロール・ポリシーの削除
   IAM → ロール: handson-sfn-execution-role, handson-sfn-lambda-role, handson-scheduler-sfn-role を削除
   IAM → ポリシー: handson-sfn-execution-policy, handson-sfn-lambda-policy, handson-scheduler-sfn-policy を削除