株式会社BeeX

LambdaでAWS利用料金をSlackへDM送信する方法

Last updated at 2025-01-25Posted at 2025-01-25

やりたいこと

最近、AWSのAI系サービスを検証しており、あまり詳しくないサービスを触っているうちに、知らないうちにAWS料金が大量にかかってしまうリスクを感じています。

毎日マネジメントコンソールで料金を確認するのが面倒なので、AWS利用料金を定期的にSlackに送信する仕組みを作ってみました。

こんなことをやりたいです↓

①Slackアプリの作成

以下のURLにアクセスして、Slackアプリを作ります
https://api.slack.com/apps?new_app=1

「From a manifest」で作成

ワークスペースを選択

manifestを設定。アプリ名を決めて、その他は全部デフォルトで大丈夫。

設定内容をレビューして、「Create」

②SlackのIncoming　Webhooksを設定

アプリ作成できたら、このような画面に遷移されます。

左メニューの「Incoming　Webhooks」をクリックし、設定を「On」に

少し下スクロールして、「Add New Webhook to Workspace」をクリック

チャンネル検索のところに、自分自身の名前を検索し設定し、「許可」
（特定のチャンネルへ投稿する場合はチャンネル名で設定しますが、今回は自分へDMを送りたいため名前で検索します。）

すると、Webhook URLが作成されました。いったんコピーしておきます。

③Parameter Storeの設定

AWSの「System Manager」を開き、左メニューの「パラメータストア」をクリックし、「パラメータの作成」

パラメータ名を入力し、「標準」利用枠にします。

タイプを「文字列」にし、タイプ型を「text」にし、値を先ほどコピーしたSlackのWebhookで入力し、「パラメータを作成」

作成に成功しました！

④Lambdaの作成

AWSのLambdaを開き、「関数を作成」

「一から作成」を選択し、関数名を入力します。今回はPython3.13で作ります。「関数の作成」をクリック

作成完了後、この画面に遷移

以下のソースでデプロイします。

import base64
import urllib3
import json
import boto3
import datetime
 
http = urllib3.PoolManager()

def lambda_handler(event, context):
    # SSMクライアントを作成
    ssm_client = boto3.client('ssm')

    # パラメータを取得
    parameter_name = 'slack-url'  # Parameter Storeに保存されているパラメータ名
    response = ssm_client.get_parameter(Name=parameter_name, WithDecryption=True)
    
    # URLを取得
    slack_endpoint = response['Parameter']['Value']
    print(f"Retrieved URL: {slack_endpoint}")

    # Cost Explorer クライアントの作成
    ce_client = boto3.client('ce')
    
    # 今日の日付と月初の日付を取得
    today = datetime.date.today()
    start_of_month = today.replace(day=1)

    try:
        response = ce_client.get_cost_and_usage(
            TimePeriod={
                'Start': start_of_month.strftime('%Y-%m-%d'),
                'End': today.strftime('%Y-%m-%d')
            },
            Granularity='MONTHLY',
            Metrics=['UnblendedCost']
        )
        
        # 合計料金を取得
        total_cost = response['ResultsByTime'][0]['Total']['UnblendedCost']['Amount']
        currency = response['ResultsByTime'][0]['Total']['UnblendedCost']['Unit']

        msg = {
            "text": f"今月のAWS料金：{total_cost} {currency}",
        }
        encoded_msg = json.dumps(msg).encode('utf-8')
        resp = http.request('POST', slack_endpoint, body=encoded_msg)

        print(resp)

        return {
            "statusCode": 200,
            "body": json.dumps("Message sent successfully!")
        }
    except Exception as e:
        return {
            "statusCode": 500,
            "body": json.dumps(str(e))
        }