Merge pull request #47 from wooparadog/fix-timeout

Should catch socket.timeout exception
Thriftpy · Oct 23, 2017 · 7584472 · 7584472
2 parents 89e5fce + e2a5d35
commit 7584472
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/gunicorn_thrift/sync_worker.py b/gunicorn_thrift/sync_worker.py
@@ -48,6 +48,8 @@ def handle(self, listener, client, addr):
                     self.notify()
             except TTransport.TTransportException:
                 pass
+        except socket.timeout:
+            self.log.warning('Client timeout: %r', addr)
         except socket.error as e:
             if e.args[0] == errno.ECONNRESET:
                 self.log.debug(e)

diff --git a/gunicorn_thrift/thriftpy_sync_worker.py b/gunicorn_thrift/thriftpy_sync_worker.py
@@ -68,6 +68,8 @@ def handle(self, listener, client, addr):
                     self.notify()
             except TTransportException:
                 pass
+        except socket.timeout:
+            self.log.warning('Client timeout: %r', addr)
         except socket.error as e:
             if e.args[0] == errno.ECONNRESET:
                 self.log.debug(e)