やりたいこと
AWSのSQSヘキューされたメッセージをEC2へポーリングしてサーバのコンソール上に通知させたい。以前「AWSのboto3を使ってサーバのコマンド結果をメールに飛ばしてた。」を書きましたが、今回はその延長で。
方法
boto3を利用して、Linuxサーバ上（EC2上で）にスクリプトを作成・実行(スクリプト内容はのちほど)。
前提
※今回は以前作成した記事の延長で、サーバのコマンド結果がSNS経由でSQSへキューされることを前提としています。
- EC2を立てる(ここでのサーバはRHEL系を想定してます。)
- EC2にaws cli、python3、boto3のインストール。
- aws configureでユーザ認証。
- SQSでキュー作成。
コード
事前にSQSのURLを記載したiniファイル作成する。（スクリプトファイルにこれを記載する事も可能だとは思いますが、それを試す余裕がなかったのでQiitaの他記事を参考に今回作成してます。）
### config.ini (拡張子.iniファイル) ###
[sqs]
url :https://sqs.ap-northeast-1.amazonaws.com/9999999999/sqs_queue_test
そして、boto3を利用したスクリプトを以下のように作成。内容としては、SQSのキューにメッセージが受信される度（本スクリプトでは1分毎）に、ポーリング処理でサーバのコンソール上にそのメッセージを表示させる。メッセージは表示後に削除。
### receive_sqs.py ###
import configparser
import boto3
import json
import time
# configファイルの読み込み（iniファイルのパスを指定）
ini = configparser.ConfigParser()
ini.read("/XXXX/config.ini")
sqs = boto3.client('sqs')
url = ini.get("sqs", "url")
# キューの先頭にあるメッセージを表示・削除（ここでは1分毎にポーリング処理）
while True:
response = sqs.receive_message(
QueueUrl=url,
AttributeNames=['SentTimestamp'],
MaxNumberOfMessages=1,
VisibilityTimeout=0,
WaitTimeSeconds=0
)
if 'Messages' in response and response['Messages']:
message = response['Messages'][0]
print(message)
receipt_handle = message['ReceiptHandle']
sqs.delete_message(
QueueUrl=url,
ReceiptHandle=receipt_handle
)
time.sleep(60)
参考にしたもの
以下参考にさせていただきました。ありがとうございました。
・CloudWatchからSNS→SQS、AWS SDK for RubyでQueue取り出し
・Boto3(AWS SDK for Python)でSQSに送信したメッセージをLambdaでポーリングして受信してみた
・AWS SDKでSQSをPythonで操作する