ALB + EC2 Auto Scaling 環境下、"スケーリングプロセスの中断" と "Lambda 関数" を利用してヘルスチェックで失敗したインスタンスを保持しつつスケールアウトも実行させてみた

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "autoscaling:DescribeAutoScalingGroups",
                "autoscaling:UpdateAutoScalingGroup"
            ],
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": "elasticloadbalancing:DescribeTargetHealth",
            "Resource": "*"
        }
    ]
}

import os
import boto3
from botocore.exceptions import ClientError

def lambda_handler(event, context):
    # 環境変数から設定値を取得
    target_asg_arn = os.getenv("TARGET_AUTO_SCALING_GROUP")
    base_desired_capacity = int(os.getenv("BASE_DESIRED_CAPACITY", 2))
    base_max_size = int(os.getenv("BASE_MAX_SIZE", 2))
    target_group_arn = os.getenv("TARGET_GROUP_ARN")

    if not target_asg_arn or not target_group_arn:
        raise ValueError("環境変数 'TARGET_AUTO_SCALING_GROUP' と 'TARGET_GROUP_ARN' を設定してください。")

    # クライアント初期化
    elb_client = boto3.client("elbv2")
    asg_client = boto3.client("autoscaling")

    try:
        # AutoScalingグループ名をARNから取得
        response_asg = asg_client.describe_auto_scaling_groups()
        target_asg_name = None
        for asg in response_asg["AutoScalingGroups"]:
            if asg["AutoScalingGroupARN"] == target_asg_arn:
                target_asg_name = asg["AutoScalingGroupName"]
                break

        if not target_asg_name:
            raise ValueError(f"指定されたARN '{target_asg_arn}' に一致するAutoScalingグループが見つかりませんでした。")

        # ALBターゲットグループのヘルスチェック情報を取得
        response = elb_client.describe_target_health(TargetGroupArn=target_group_arn)
        targets = response.get("TargetHealthDescriptions", [])

        # UnHealthy 状態のインスタンス数をカウント
        unhealthy_count = sum(
            1 for target in targets if target["TargetHealth"]["State"] != "healthy"
        )

        # 希望容量と最大数を計算
        new_desired_capacity = base_desired_capacity + unhealthy_count
        new_max_size = base_max_size + unhealthy_count

        # AutoScalingグループの更新
        asg_client.update_auto_scaling_group(
            AutoScalingGroupName=target_asg_name,
            DesiredCapacity=new_desired_capacity,
            MaxSize=new_max_size
        )

        return {
            "statusCode": 200,
            "message": f"AutoScalingグループ '{target_asg_name}' を更新しました。",
            "desired_capacity": new_desired_capacity,
            "max_size": new_max_size
        }

    except ClientError as e:
        return {
            "statusCode": 500,
            "message": f"エラーが発生しました: {e.response['Error']['Message']}"
        }