fix connection pool check_close_callback

snower · Oct 16, 2015 · c18a32f · c18a32f
1 parent 7d52c2d
commit c18a32f
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tormysql/pool.py b/tormysql/pool.py
@@ -167,7 +167,7 @@ def check_close_callback(self):
         self._check_close_callback = False
         for connection in list(self._connections):
             if connection.open:
-                connection._handle_read()
+                connection.socket._handle_read()
 
     def connection_close_callback(self, connection):
         try:
@@ -180,10 +180,9 @@ def connection_close_callback(self, connection):
             except ValueError:
                 logging.warning("Close unknown Connection %s", connection)
         if self._close_future and not self._used_connections and not self._connections:
-            def do_close():
-                self._close_future.set_result(None)
-                self._close_future = None
-            IOLoop.current().add_callback(do_close)
+            close_future = self._close_future
+            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)