review fixes

Author: vgvoleg

ydb/aio/query/pool.py

@@ -1,6 +1,5 @@
 import asyncio
 import logging
-import functools
 from typing import (
     Callable,
     Optional,
@@ -61,7 +60,8 @@ async def acquire(self) -> QuerySession:
             done, _ = await asyncio.wait((queue_get, task_stop), return_when=asyncio.FIRST_COMPLETED)
             if task_stop in done:
                 queue_get.cancel()
-                return await self._create_new_session()
+                raise RuntimeError("An attempt to take session from closed session pool.")
+
             task_stop.cancel()
             session = queue_get.result()
 
@@ -163,7 +163,7 @@ def __del__(self):
         if self._should_stop.is_set() or self._loop.is_closed():
             return
 
-        self._loop.call_soon(functools.partial(self.stop))
+        self._loop.call_soon(self.stop)
 
 
 class SimpleQuerySessionCheckoutAsync:

ydb/query/pool.py

@@ -47,6 +47,8 @@ def _create_new_session(self, timeout: Optional[float]):
         return session
 
     def acquire(self, timeout: Optional[float] = None) -> QuerySession:
+        start = time.monotonic()
+
         lock_acquire_timeout = timeout if timeout is not None else -1
         acquired = self._lock.acquire(timeout=lock_acquire_timeout)
         try:
@@ -60,6 +62,9 @@ def acquire(self, timeout: Optional[float] = None) -> QuerySession:
             except queue.Empty:
                 pass
 
+            finish = time.monotonic()
+            timeout = timeout - (finish - start) if timeout is not None else None
+
             start = time.monotonic()
             if session is None and self._current_size == self._size:
                 try: