AWS CDKとEventBridge Schedulerで実装するAurora Serverless v2のキャパシティ自動調整

import * as cdk from "aws-cdk-lib";
import * as iam from "aws-cdk-lib/aws-iam";
import * as rds from "aws-cdk-lib/aws-rds";
import * as scheduler from "aws-cdk-lib/aws-scheduler";
import * as scheduler_targets from "aws-cdk-lib/aws-scheduler-targets";
import { Construct } from "constructs";

interface AcuSchedulerConstructProps {
  cluster: rds.DatabaseCluster;
}

export class AcuSchedulerConstruct extends Construct {
  constructor(scope: Construct, id: string, props: AcuSchedulerConstructProps) {
    super(scope, id);

    const { cluster } = props;

    /**
     * Aurora データベースのキャパシティを自動的に切り替えるスケジューラー
     */

    // EventBridge Scheduler がAuroraクラスターを操作するためのIAMロールを作成
    const role = new iam.Role(this, "SchedulerRole", {
      assumedBy: new iam.ServicePrincipal("scheduler.amazonaws.com"),
    });

    // Aurora クラスターのキャパシティ変更権限をIAMロールに付与
    role.addToPolicy(
      new iam.PolicyStatement({
        actions: ["rds:ModifyDBCluster"],
        resources: [cluster.clusterArn],
      }),
    );

    // 日本時間23:40に最小キャパシティを0.5に設定
    new scheduler.Schedule(this, "SetMinCapacity0.5", {
      schedule: scheduler.ScheduleExpression.cron({
        minute: "40",
        hour: "23",
        timeZone: cdk.TimeZone.ASIA_TOKYO,
      }),
      target: new scheduler_targets.Universal({
        service: "rds",
        action: "modifyDBCluster",
        role,
        input: scheduler.ScheduleTargetInput.fromObject({
          DbClusterIdentifier: cluster.clusterIdentifier,
          ServerlessV2ScalingConfiguration: {
            MinCapacity: 0.5,
          },
          ApplyImmediately: true,
        }),
      }),
    });

    // 日本時間23:45に最小キャパシティを0に設定
    new scheduler.Schedule(this, "SetMinCapacity0", {
      schedule: scheduler.ScheduleExpression.cron({
        minute: "45",
        hour: "23",
        timeZone: cdk.TimeZone.ASIA_TOKYO,
      }),
      target: new scheduler_targets.Universal({
        service: "rds",
        action: "modifyDBCluster",
        role,
        input: scheduler.ScheduleTargetInput.fromObject({
          DbClusterIdentifier: cluster.clusterIdentifier,
          ServerlessV2ScalingConfiguration: {
            MinCapacity: 0,
          },
          ApplyImmediately: true,
        }),
      }),
    });
  }
}