timeout style

Author: rekby

ydb/_topic_common/common.py

@@ -76,9 +76,7 @@ def unsafe_call_with_future(
         """
         return asyncio.run_coroutine_threadsafe(coro, self._loop)
 
-    def unsafe_call_with_result(
-        self, coro: typing.Coroutine, timeout: typing.Union[int, float, None]
-    ):
+    def unsafe_call_with_result(self, coro: typing.Coroutine, timeout: TimeoutType):
         """
         returned result from coro may be lost by race future cancel by timeout and return value from coroutine
         """

ydb/_topic_reader/topic_reader_sync.py

@@ -4,7 +4,11 @@
 from typing import List, Union, Iterable, Optional
 
 from ydb._grpc.grpcwrapper.common_utils import SupportedDriverType
-from ydb._topic_common.common import _get_shared_event_loop, CallFromSyncToAsync
+from ydb._topic_common.common import (
+    _get_shared_event_loop,
+    CallFromSyncToAsync,
+    TimeoutType,
+)
 from ydb._topic_reader import datatypes
 from ydb._topic_reader.datatypes import PublicMessage, PublicBatch, ICommittable
 from ydb._topic_reader.topic_reader import (
@@ -156,7 +160,7 @@ def commit(
         self._caller.call_sync(self._async_reader.commit(mess))
 
     def commit_with_ack(
-        self, mess: ICommittable, timeout: typing.Union[int, float, None] = None
+        self, mess: ICommittable, timeout: TimeoutType = None
     ) -> Union[CommitResult, List[CommitResult]]:
         """
         write commit message to a buffer and wait ack from the server.
@@ -193,7 +197,7 @@ def flush(self):
         """
         raise NotImplementedError()
 
-    def close(self, *, timeout: typing.Union[int, float, None] = None):
+    def close(self, *, timeout: TimeoutType = None):
         if self._closed:
             return
 

ydb/_topic_writer/topic_writer_sync.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import asyncio
-import typing
 from concurrent.futures import Future
 from typing import Union, List, Optional
 
@@ -57,9 +56,7 @@ def __enter__(self):
     def __exit__(self, exc_type, exc_val, exc_tb):
         self.close()
 
-    def close(
-        self, *, flush: bool = True, timeout: typing.Union[int, float, None] = None
-    ):
+    def close(self, *, flush: bool = True, timeout: TimeoutType = None):
         if self._closed:
             return