Lambdaでヘルスチェックするコードのメモ

完全自分用メモですよ。

# -*- coding: utf-8 -*-
 
import commands
import os
import boto3
 
print('Loading function')

## SNS送信関数
def sendmsg():
    TOPIC_ARN = u'arn:aws:sns:ap-northeast-1:＊＊＊＊＊＊'
 
    msg = "Please check for https://www.example.com/"
    subject = u'Site is down！!'
    client = boto3.client('sns')
 
    request = {
        'TopicArn': TOPIC_ARN,
        'Message': msg,
        'Subject': subject
    }
 
    response = client.publish(**request)

#Linuxのコマンド実行関数
def _(cmd):
    return commands.getoutput(cmd)
 
def lambda_handler(event, context):
    #ヘルスチェックで問題がない場合に表示する文字列
    ok_string = "ok"
     
    #ヘルスチェックで問題がある場合に表示する文字列
    ng_string = "ng"
     
    #実行するcurlコマンド
    curl_cmd = 'curl -LI https://www.example.com/ -o /dev/null -w \'%{http_code}\\n\' -s'

    #curlコマンドの実行
    response = _(curl_cmd)
     
    #レスポンスのログへの記録
    print("----------response body start.----------")
    print(response)
    print("----------response body end.------------")
     
    #ステータスコードの判定
    if response == ("200"):
        print("Status is 200")
        return {"result": ok_string}
    else:
        print("Status is not 200")
        sendmsg()
        return {"result": ng_string}