add back set_timeout api for backward compat

Thriftpy · Jan 21, 2016 · 70bfa0b · 70bfa0b
1 parent 880695c
commit 70bfa0b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/thriftpy/transport/socket.py b/thriftpy/transport/socket.py
@@ -71,6 +71,14 @@ def _init_sock(self):
     def set_handle(self, sock):
         self.sock = sock
 
+    def set_timeout(self, ms):
+        """Backward compat api, will bind the timeout to both connect_timeout
+        and socket_timeout.
+        """
+        if ms and ms > 0:
+            self.socket_timeout = ms / 1000
+            self.connect_timeout = self.socket_timeout
+
     def is_open(self):
         return bool(self.sock)