From e0c5cab5f94032da3d770091f93a7b7643965059 Mon Sep 17 00:00:00 2001
From: Oleg Ovcharuk <vgvoleg@gmail.com>
Date: Fri, 1 Nov 2024 14:09:12 +0300
Subject: [PATCH] Add an ability to pass custom event loop to QuerySessionPool

---
 ydb/aio/query/pool.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ydb/aio/query/pool.py b/ydb/aio/query/pool.py
index db01adce..456896db 100644
--- a/ydb/aio/query/pool.py
+++ b/ydb/aio/query/pool.py
@@ -29,6 +29,7 @@ def __init__(
         size: int = 100,
         *,
         query_client_settings: Optional[QueryClientSettings] = None,
+        loop: Optional[asyncio.AbstractEventLoop] = None,
     ):
         """
         :param driver: A driver instance
@@ -42,7 +43,7 @@ def __init__(
         self._queue = asyncio.Queue()
         self._current_size = 0
         self._waiters = 0
-        self._loop = asyncio.get_running_loop()
+        self._loop = asyncio.get_running_loop() if loop is None else loop
         self._query_client_settings = query_client_settings
 
     async def _create_new_session(self):