rename interfaces

Author: vgvoleg

examples/basic_example_v2/basic_example.py

@@ -140,7 +140,7 @@ def select_with_parameters(pool: ydb.QuerySessionPool, series_id, season_id, epi
 # calls instead to avoid additional hops to YDB cluster and allow more efficient
 # execution of queries.
 def explicit_transaction_control(pool: ydb.QuerySessionPool, series_id, season_id, episode_id):
-    def callee(session: ydb.QuerySessionSync):
+    def callee(session: ydb.QuerySession):
         query = """
         DECLARE $seriesId AS Int64;
         DECLARE $seasonId AS Int64;
@@ -175,7 +175,7 @@ def callee(session: ydb.QuerySessionSync):
 
 
 def huge_select(pool: ydb.QuerySessionPool):
-    def callee(session: ydb.QuerySessionSync):
+    def callee(session: ydb.QuerySession):
         query = """SELECT * from episodes;"""
 
         with session.transaction().execute(

examples/basic_example_v2/basic_example_async.py

@@ -58,7 +58,7 @@
 """
 
 
-async def fill_tables_with_data(pool: ydb.aio.QuerySessionPoolAsync):
+async def fill_tables_with_data(pool: ydb.aio.QuerySessionPool):
     print("\nFilling tables with data...")
     await pool.execute_with_retries(
         FillDataQuery,
@@ -70,7 +70,7 @@ async def fill_tables_with_data(pool: ydb.aio.QuerySessionPoolAsync):
     )
 
 
-async def select_simple(pool: ydb.aio.QuerySessionPoolAsync):
+async def select_simple(pool: ydb.aio.QuerySessionPool):
     print("\nCheck series table...")
     result_sets = await pool.execute_with_retries(
         """
@@ -96,7 +96,7 @@ async def select_simple(pool: ydb.aio.QuerySessionPoolAsync):
     return first_set
 
 
-async def upsert_simple(pool: ydb.aio.QuerySessionPoolAsync):
+async def upsert_simple(pool: ydb.aio.QuerySessionPool):
     print("\nPerforming UPSERT into episodes...")
 
     await pool.execute_with_retries(
@@ -106,7 +106,7 @@ async def upsert_simple(pool: ydb.aio.QuerySessionPoolAsync):
     )
 
 
-async def select_with_parameters(pool: ydb.aio.QuerySessionPoolAsync, series_id, season_id, episode_id):
+async def select_with_parameters(pool: ydb.aio.QuerySessionPool, series_id, season_id, episode_id):
     result_sets = await pool.execute_with_retries(
         """
         DECLARE $seriesId AS Int64;
@@ -138,8 +138,8 @@ async def select_with_parameters(pool: ydb.aio.QuerySessionPoolAsync, series_id,
 # In most cases it's better to use transaction control settings in session.transaction
 # calls instead to avoid additional hops to YDB cluster and allow more efficient
 # execution of queries.
-async def explicit_transaction_control(pool: ydb.aio.QuerySessionPoolAsync, series_id, season_id, episode_id):
-    async def callee(session: ydb.aio.QuerySessionAsync):
+async def explicit_transaction_control(pool: ydb.aio.QuerySessionPool, series_id, season_id, episode_id):
+    async def callee(session: ydb.aio.QuerySession):
         query = """
         DECLARE $seriesId AS Int64;
         DECLARE $seasonId AS Int64;
@@ -173,8 +173,8 @@ async def callee(session: ydb.aio.QuerySessionAsync):
     return await pool.retry_operation_async(callee)
 
 
-async def huge_select(pool: ydb.aio.QuerySessionPoolAsync):
-    async def callee(session: ydb.aio.QuerySessionAsync):
+async def huge_select(pool: ydb.aio.QuerySessionPool):
+    async def callee(session: ydb.aio.QuerySession):
         query = """SELECT * from episodes;"""
 
         async with await session.transaction().execute(
@@ -189,12 +189,12 @@ async def callee(session: ydb.aio.QuerySessionAsync):
     return await pool.retry_operation_async(callee)
 
 
-async def drop_tables(pool: ydb.aio.QuerySessionPoolAsync):
+async def drop_tables(pool: ydb.aio.QuerySessionPool):
     print("\nCleaning up existing tables...")
     await pool.execute_with_retries(DropTablesQuery)
 
 
-async def create_tables(pool: ydb.aio.QuerySessionPoolAsync):
+async def create_tables(pool: ydb.aio.QuerySessionPool):
     print("\nCreating table series...")
     await pool.execute_with_retries(
         """
@@ -246,7 +246,7 @@ async def run(endpoint, database):
     ) as driver:
         await driver.wait(timeout=5, fail_fast=True)
 
-        async with ydb.aio.QuerySessionPoolAsync(driver) as pool:
+        async with ydb.aio.QuerySessionPool(driver) as pool:
             await drop_tables(pool)
 
             await create_tables(pool)

examples/query-service/basic_example.py

@@ -82,7 +82,7 @@ def callee(session):
 
     pool.retry_operation_sync(callee)
 
-    def callee(session: ydb.QuerySessionSync):
+    def callee(session: ydb.QuerySession):
         query_print = """select $a"""
 
         print("=" * 50)

examples/query-service/basic_example_asyncio.py

@@ -15,7 +15,7 @@ async def main():
     except TimeoutError:
         raise RuntimeError("Connect failed to YDB")
 
-    pool = ydb.aio.QuerySessionPoolAsync(driver)
+    pool = ydb.aio.QuerySessionPool(driver)
 
     print("=" * 50)
     print("DELETE TABLE IF EXISTS")
@@ -83,7 +83,7 @@ async def callee(session):
 
     await pool.retry_operation_async(callee)
 
-    async def callee(session: ydb.aio.QuerySessionAsync):
+    async def callee(session: ydb.aio.QuerySession):
         query_print = """select $a"""
 
         print("=" * 50)

tests/aio/query/conftest.py

@@ -1,11 +1,11 @@
 import pytest
-from ydb.aio.query.session import QuerySessionAsync
-from ydb.aio.query.pool import QuerySessionPoolAsync
+from ydb.aio.query.session import QuerySession
+from ydb.aio.query.pool import QuerySessionPool
 
 
 @pytest.fixture
 async def session(driver):
-    session = QuerySessionAsync(driver)
+    session = QuerySession(driver)
 
     yield session
 
@@ -30,5 +30,5 @@ async def tx(session):
 
 @pytest.fixture
 async def pool(driver):
-    async with QuerySessionPoolAsync(driver) as pool:
+    async with QuerySessionPool(driver) as pool:
         yield pool

tests/aio/query/test_query_session.py

@@ -1,22 +1,22 @@
 import pytest
-from ydb.aio.query.session import QuerySessionAsync
+from ydb.aio.query.session import QuerySession
 
 
-def _check_session_state_empty(session: QuerySessionAsync):
+def _check_session_state_empty(session: QuerySession):
     assert session._state.session_id is None
     assert session._state.node_id is None
     assert not session._state.attached
 
 
-def _check_session_state_full(session: QuerySessionAsync):
+def _check_session_state_full(session: QuerySession):
     assert session._state.session_id is not None
     assert session._state.node_id is not None
     assert session._state.attached
 
 
 class TestAsyncQuerySession:
     @pytest.mark.asyncio
-    async def test_session_normal_lifecycle(self, session: QuerySessionAsync):
+    async def test_session_normal_lifecycle(self, session: QuerySession):
         _check_session_state_empty(session)
 
         await session.create()
@@ -26,7 +26,7 @@ async def test_session_normal_lifecycle(self, session: QuerySessionAsync):
         _check_session_state_empty(session)
 
     @pytest.mark.asyncio
-    async def test_second_create_do_nothing(self, session: QuerySessionAsync):
+    async def test_second_create_do_nothing(self, session: QuerySession):
         await session.create()
         _check_session_state_full(session)
 
@@ -40,30 +40,30 @@ async def test_second_create_do_nothing(self, session: QuerySessionAsync):
         assert session._state.node_id == node_id_before
 
     @pytest.mark.asyncio
-    async def test_second_delete_do_nothing(self, session: QuerySessionAsync):
+    async def test_second_delete_do_nothing(self, session: QuerySession):
         await session.create()
 
         await session.delete()
         await session.delete()
 
     @pytest.mark.asyncio
-    async def test_delete_before_create_not_possible(self, session: QuerySessionAsync):
+    async def test_delete_before_create_not_possible(self, session: QuerySession):
         with pytest.raises(RuntimeError):
             await session.delete()
 
     @pytest.mark.asyncio
-    async def test_create_after_delete_not_possible(self, session: QuerySessionAsync):
+    async def test_create_after_delete_not_possible(self, session: QuerySession):
         await session.create()
         await session.delete()
         with pytest.raises(RuntimeError):
             await session.create()
 
-    def test_transaction_before_create_raises(self, session: QuerySessionAsync):
+    def test_transaction_before_create_raises(self, session: QuerySession):
         with pytest.raises(RuntimeError):
             session.transaction()
 
     @pytest.mark.asyncio
-    async def test_transaction_after_delete_raises(self, session: QuerySessionAsync):
+    async def test_transaction_after_delete_raises(self, session: QuerySession):
         await session.create()
 
         await session.delete()
@@ -72,24 +72,24 @@ async def test_transaction_after_delete_raises(self, session: QuerySessionAsync)
             session.transaction()
 
     @pytest.mark.asyncio
-    async def test_transaction_after_create_not_raises(self, session: QuerySessionAsync):
+    async def test_transaction_after_create_not_raises(self, session: QuerySession):
         await session.create()
         session.transaction()
 
     @pytest.mark.asyncio
-    async def test_execute_before_create_raises(self, session: QuerySessionAsync):
+    async def test_execute_before_create_raises(self, session: QuerySession):
         with pytest.raises(RuntimeError):
             await session.execute("select 1;")
 
     @pytest.mark.asyncio
-    async def test_execute_after_delete_raises(self, session: QuerySessionAsync):
+    async def test_execute_after_delete_raises(self, session: QuerySession):
         await session.create()
         await session.delete()
         with pytest.raises(RuntimeError):
             await session.execute("select 1;")
 
     @pytest.mark.asyncio
-    async def test_basic_execute(self, session: QuerySessionAsync):
+    async def test_basic_execute(self, session: QuerySession):
         await session.create()
         it = await session.execute("select 1;")
         result_sets = [result_set async for result_set in it]
@@ -100,7 +100,7 @@ async def test_basic_execute(self, session: QuerySessionAsync):
         assert list(result_sets[0].rows[0].values()) == [1]
 
     @pytest.mark.asyncio
-    async def test_two_results(self, session: QuerySessionAsync):
+    async def test_two_results(self, session: QuerySession):
         await session.create()
         res = []
 

tests/aio/query/test_query_session_pool.py

@@ -1,42 +1,42 @@
 import asyncio
 import pytest
 import ydb
-from ydb.aio.query.pool import QuerySessionPoolAsync
-from ydb.aio.query.session import QuerySessionAsync, QuerySessionStateEnum
+from ydb.aio.query.pool import QuerySessionPool
+from ydb.aio.query.session import QuerySession, QuerySessionStateEnum
 
 
-class TestQuerySessionPoolAsync:
+class TestQuerySessionPool:
     @pytest.mark.asyncio
-    async def test_checkout_provides_created_session(self, pool: QuerySessionPoolAsync):
+    async def test_checkout_provides_created_session(self, pool: QuerySessionPool):
         async with pool.checkout() as session:
             assert session._state._state == QuerySessionStateEnum.CREATED
 
     @pytest.mark.asyncio
-    async def test_oneshot_query_normal(self, pool: QuerySessionPoolAsync):
+    async def test_oneshot_query_normal(self, pool: QuerySessionPool):
         res = await pool.execute_with_retries("select 1;")
         assert len(res) == 1
 
     @pytest.mark.asyncio
-    async def test_oneshot_ddl_query(self, pool: QuerySessionPoolAsync):
+    async def test_oneshot_ddl_query(self, pool: QuerySessionPool):
         await pool.execute_with_retries("drop table if exists Queen;")
         await pool.execute_with_retries("create table Queen(key UInt64, PRIMARY KEY (key));")
         await pool.execute_with_retries("drop table Queen;")
 
     @pytest.mark.asyncio
-    async def test_oneshot_query_raises(self, pool: QuerySessionPoolAsync):
+    async def test_oneshot_query_raises(self, pool: QuerySessionPool):
         with pytest.raises(ydb.GenericError):
             await pool.execute_with_retries("Is this the real life? Is this just fantasy?")
 
     @pytest.mark.asyncio
-    async def test_retry_op_uses_created_session(self, pool: QuerySessionPoolAsync):
-        async def callee(session: QuerySessionAsync):
+    async def test_retry_op_uses_created_session(self, pool: QuerySessionPool):
+        async def callee(session: QuerySession):
             assert session._state._state == QuerySessionStateEnum.CREATED
 
         await pool.retry_operation_async(callee)
 
     @pytest.mark.asyncio
-    async def test_retry_op_normal(self, pool: QuerySessionPoolAsync):
-        async def callee(session: QuerySessionAsync):
+    async def test_retry_op_normal(self, pool: QuerySessionPool):
+        async def callee(session: QuerySession):
             async with session.transaction() as tx:
                 iterator = await tx.execute("select 1;", commit_tx=True)
                 return [result_set async for result_set in iterator]
@@ -45,18 +45,18 @@ async def callee(session: QuerySessionAsync):
         assert len(res) == 1
 
     @pytest.mark.asyncio
-    async def test_retry_op_raises(self, pool: QuerySessionPoolAsync):
+    async def test_retry_op_raises(self, pool: QuerySessionPool):
         class CustomException(Exception):
             pass
 
-        async def callee(session: QuerySessionAsync):
+        async def callee(session: QuerySession):
             raise CustomException()
 
         with pytest.raises(CustomException):
             await pool.retry_operation_async(callee)
 
     @pytest.mark.asyncio
-    async def test_pool_size_limit_logic(self, pool: QuerySessionPoolAsync):
+    async def test_pool_size_limit_logic(self, pool: QuerySessionPool):
         target_size = 5
         pool._size = target_size
         ids = set()
@@ -78,7 +78,7 @@ async def test_pool_size_limit_logic(self, pool: QuerySessionPoolAsync):
         assert pool._current_size == target_size
 
     @pytest.mark.asyncio
-    async def test_checkout_do_not_increase_size(self, pool: QuerySessionPoolAsync):
+    async def test_checkout_do_not_increase_size(self, pool: QuerySessionPool):
         session_id = None
         for _ in range(10):
             async with pool.checkout() as session:
@@ -88,7 +88,7 @@ async def test_checkout_do_not_increase_size(self, pool: QuerySessionPoolAsync):
                 assert session_id == session._state.session_id
 
     @pytest.mark.asyncio
-    async def test_pool_recreates_bad_sessions(self, pool: QuerySessionPoolAsync):
+    async def test_pool_recreates_bad_sessions(self, pool: QuerySessionPool):
         async with pool.checkout() as session:
             session_id = session._state.session_id
             await session.delete()
@@ -98,20 +98,20 @@ async def test_pool_recreates_bad_sessions(self, pool: QuerySessionPoolAsync):
             assert pool._current_size == 1
 
     @pytest.mark.asyncio
-    async def test_acquire_from_closed_pool_raises(self, pool: QuerySessionPoolAsync):
+    async def test_acquire_from_closed_pool_raises(self, pool: QuerySessionPool):
         await pool.stop()
         with pytest.raises(RuntimeError):
             await pool.acquire()
 
     @pytest.mark.asyncio
-    async def test_acquire_with_timeout_from_closed_pool_raises(self, pool: QuerySessionPoolAsync):
+    async def test_acquire_with_timeout_from_closed_pool_raises(self, pool: QuerySessionPool):
         await pool.stop()
         with pytest.raises(RuntimeError):
             await asyncio.wait_for(pool.acquire(), timeout=0.1)
 
     @pytest.mark.asyncio
     async def test_no_session_leak(self, driver, docker_project):
-        pool = ydb.aio.QuerySessionPoolAsync(driver, 1)
+        pool = ydb.aio.QuerySessionPool(driver, 1)
         docker_project.stop()
         try:
             await asyncio.wait_for(pool.acquire(), timeout=0.1)