now able to set client_timeout in make_server, fixes #204

expose client_timeout settings in thriftpy.rpc helper func

thriftpy/rpc.py

@@ -47,7 +47,7 @@ def make_server(service, handler,
                 host="localhost", port=9090, unix_socket=None,
                 proto_factory=TBinaryProtocolFactory(),
                 trans_factory=TBufferedTransportFactory(),
-                certfile=None):
+                client_timeout=3000, certfile=None):
     processor = TProcessor(service, handler)
 
     if unix_socket:
@@ -56,10 +56,12 @@ def make_server(service, handler,
             warnings.warn("SSL only works with host:port, not unix_socket.")
     elif host and port:
         if certfile:
-            server_socket = TSSLServerSocket(host=host, port=port,
-                                             certfile=certfile)
+            server_socket = TSSLServerSocket(
+                host=host, port=port, client_timeout=client_timeout,
+                certfile=certfile)
         else:
-            server_socket = TServerSocket(host=host, port=port)
+            server_socket = TServerSocket(
+                host=host, port=port, client_timeout=client_timeout)
     else:
         raise ValueError("Either host/port or unix_socket must be provided.")
 

thriftpy/transport/socket.py

@@ -204,7 +204,8 @@ def listen(self):
 
     def accept(self):
         client, _ = self.sock.accept()
-        client.settimeout(self.client_timeout)
+        if self.client_timeout:
+            client.settimeout(self.client_timeout)
         return TSocket(sock=client)
 
     def close(self):

thriftpy/transport/sslsocket.py

@@ -118,5 +118,6 @@ def accept(self):
             sock.close()
             raise
         else:
-            ssl_sock.settimeout(self.client_timeout)
+            if self.client_timeout:
+                ssl_sock.settimeout(self.client_timeout)
             return TSocket(sock=ssl_sock)