Merge pull request #35 from mokshaproject/stomp-headers

Pass STOMP headers along to consumers.
mokshaproject · Apr 28, 2016 · 433df3e · 433df3e
2 parents c85a0e7 + 53cfe1c
commit 433df3e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/moksha.hub/moksha/hub/hub.py b/moksha.hub/moksha/hub/hub.py
@@ -190,7 +190,8 @@ def consume_amqp_message(self, message):
     def consume_stomp_message(self, message):
         from moksha.hub.reactor import reactor
 
-        topic = message['headers'].get('destination')
+        headers = message['headers']
+        topic = headers.get('destination')
         if not topic:
             log.debug("Got message without a topic: %r" % message)
             return
@@ -204,8 +205,9 @@ def consume_stomp_message(self, message):
             body = message['body']
 
         # feed all of our consumers
+        envelope = {'body': body, 'topic': topic, 'headers': headers}
         for callback in self.topics.get(topic, []):
-            reactor.callInThread(callback, {'body': body, 'topic': topic})
+            reactor.callInThread(callback, envelope)
 
 
 class CentralMokshaHub(MokshaHub):