From 3708f10bf033bfd3ea6ab48743f3cc127fed0839 Mon Sep 17 00:00:00 2001
From: mckadesorensen <dmsorensen@alaska.edu>
Date: Fri, 3 May 2024 09:55:50 -0800
Subject: [PATCH] feat: add MessageAttribute

---
 src/granule_to_sns.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/granule_to_sns.py b/src/granule_to_sns.py
index f1310e8..f30e297 100644
--- a/src/granule_to_sns.py
+++ b/src/granule_to_sns.py
@@ -38,6 +38,12 @@ def granule_to_sns(event: dict, _) -> dict:
         client.publish(
             TopicArn=sns_topic_arn,
             Message=message,
+            MessageAttributes={
+                "collection": {
+                    "DataType": "String",
+                    "StringValue": granule["dataType"],
+                },
+            },
         )
 
     return event