File tree Expand file tree Collapse file tree 2 files changed +12
-4
lines changed Expand file tree Collapse file tree 2 files changed +12
-4
lines changed Original file line number Diff line number Diff line change @@ -50,8 +50,12 @@ async def acquire(self, timeout: float) -> QuerySessionAsync:
50
50
51
51
try :
52
52
_ , session = self ._queue .get_nowait ()
53
- logger .debug (f"Acquired active session from queue: { session ._state .session_id } " )
54
- return session if session ._state .attached else await self ._create_new_session ()
53
+ if session ._state .attached :
54
+ logger .debug (f"Acquired active session from queue: { session ._state .session_id } " )
55
+ return session
56
+ else :
57
+ self ._current_size -= 1
58
+ logger .debug (f"Acquired dead session from queue: { session ._state .session_id } " )
55
59
except asyncio .QueueEmpty :
56
60
pass
57
61
Original file line number Diff line number Diff line change @@ -52,8 +52,12 @@ def acquire(self, timeout: float) -> QuerySessionSync:
52
52
53
53
try :
54
54
_ , session = self ._queue .get_nowait ()
55
- logger .debug (f"Acquired active session from queue: { session ._state .session_id } " )
56
- return session if session ._state .attached else self ._create_new_session ()
55
+ if session ._state .attached :
56
+ logger .debug (f"Acquired active session from queue: { session ._state .session_id } " )
57
+ return session
58
+ else :
59
+ self ._current_size -= 1
60
+ logger .debug (f"Acquired dead session from queue: { session ._state .session_id } " )
57
61
except queue .Empty :
58
62
pass
59
63
You can’t perform that action at this time.
0 commit comments