bugfix(aws-sqs-signal-consumer-plugin): fixes logic (#5589)

Netflix · Dec 10, 2024 · c85f3b0 · c85f3b0
1 parent a8b57b6
commit c85f3b0
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/dispatch/plugins/dispatch_aws/plugin.py b/src/dispatch/plugins/dispatch_aws/plugin.py
@@ -74,15 +74,16 @@ def consume(self, db_session: Session, project: Project) -> None:
 
             entries: list[SqsEntries] = []
             for message in response["Messages"]:
-                message_attributes = message.get("MessageAttributes", {})
-                message_body = message["Body"]
+                message_body = json.loads(message["Body"])
+                message_body_message = message_body.get("Message")
+                message_attributes = message_body.get("MessageAttributes", {})
 
-                if message_attributes.get("compressed", {}).get("StringValue") == "zlib":
+                if message_attributes.get("compressed", {}).get("Value") == "zlib":
                     # Message is compressed, decompress it
-                    message_body = decompress_json(message_body)
-
-                message_body = json.loads(message_body)
-                signal_data = json.loads(message_body["Message"])
+                    message_body_message = decompress_json(message_body_message)
+                    signal_data = json.loads(message_body_message)
+                else:
+                    signal_data = message_body_message
 
                 try:
                     signal_instance_in = SignalInstanceCreate(