Merge pull request #179 from air-upc/add_handshake_logic

add handshake logic in tracker

thriftpy/contrib/tracking/__init__.py

@@ -69,6 +69,7 @@ def _negotiation(self):
         self._oprot.write_message_begin(track_method, TMessageType.CALL,
                                         self._seqid)
         args = track_thrift.UpgradeArgs()
+        self.tracker.init_handshake_info(args)
         args.write(self._oprot)
         self._oprot.write_message_end()
         self._oprot.trans.flush()
@@ -148,6 +149,7 @@ def _try_upgrade(self, iprot):
 
             args = track_thrift.UpgradeArgs()
             args.read(iprot)
+            self.tracker.handle_handshake_info(args)
             result = track_thrift.UpgradeReply()
             result.oneway = False
 

thriftpy/contrib/tracking/tracker.py

@@ -101,6 +101,12 @@ def get_request_id(self):
             return ctx.header.request_id
         return str(uuid.uuid4())
 
+    def init_handshake_info(self, handshake_obj):
+        pass
+
+    def handle_handshake_info(self, handshake_obj):
+        pass
+
 
 class ConsoleTracker(TrackerBase):
     def record(self, header, exception):

thriftpy/contrib/tracking/tracking.thrift

@@ -11,4 +11,6 @@ struct RequestHeader {
  * This is the struct that a successful upgrade will reply with.
  */
 struct UpgradeReply {}
-struct UpgradeArgs {}
+struct UpgradeArgs {
+    1: string app_id
+}