Fix SNS publish for non-default message types

alexandria-oss · May 1, 2023 · a50e48d · a50e48d
1 parent df1e86d
commit a50e48d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/driver/amazon/sns/writer.go b/driver/amazon/sns/writer.go
@@ -40,8 +40,14 @@ func (w Writer) write(ctx context.Context, stream string, msgBatch []streams.Mes
 	isTopicFIFO := strings.HasSuffix(stream, ".fifo")
 	batchBuf := make([]types.PublishBatchRequestEntry, len(msgBatch))
 	for i, msg := range msgBatch {
+		msgStr := string(msg.Data)
 		msgJSON, err := jsoniter.Marshal(message{
-			Default: string(msg.Data),
+			Default:   msgStr,
+			Email:     msgStr,
+			EmailJSON: msgStr,
+			HTTP:      msgStr,
+			HTTPS:     msgStr,
+			SQS:       msgStr,
 		})
 		if err != nil {
 			return err