More than 3 years have passed since last update.

AWS Lambda + Python + Slack でサイトの監視・アラート通知

Posted at 2020-03-22

運用中のサイトがたまに落ちていることがあったので、
簡易的にですがサイト監視を用意してみました。
（世の中にはサーバーのメトリクス取得含め手厚い監視サービスが様々ありますが、今回は低コストかつSlackとも簡単に連携できる方法という位置づけで作っています）

サイトのstatusをチェックしてSlackに通知する

では、サイトチェックをするコードを書いていきます。
おおまかな流れとしては、指定されたURLのHTTP statusをチェックし、200以外のサイトがあれば、Slackに通知するというようになっています。
（逆に問題が無ければ、Slackには何も通知されません。テストの際には404ページのURLなどを含めてあげてください。）
全体は以下のようになります。

health_checker.py

# -*- coding: utf-8 -*-
import requests
import json

# 監視したいサイトのURLを配列で指定
urls = ['https://www.rakuten.co.jp/']


def post_slack_h(c):
    # SlackでIncoming Webhooksを有効にして取得したWebhook URLを記載
    post_url = 'SlackのWebhook URL'
    requests.post(post_url, data=json.dumps({
        "username": "サイト見守るくん",
        'text': c
    }))


targets = []


def status_check(a="", b=""):
    content = ""
    for url in urls:
        try:
            s = requests.get(url, timeout=10).status_code
        except requests.exceptions.ReadTimeout:
            content += url + " Time out(10s)\n"
            targets.append(url)
        except requests.exceptions.ConnectionError:
            content += url + " Connection error\n"
            targets.append(url)
        else:
            if s == 200:
                content += url + " " + str(s) + "\n"
            else:
                content += url + " " + str(s) + "\n"
                targets.append(url)

    if not targets:
        print("All Green!!")
        pass
    else:
        post_slack_h(content)


if __name__ == '__main__':
    status_check()

Slackへの通知はIncoming webhookを利用

Incoming webhookの利用の手順は、以下の公式サポートをご確認ください。
https://slack.com/intl/ja-jp/help/articles/115005265063

コードのzipファイルを準備

AWS Lambda上で外部パッケージもimportするには外部パッケージのコードもあわせてアップロードする必要があります。
今回使用しているrequestsは外部パッケージなので、メインのコードと一緒にzipファイルにまとめてしまいます。

# メインコードのあるディレクトリにパッケージファイルを保存
$ pip install requests -t ./
# 圧縮してzipファイルにします
$ zip -r health_checker ./*

これでコードと外部パッケージを内包した「health_checker.zip」というzipファイルが出来上がります。

AWS Lambdaで関数を作成する

AWSでLambdaのリンクをクリックすると、関数一覧の画面に遷移します。
ここの画面から、「関数の作成」をクリックします。

作成時の初期設定画面では、以下のように設定します

オプション: 一から作成
関数名: 任意
ランタイム: Python 3.7

続いて、関数の編集画面にある「関数コード」という項目からファイルのアップロードをおこないます。
設定は以下の画面のようにします。
特に注意したいポイントは、「ハンドラ」です。
ハンドラは、Lambdaで実行する関数を指定する箇所です。
指定の仕方はファイル名(拡張子無し).(指定したファイル内の)関数名となります。
ですので、ここでは「health_checker.status_check」となります。