📘

SageMaker Canvas を自動停止させる仕組みを EventBridge + Lambda で構築する

2023/06/15に公開

AWS

SageMaker

tech

はじめに

SageMaker Canvas は明示的にログアウトをしない限り、セッションが残り続けます。
この記事は EventBridge と Lambda を使って、SageMaker Canvas のセッションを自動停止する仕組みを作る備忘録です。

実装する Lambda 関数のコード

aws-samples の以下のリポジトリにある Lambda の実装をそのまま使います。

import boto3
import logging
from botocore.config import Config

logger = logging.getLogger()
logger.setLevel(logging.INFO)

config = Config(retries={"max_attempts": 10, "mode": "standard"})
sagemaker = boto3.client("sagemaker", config=config)
paginator = sagemaker.get_paginator("list_apps")


def delete_app(domain_id, user_profile_name, app_type, app_name):
    logger.info(
        f"deleting {app_type}: {app_name} for user: {user_profile_name} in Domain: {domain_id}"
    )

    sagemaker.delete_app(
        DomainId=domain_id,
        UserProfileName=user_profile_name,
        AppType=app_type,
        AppName=app_name,
    )


def lambda_handler(event, context):
    try:
        app_page_iterator = paginator.paginate(PaginationConfig={"PageSize": 50})

        for app_page in app_page_iterator:
            for app in app_page["Apps"]:
                if app["AppType"] == "Canvas" and app["Status"] != "Deleted":
                    delete_app(
                        app["DomainId"],
                        app["UserProfileName"],
                        app["AppType"],
                        app["AppName"],
                    )

    except Exception as e:
        logger.error(e)

    logger.info("Canvas apps deleted")