gooddata-flight-server 1.69.1.dev2__tar.gz → 1.70.1.dev1__tar.gz

{gooddata_flight_server-1.69.1.dev2 → gooddata_flight_server-1.70.1.dev1}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: gooddata-flight-server
-Version: 1.69.1.dev2
+Version: 1.70.1.dev1
 Summary: Flight RPC server to host custom functions
-Project-URL: Documentation, https://gooddata-flight-server.readthedocs.io/en/v1.69.1.dev2
+Project-URL: Documentation, https://gooddata-flight-server.readthedocs.io/en/v1.70.1.dev1
 Project-URL: Source, https://github.com/gooddata/gooddata-python-sdk
 Author-email: GoodData <support@gooddata.com>
 License-Expression: MIT

{gooddata_flight_server-1.69.1.dev2 → gooddata_flight_server-1.70.1.dev1}/pyproject.toml

@@ -1,7 +1,7 @@
 # (C) 2025 GoodData Corporation
 [project]
 name = "gooddata-flight-server"
-version = "1.69.1.dev2"
+version = "1.70.1.dev1"
 description = "Flight RPC server to host custom functions"
 readme = "README.md"
 license = "MIT"
@@ -52,7 +52,7 @@ classifiers = [
 ]
 
 [project.urls]
-Documentation = "https://gooddata-flight-server.readthedocs.io/en/v1.69.1.dev2"
+Documentation = "https://gooddata-flight-server.readthedocs.io/en/v1.70.1.dev1"
 Source = "https://github.com/gooddata/gooddata-python-sdk"
 
 [project.scripts]

{gooddata_flight_server-1.69.1.dev2 → gooddata_flight_server-1.70.1.dev1}/src/gooddata_flight_server/server/flight_rpc/server_methods.py

@@ -101,6 +101,8 @@ class FlightServerMethods:
                     f"While the result exists, it is of an unexpected type: {type(result).__name__} ",
                 ).to_internal_error()
 
+            finalizer = FlightServerMethods.call_finalizer_middleware(context)
+
             rlock, data = result.acquire_data()
 
             def _on_end(_: pyarrow.ArrowException | None) -> None:
@@ -121,7 +123,6 @@ class FlightServerMethods:
                         # log and sink these Exceptions - not much to do
                         _LOGGER.error("do_get_close_failed", exc_info=True)
 
-            finalizer = FlightServerMethods.call_finalizer_middleware(context)
             finalizer.register_on_end(_on_end)
 
             if isinstance(data, pyarrow.Table):

{gooddata_flight_server-1.69.1.dev2 → gooddata_flight_server-1.70.1.dev1}/src/gooddata_flight_server/tasks/thread_task_executor.py

@@ -158,6 +158,7 @@ class _TaskExecution:
         "_result_future",
         "_lock",
         "_completed",
+        "_done",
         "_stats",
     )
 
@@ -189,6 +190,8 @@ class _TaskExecution:
         # all these are protected using the lock
         self._result_future: Future[Union[TaskResult, TaskError]] | None = None
         self._completed: threading.Condition = threading.Condition(self._lock)
+        # indicates the task actually finished
+        self._done: bool = False
 
     @property
     def task(self) -> Task:
@@ -235,6 +238,7 @@ class _TaskExecution:
 
         with self._lock:
             execution_result = self._cb.process_task_result(self, self._result_future)
+            self._done = True
             self._completed.notify_all()
 
         self._complete_execution_span(execution_result)
@@ -291,7 +295,7 @@ class _TaskExecution:
 
     def wait_for_completion(self, timeout: float | None = None) -> None:
         with self._lock:
-            completed = self._completed.wait(timeout=timeout)
+            completed = self._completed.wait_for(lambda: self._done, timeout=timeout)
 
         if not completed:
             raise TaskWaitTimeoutError(task_id=self._task.task_id, cmd=self._task.cmd)