make pool init lazy

Author: vgvoleg

ydb/aio/driver.py

@@ -61,5 +61,5 @@ def __init__(
         self.topic_client = topic.TopicClientAsyncIO(self, config.topic_client_settings)
 
     async def stop(self, timeout=10):
-        await self.table_client._pool.stop(timeout=timeout)
+        await self.table_client._stop_pool_if_needed(timeout=timeout)
         await super().stop(timeout=timeout)

ydb/aio/table.py

@@ -151,11 +151,11 @@ class TableClient(BaseTableClient):
     def __init__(self, driver, table_client_settings=None):
         # type:(ydb.Driver, ydb.TableClientSettings) -> None
         super().__init__(driver=driver, table_client_settings=table_client_settings)
-        self._pool: SessionPool = SessionPool(self._driver, 10)
+        self._pool: Optional[SessionPool] = None
 
     def __del__(self):
         if not self._pool._terminating:
-            asyncio.get_running_loop.call_soon(self._pool.stop)
+            asyncio.get_running_loop.call_soon(self._stop_pool_if_needed)
 
     def session(self):
         return Session(self._driver, self._table_client_settings)
@@ -176,6 +176,14 @@ async def scan_query(self, query, parameters=None, settings=None):  # pylint: di
             lambda resp: _wrap_scan_query_response(resp, self._table_client_settings),
         )
 
+    def _init_pool_if_needed(self):
+        if self._pool is None:
+            self._pool = SessionPool(self._driver, 10)
+
+    async def _stop_pool_if_needed(self, timeout=10):
+        if self._pool is not None:
+            await self._pool.stop(timeout=timeout)
+
     async def create_table(
         self,
         path: str,
@@ -192,6 +200,8 @@ async def create_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.create_table(path=path, table_description=table_description, settings=settings)
 
@@ -211,6 +221,8 @@ async def drop_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.drop_table(path=path, settings=settings)
 
@@ -257,6 +269,8 @@ async def alter_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.alter_table(
                 path=path,
@@ -293,6 +307,8 @@ async def describe_table(
         :return: TableSchemeEntry or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.describe_table(path=path, settings=settings)
 
@@ -314,6 +330,8 @@ async def copy_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.copy_table(
                 source_path=source_path,
@@ -337,6 +355,8 @@ async def copy_tables(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.copy_tables(source_destination_pairs=source_destination_pairs, settings=settings)
 
@@ -356,6 +376,8 @@ async def rename_tables(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         async def callee(session: Session):
             return await session.rename_tables(rename_items=rename_items, settings=settings)
 

ydb/driver.py

@@ -284,5 +284,5 @@ def __init__(
         self.topic_client = topic.TopicClient(self, driver_config.topic_client_settings)
 
     def stop(self, timeout=10):
-        self.table_client._pool.stop(timeout=timeout)
+        self.table_client._stop_pool_if_needed(timeout=timeout)
         super().stop(timeout=timeout)

ydb/table.py

@@ -1205,10 +1205,10 @@ class TableClient(BaseTableClient):
     def __init__(self, driver, table_client_settings=None):
         # type:(ydb.Driver, ydb.TableClientSettings) -> None
         super().__init__(driver=driver, table_client_settings=table_client_settings)
-        self._pool: SessionPool = SessionPool(self._driver, 10)
+        self._pool: Optional[SessionPool] = None
 
     def __del__(self):
-        self._pool.stop()
+        self._stop_pool_if_needed()
 
     def async_scan_query(self, query, parameters=None, settings=None):
         # type: (ydb.ScanQuery, tuple, ydb.BaseRequestSettings) -> ydb.AsyncResponseIterator
@@ -1236,6 +1236,14 @@ def async_bulk_upsert(self, table_path, rows, column_types, settings=None):
             (),
         )
 
+    def _init_pool_if_needed(self):
+        if self._pool is None:
+            self._pool = SessionPool(self._driver, 10)
+
+    def _stop_pool_if_needed(self, timeout=10):
+        if self._pool is not None:
+            self._pool.stop(timeout=timeout)
+
     def create_table(
         self,
         path: str,
@@ -1252,6 +1260,8 @@ def create_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.create_table(path=path, table_description=table_description, settings=settings)
 
@@ -1271,6 +1281,8 @@ def drop_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.drop_table(path=path, settings=settings)
 
@@ -1317,6 +1329,8 @@ def alter_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.alter_table(
                 path=path,
@@ -1353,6 +1367,8 @@ def describe_table(
         :return: TableSchemeEntry or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.describe_table(path=path, settings=settings)
 
@@ -1374,6 +1390,8 @@ def copy_table(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.copy_table(
                 source_path=source_path,
@@ -1397,6 +1415,8 @@ def copy_tables(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.copy_tables(source_destination_pairs=source_destination_pairs, settings=settings)
 
@@ -1416,6 +1436,8 @@ def rename_tables(
         :return: Operation or YDB error otherwise.
         """
 
+        self._init_pool_if_needed()
+
         def callee(session: Session):
             return session.rename_tables(rename_items=rename_items, settings=settings)