Remove connection from pool before waiting for results

Author: vgvoleg

ydb/aio/connection.py

@@ -239,12 +239,12 @@ async def close(self, grace: float = None):
 
         self.closing = True
 
-        if self.calls:
-            await asyncio.wait(self.calls.values(), timeout=grace)
-
         for callback in self._cleanup_callbacks:
             callback(self)
 
+        if self.calls:
+            await asyncio.wait(self.calls.values(), timeout=grace)
+
         await self.destroy()
 
     async def __aenter__(self):