set timeout on socket

CurleySamuel · Jan 14, 2016 · aa3d11e · aa3d11e
1 parent 5ad1d3b
commit aa3d11e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/region/client.py b/region/client.py
@@ -258,6 +258,7 @@ def NewClient(host, port, pool_size):
             s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
             s.connect((host, int(port)))
             _send_hello(s)
+            s.settimeout(2)
             c.sock_pool.append(s)
             c.read_lock_pool.append(Lock())
             c.write_lock_pool.append(Lock())