fix close connection when mysql server is down

Author: snower

tormysql/connections.py

@@ -154,6 +154,9 @@ def _read_bytes(self, num_bytes):
             self._rbuffer = StringIO(last_buf + b''.join(self._rfile._read_buffer))
             self._rfile._read_buffer.clear()
             self._rfile._read_buffer_size = 0
+            if not self._rfile._state & self._rfile.io_loop.READ:
+                self._rfile._state |= self._rfile.io_loop.READ
+                self._rfile.io_loop.update_handler(self._rfile.fileno(), self._rfile._state)
             return self._rbuffer.read(num_bytes)
 
         child_gr = greenlet.getcurrent()

tormysql/pool.py

@@ -66,7 +66,6 @@ def __init__(self, *args, **kwargs):
         self._closed = False
         self._close_future = None
         self._check_idle_callback = False
-        self._check_close_callback = False
 
     @property
     def closed(self):
@@ -157,12 +156,6 @@ def close_connection(self, connection):
         except ValueError:
             raise ConnectionUsedError()
 
-    def check_close_callback(self):
-        self._check_close_callback = False
-        for connection in list(self._connections):
-            if connection.open:
-                connection.socket._handle_read()
-
     def connection_close_callback(self, connection):
         try:
             del self._used_connections[id(connection)]
@@ -178,10 +171,6 @@ def connection_close_callback(self, connection):
             IOLoop.current().add_callback(close_future.set_result, None)
             self._close_future = None
 
-        if not self._check_close_callback:
-            IOLoop.current().add_callback(self.check_close_callback)
-            self._check_close_callback = True
-
     def close(self):
         if self._closed:
             raise ConnectionPoolClosedError()