Skip to content

Commit f5e568e

Browse files
committed
close pool on driver close
1 parent 2480312 commit f5e568e

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

ydb/aio/driver.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,3 +59,7 @@ def __init__(
5959
self.scheme_client = scheme.SchemeClient(self)
6060
self.table_client = table.TableClient(self, config.table_client_settings)
6161
self.topic_client = topic.TopicClientAsyncIO(self, config.topic_client_settings)
62+
63+
async def stop(self, timeout=10):
64+
await self.table_client._pool.stop(timeout)
65+
await super().stop(timeout=timeout)

ydb/driver.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,3 +282,7 @@ def __init__(
282282
self.scheme_client = scheme.SchemeClient(self)
283283
self.table_client = table.TableClient(self, driver_config.table_client_settings)
284284
self.topic_client = topic.TopicClient(self, driver_config.topic_client_settings)
285+
286+
def stop(self):
287+
self.table_client._pool.stop()
288+
super().stop()

0 commit comments

Comments
 (0)