refactor dead session handle logic

Author: vgvoleg

ydb/aio/query/pool.py

@@ -50,8 +50,12 @@ async def acquire(self, timeout: float) -> QuerySessionAsync:
 
         try:
             _, session = self._queue.get_nowait()
-            logger.debug(f"Acquired active session from queue: {session._state.session_id}")
-            return session if session._state.attached else await self._create_new_session()
+            if session._state.attached:
+                logger.debug(f"Acquired active session from queue: {session._state.session_id}")
+                return session
+            else:
+                self._current_size -= 1
+                logger.debug(f"Acquired dead session from queue: {session._state.session_id}")
         except asyncio.QueueEmpty:
             pass
 

ydb/query/pool.py

@@ -52,8 +52,12 @@ def acquire(self, timeout: float) -> QuerySessionSync:
 
             try:
                 _, session = self._queue.get_nowait()
-                logger.debug(f"Acquired active session from queue: {session._state.session_id}")
-                return session if session._state.attached else self._create_new_session()
+                if session._state.attached:
+                    logger.debug(f"Acquired active session from queue: {session._state.session_id}")
+                    return session
+                else:
+                    self._current_size -= 1
+                    logger.debug(f"Acquired dead session from queue: {session._state.session_id}")
             except queue.Empty:
                 pass