Amazon Aurora のスロークエリを Slack に通知する

スロークエリを簡単に確認できるようにしたかったので、Lambda を使って Aurora でスロークエリが発生するたびに Slack に通知するようにしてみました。

ログのエクスポート設定

まずは、クラスターの変更から Amazon CloudWatch Logs にスロークエリログを発行するようにします。

f:id:enomotodev:20181212212733p:plain

パラメータグループを編集

デフォルトのパラメータグループでは slow_query_log が OFF になっており、スロークエリが出力されないようになっているので、パラメータグループを編集して slow_query_log を 1 にして有効にしましょう。

ついでにデフォルトでは long_query_time が 10 になっており、10 秒以上のクエリではないとスロークエリに出力されないので、long_query_time も 1 など小さい数値に変更します。

IAM ロールの作成

Lambda 用の IAM ロールを作成します。

ロール名：slowquery-to-slack-role
このロールを使用するサービスを選択：Lambda
ポリシー：AWSLambdaBasicExecutionRole

として作成します。

Webhook URL の発行

Slack の Webhook URL を用意していない場合は Slack の Incoming Webhooks のページから Webhook URL を発行してください。

次の Lambda を作成する際に Slack の Webhook URL を使用します。

Lambda の作成

IAM ロールと Slack の Webhook URL が作成できたら Lambda 関数を作成します。

一から作成を選択し、

名前：slowquery-to-slack
ランタイム：Python 3.7
ロール：既存のロールを選択
既存のロール：slowquery-to-slack-role

として作成します。

関数コードは下記をそのままコピペして使用してください。

import os
import base64
import gzip
import json
import logging
from io import BytesIO
from urllib.request import Request, urlopen
from urllib.error import URLError, HTTPError

Slack_Webhook_URL = os.environ['SLACK_WEBHOOK_URL']

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

def lambda_handler(event, context):
    buffer = BytesIO(base64.b64decode(json.dumps(event['awslogs']['data'])))
    with gzip.GzipFile(mode='rb', fileobj=buffer) as sb:
        data = sb.read()

    log = json.loads(data.decode('utf-8'))
    message = str(log['logEvents'][0]['message'])

    slack_message = {
        'text': '*New Slow Query Alert*',
        'attachments': [
            {
                'color': '#36464f',
                'text': message,
                'footer': 'Amazon RDS'
            }
        ]
    }

    req = Request(Slack_Webhook_URL, json.dumps(slack_message).encode('utf-8'))
    try:
        response = urlopen(req)
        response.read()
        logger.info('Message posted to %s', Slack_Webhook_URL)
    except HTTPError as e:
        logger.error('Request failed: %d %s', e.code, e.reason)
    except URLError as e:
        logger.error('Server connection failed: %s', e.reason)