extra routing info

Author: acn-ericlaw

mercury/platform.py

@@ -239,6 +239,8 @@ def handle_event(self, event, instance):
             response = EventEnvelope().set_to(reply_to)
             if not error_code:
                 response.set_exec_time((end - begin) * 1000)
+            if 'extra' in event:
+                response.set_extra(event['extra'])
             if 'cid' in event:
                 response.set_correlation_id(event['cid'])
             if isinstance(result, EventEnvelope):

mercury/system/models.py

@@ -44,6 +44,7 @@ def __init__(self):
         self.to = None
         self.sender = None
         self.reply_to = None
+        self.extra = None
         self.correlation_id = None
         self.broadcast = False
         self.exec_time = -1.0
@@ -93,6 +94,16 @@ def set_reply_to(self, reply_to: str, me=False):
     def get_reply_to(self):
         return self.reply_to
 
+    def set_extra(self, extra: str):
+        if isinstance(extra, str):
+            self.extra = extra
+        else:
+            raise ValueError('extra must be str')
+        return self
+
+    def get_extra(self):
+        return self.extra
+
     def set_correlation_id(self, correlation_id: str):
         self.correlation_id = correlation_id if isinstance(correlation_id, str) else str(correlation_id)
         return self
@@ -153,6 +164,8 @@ def to_map(self):
             result['body'] = self.body
         if self.reply_to:
             result['reply_to'] = self.reply_to
+        if self.extra:
+            result['extra'] = self.extra
         if self.correlation_id:
             result['cid'] = self.correlation_id
         if self.broadcast:
@@ -178,6 +191,8 @@ def from_map(self, data: dict):
             self.body = data['body']
         if 'reply_to' in data and isinstance(data['reply_to'], str):
             self.reply_to = data['reply_to']
+        if 'extra' in data and isinstance(data['extra'], str):
+            self.extra = data['extra']
         if 'cid' in data:
             self.correlation_id = data['cid']
         if 'status' in data: