Architecture/AWS

AWS Lambda로 Slack과 연동해서 알람 설정 (RDS PostgreSQL, 람다 슬랙 연동, SNS 구독 설정)

탱젤 2023. 9. 24. 16:30

RDS PostgreSQL Slack 알람 테스트 (slack incoming-webhook 앱 이용)

Slack에 알림 수신을 위한 앱 추가 (incoming-webhook)

구성 선택해서 구독을 원하는 슬랙 채널을 선택

선택 후 웹후크 URL 복사해둔 뒤 (람다 생성 시 필요) 설정 저장

Slack 수신을 위한 AWS Lambda 함수 생성

역할 선택 후 SNS 트리거는 아직 주제 생성 전이기에 제거한다.

람다 생성 시 환경 변수 설정

kmsEncryptedHookUrl과 slackChannel 값을 입력해준다

Url 정보와 slackChannel은 위에 슬랙에서 앱 설정할 시 정보들을 입력해준다

생성 완료

Amazon SNS 주제 생성

표준 선택, 이름 입력 후 생성

생성된 주제로 들어가서 구독 생성

주제 ARN으로 위에서 생성한 람다 선택 후 프로토콜 AWS Lambda 선택 → 구독 생성

구독 생성된 것 확인

테스트 알람 수신

생성된 람다 함수로 들어가서 테스트 이벤트 생성 후 테스트 진행

슬랙 테스트 오류 수정 -> hook_url 수정 및 동시성 편집

AWS Lambda를 이용한 Sequence Check 슬랙 알람까지

rds slack 시퀀스 test -> psycopg2 모듈 찾을 수 없다고 뜨는 에러 수정 -> layer 추가하면됨
- AWS Lambda 에서는 기본적으로 DB 커넥션에 필요한 모듈 기능 제공을 안한다.(포스트그레 - psycopg2, mysql - psmyqsl 등) → psycopg2를 개별적으로 받아서 Lambda내 Layer(계층)라는 곳에 올려야한다.
- Add a layer(계층생성) 에 아래에 추가된 psycopg2-layger.zip 파일 업로드

Layers 생성 확인!

postgre_alarm_test_ty 람다 생성 시 코드

import json
import os
import psycopg2
import urllib.request

ENCRYPTED_HOOK_URL = os.environ['kmsEncryptedHookUrl']#해당 환경변수들은 구성에서 추가
SLACK_CHANNEL =os.environ['slack_channel']

def post_slack(argStr):
    message = argStr
    send_data = {
        "text": message,
    }
    send_text = json.dumps(send_data)
    request = urllib.request.Request(
        ENCRYPTED_HOOK_URL,
        data=send_text.encode('utf-8'),
    )

    with urllib.request.urlopen(request) as response:
        slack_message = response.read()

def lambda_handler(event, context):
# TODO implement
    sql = 'select * from (    select sequencename , round(("last_value")::numeric/max_value::numeric, 10) as pct from pg_catalog.pg_sequences) a where a.pct > 0.07;'

    conn = psycopg2.connect(
        host = os.environ['DB_HOST'],
        dbname = os.environ['DB_NAME'],
        user = os.environ['DB_USER'],
        password = os.environ['DB_PASSWORD'],
        port = os.environ['DB_PORT']
        )

    cur = conn.cursor()

#print(cur.mogrify(sql))
    cur.execute(sql)

    result = cur.fetchall()

    for cur in result:
        if cur is not None:
            msg = cur[0] + ' is urgent'
            post_slack(os.environ['DB_NAME'] + ' ' + msg + '(taeyeong)')
        else:
            post_slack("sequence is normal.")

    conn.close()
    return {
        'statusCode': 200,
        'body': json.dumps('Hello from Lambda!')
    }