Go言語でAWS SNSにメッセージを送信する

概要

Amazon Simple Notification Service(SNS)で、Go言語のSDKを使ってトピックに対してメッセージを送信します。

環境

• Go : 1.14.3
• Mac OS X : 10.15.4
• AWS SDK for Go

シンプルな実装

シンプルなメッセージ送信

package main

import (
	"fmt"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/sns"
)

// TopicARNとRegionは環境に合わせます
const (
	TopicARN  = "arn:aws:sns:XXXXXXXXXXXXXXXXX"
	AwsRegion = "XXXXXX"
)

func main() {
    // SNSクライアントの作成
	mySession := session.Must(session.NewSession())
	svc := sns.New(mySession, aws.NewConfig().WithRegion(AwsRegion))

	message := "This is sample message."

    // メッセージを送信するための構造体を作成
	inputPublish := &sns.PublishInput{
		Message:  aws.String(message),
		TopicArn: aws.String(TopicARN),
	}

    // メッセージの送信(Publish)
	MessageId, err := svc.Publish(inputPublish)
	if err != nil {
		fmt.Println("Publish Error: ", err)
	}

	fmt.Println(MessageId)

}

シンプルにメッセージを送る場合の実装です。登録されたサブスクリプションに対してメッセージが送信されます。例として、SQSをサブスクリプションにした場合、以下の通り確認できます。

サブスクリプションごとにメッセージを変える場合

サブスクライブするプロトコルごとのメッセージ

package main

import (
	"encoding/json"
	"fmt"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/sns"
)

const (
	TopicARN  = "arn:aws:sns:XXXXXXXXXXXXXXXXX"
	AwsRegion = "XXXXXX"
)

func main() {
	mySession := session.Must(session.NewSession())
	svc := sns.New(mySession, aws.NewConfig().WithRegion(AwsRegion))

	message := "This is sample message."

    // サブスクリプションのプロトコルごとにメッセージを指定
	messageJson := map[string]string{
		"default": message,
		"sqs": "This is sample message for sqs.",
	}
    // メッセージ構造体はJSON文字列にする
	bytes, err := json.Marshal(messageJson)
	if err != nil {
		fmt.Println("JSON marshal Error: ", err)
	}
	messageForSQS := string(bytes)

	inputPublish := &sns.PublishInput{
		Message:  aws.String(messageForSQS),
		MessageStructure: aws.String("json"),    // MessageStructureにjsonを指定
		TopicArn: aws.String(TopicARN),
	}

	MessageId, err := svc.Publish(inputPublish)
	if err != nil {
		fmt.Println("Publish Error: ", err)
	}

	fmt.Println(MessageId)

}

このソースの実行結果は以下の通りです。SQS用に設定したメッセージが送信されています。

このようにサブスクリプションのプロトコルごとに異なるメッセージを送る場合、MessageStructureにjsonを設定します。また、メッセージはJSON文字列として、トップレベルにプロトコル(今回の例ではsqs)を指定します。トップレベルにdefaultは必須です。
プロトコルとして指定する値はCLIのリファレンスを参照。
subscribe - Options

なお、メッセージ属性を設定する場合はMessageAttributesを使います。詳細は以下の公式ドキュメントを参照。
type PublishInput

参考

• AWS SDK for Go API Reference
• AWS SNSで複数プラットフォーム上のアプリへのメッセージを送るときにはJSON文字列を渡さないといけない