SNSからLambdaを呼び出す

以下のページでSNSについて取り上げました。

上記のページでは、サブスクライバーとしてメールアドレスを指定しました。

今回はサブスクライバーにLambda関数を指定する方法を確認します。

構築する環境

構成は冒頭にご紹介したページと概ね同様です。

Lambda関数2からSNSトピックにメッセージを送信します。
つまりこの関数はパブリッシャーとして動作します。
この関数の働きは、現在日時からUNIX時間を取得し、メッセージとしてこれを送信します。

SNSトピックのサブスクライバーとしてLambda関数1を指定します。
この関数の働きは、メッセージとして渡されたUNIX時間が偶数/奇数か判定します。

CloudFormationテンプレートファイル

上記の構成をCloudFormationで構築します。
以下のURLにCloudFormationテンプレートを配置しています。

https://github.com/awstut-an-r/awstut-fa/blob/main/100/fa-100.yaml

テンプレートファイルのポイント解説

(参考)Lambda関数でSNSにパブリッシュする

Resources:
  Function2:
    Type: AWS::Lambda::Function
    Properties:
      Architectures:
        - !Ref Architecture
      Environment:
        Variables:
          REGION: !Ref AWS::Region
          TOPIC: !Ref TopicArn
      Code:
        ZipFile: |
          import boto3
          import datetime
          import os
          import time

          topic = os.environ['TOPIC']
          region = os.environ['REGION']

          client = boto3.client('sns', region_name=region)

          def lambda_handler(event, context):
            now = datetime.datetime.now()
            now_str = now.strftime('%Y-%m-%d %H:%M:%S.%f')
            unix_time = str(time.mktime(now.timetuple()))
            print(now_str, unix_time)

            response = client.publish(
              TopicArn=topic,
              Subject=now_str,
              Message=unix_time
              )
            print(response)
      FunctionName: !Sub "${Prefix}-function-02"
      Handler: !Ref Handler
      Runtime: !Ref Runtime
      Role: !GetAtt FunctionRole2.Arn

  FunctionRole2:
    Type: AWS::IAM::Role
    DeletionPolicy: Delete
    Properties:
      AssumeRolePolicyDocument:
        Version: 2012-10-17
        Statement:
          - Effect: Allow
            Action: sts:AssumeRole
            Principal:
              Service:
                - lambda.amazonaws.com
      ManagedPolicyArns:
        - arn:aws:iam::aws:policy/service-role/AWSLambdaBasicExecutionRole
      Policies:
        - PolicyName: SNSPublishPolicy
          PolicyDocument:
            Version: 2012-10-17
            Statement:
              - Effect: Allow
                Action:
                  - sns:Publish
                Resource:
                  - !Ref TopicArn
Code language: YAML (yaml)

Lambda関数からSNSにメッセージをパブリッシュします。
詳細は冒頭でご紹介したページをご確認ください。

なおCloudFormationでLambda関数を作成する方法については、以下のページをご確認ください。

今回はインラインで実行するコードを記載します。

SNSサブスクライバーにLambda関数を指定する

Resources:
  Topic:
    Type: AWS::SNS::Topic
    Properties:
      Subscription:
        - Endpoint: !Ref FunctionArn1
          Protocol: lambda
      TopicName: !Ref Prefix
Code language: YAML (yaml)

Subscriptionプロパティでサブスクライバーを設定します。
今回はLambda関数でSNSメッセージを受け取りますから、Endpointプロパティに関数のARNを、Protocolプロパティに「lambda」を指定します。

サブスクライバーにLambda関数を指定する場合、内部的にはSNSが関数を呼び出すという挙動になります。
そのためSNSに関数を呼び出す権限を与える必要があります。

Resources:
  SNSPermission:
    Type: AWS::Lambda::Permission
    Properties:
      FunctionName: !Ref Function1
      Action: lambda:InvokeFunction
      Principal: sns.amazonaws.com
      SourceArn: !Ref Topic
Code language: YAML (yaml)

以下のAWS公式ページを参考にして設定しました。

https://aws.amazon.com/jp/premiumsupport/knowledge-center/lambda-subscribe-sns-topic-same-account/

Lambda関数でSNSメッセージを受け取る

Resources:
  Function1:
    Type: AWS::Lambda::Function
    Properties:
      Architectures:
        - !Ref Architecture
      Code:
        ZipFile: |
          def lambda_handler(event, context):
            subject = event['Records'][0]['Sns']['Subject']
            message = event['Records'][0]['Sns']['Message']
            print(subject, message)

            unix_time = int(float(message))

            if (unix_time % 2) == 0:
              print('{int} is even !'.format(int=unix_time))
            else:
              print('{int} is odd !'.format(int=unix_time))
      FunctionName: !Sub "${Prefix}-function-01"
      Handler: !Ref Handler
      Runtime: !Ref Runtime
      Role: !GetAtt FunctionRole1.Arn
Code language: YAML (yaml)