gooddata-flight-server 1.34.0__py3-none-any.whl → 1.56.0__py3-none-any.whl

gooddata_flight_server/errors/error_info.py

@@ -1,13 +1,29 @@
 #  (C) 2024 GoodData Corporation
 import base64
 import traceback
-from typing import Callable, Optional, Union
+from typing import Callable, Optional, Union, cast
 
 import orjson
 import pyarrow.flight
 
 from gooddata_flight_server.errors.error_code import ErrorCode
 
+_ERROR_INFO_MAX_MSG = 256
+_ERROR_INFO_MAX_DETAIL = 512
+
+
+def _truncate_str_value(val: Optional[str], max_len: int) -> Optional[str]:
+    if val is None:
+        return None
+
+    if len(val) <= max_len:
+        return val
+
+    # no big deal that the actual max length is slightly exceeded
+    # all this truncating happens because ErrorInfo is eventually
+    # passed via gRPC headers which have 16k hard limit
+    return val[:max_len] + " [truncated]"
+
 
 class ErrorInfo:
     """
@@ -22,15 +38,15 @@ class ErrorInfo:
         body: Optional[bytes] = None,
         code: int = 0,
     ) -> None:
-        self._msg = msg
-        self._detail: Optional[str] = detail
+        self._msg = cast(str, _truncate_str_value(msg, _ERROR_INFO_MAX_MSG))
+        self._detail: Optional[str] = _truncate_str_value(detail, _ERROR_INFO_MAX_DETAIL)
         self._body: Optional[bytes] = body
         self._code: int = code
 
     @property
     def msg(self) -> str:
         """
-        :return: human readable error message
+        :return: human-readable error message
         """
         return self._msg
 
@@ -60,26 +76,36 @@ class ErrorInfo:
         """
         Updates error message.
 
-        :param msg: new message
+        :param msg: new message, up to 256 characters; will be truncated if the limit is exceeded
         :return: self, for call chaining sakes
         """
-        self._msg = msg
+        self._msg = cast(str, _truncate_str_value(msg, _ERROR_INFO_MAX_MSG))
+
         return self
 
     def with_detail(self, detail: Optional[str] = None) -> "ErrorInfo":
         """
         Updates or resets the error detail.
 
-        :param detail: detail to set; if None, the detail stored in the meta will be removed; default is None
+        :param detail: detail to set; if None, the detail stored in the meta will be removed; default is None;
+         detail can be up to 512 characters; will be truncated if the limit is exceeded
         :return: self, for call chaining sakes
         """
-        self._detail = detail
+        self._detail = _truncate_str_value(detail, _ERROR_INFO_MAX_DETAIL)
+
         return self
 
     def with_body(self, body: Optional[Union[bytes, str]]) -> "ErrorInfo":
         """
         Updates or resets the error body.
 
+        IMPORTANT: the ErrorInfo (and thus the contents of `body`) are passed out via FlightError.extra_info
+        property. The Flight RPC implementations pass the `extra_info` via gRPC headers. In turn, the gRPC headers
+        do have size limit. Keep this in mind when designing the value of `body`.
+
+        If you set body that is too large, you will run into problems like this:
+         https://github.com/grpc/grpc/issues/37852.
+
         :param body: body to set; if None, the body stored in the meta will be removed; default is None
         :return: self, for call chaining sakes
         """

gooddata_flight_server/server/flight_rpc/flight_server.py

@@ -30,12 +30,10 @@ T = TypeVar("T")
 P = ParamSpec("P")
 
 
-def rpc_decorator() -> (
-    Callable[
-        [Callable[Concatenate[Any, pyarrow.flight.ServerCallContext, P], T]],
-        Callable[Concatenate[Any, pyarrow.flight.ServerCallContext, P], T],
-    ]
-):
+def rpc_decorator() -> Callable[
+    [Callable[Concatenate[Any, pyarrow.flight.ServerCallContext, P], T]],
+    Callable[Concatenate[Any, pyarrow.flight.ServerCallContext, P], T],
+]:
     def _factory(
         fun: Callable[Concatenate[Any, pyarrow.flight.ServerCallContext, P], T],
     ) -> Callable[Concatenate[Any, pyarrow.flight.ServerCallContext, P], T]:

gooddata_flight_server/server/flight_rpc/flight_service.py

@@ -264,9 +264,7 @@ class FlightRpcService:
     def wait_for_stop(self, timeout: Optional[float] = None) -> bool:
         if self._flight_shutdown_thread is None:
             # this is really some mess in the caller code.. did not call stop() but tries to wait for it..
-            raise AssertionError(
-                "Flight server stop() was not issued yet attempting to wait for " "the server to stop."
-            )
+            raise AssertionError("Flight server stop() was not issued yet attempting to wait for the server to stop.")
 
         if self._flight_shutdown_thread.is_alive():
             self._flight_shutdown_thread.join(timeout=timeout)

gooddata_flight_server/tasks/task_executor.py

@@ -54,6 +54,15 @@ class TaskExecutor(abc.ABC):
         """
         raise NotImplementedError
 
+    @abc.abstractmethod
+    def get_task_submitted_timestamp(self, task_id: str) -> Optional[float]:
+        """
+        Returns the timestamp of when the task with the given id was submitted.
+        :param task_id: task id to get the timestamp for
+        :return: Timestamp in seconds since epoch of when the task was submitted or None if there is no such task
+        """
+        raise NotImplementedError
+
     @abc.abstractmethod
     def wait_for_result(self, task_id: str, timeout: Optional[float] = None) -> Optional[TaskExecutionResult]:
         """

gooddata_flight_server/tasks/thread_task_executor.py

@@ -565,6 +565,14 @@ class ThreadTaskExecutor(TaskExecutor, _TaskExecutionCallbacks):
         execution.start()
         self._metrics.queue_size.set(self._queue_size)
 
+    def get_task_submitted_timestamp(self, task_id: str) -> Optional[float]:
+        with self._task_lock:
+            execution = self._executions.get(task_id)
+
+        if execution is not None:
+            return execution.stats.created
+        return None
+
     def wait_for_result(self, task_id: str, timeout: Optional[float] = None) -> Optional[TaskExecutionResult]:
         with self._task_lock:
             execution = self._executions.get(task_id)
@@ -597,8 +605,8 @@ class ThreadTaskExecutor(TaskExecutor, _TaskExecutionCallbacks):
             return True
 
         if execution is None:
-            # the task was not and is not running - cancel not possible
-            return False
+            # the task was not and is not running - cancel not necessary
+            return True
 
         return execution.cancel()
 

{gooddata_flight_server-1.34.0.dist-info → gooddata_flight_server-1.56.0.dist-info}/METADATA

@@ -1,28 +1,25 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: gooddata-flight-server
-Version: 1.34.0
+Version: 1.56.0
 Summary: Flight RPC server to host custom functions
-Author: GoodData
-Author-email: support@gooddata.com
-License: MIT
-Project-URL: Documentation, https://gooddata-flight-server.readthedocs.io/en/v1.34.0
+Project-URL: Documentation, https://gooddata-flight-server.readthedocs.io/en/v1.56.0
 Project-URL: Source, https://github.com/gooddata/gooddata-python-sdk
-Keywords: gooddata,flight,rpc,flight rpc,custom functions,analytics,headless,business,intelligence,headless-bi,cloud,native,semantic,layer,sql,metrics
+Author-email: GoodData <support@gooddata.com>
+License-Expression: MIT
+License-File: LICENSE.txt
+Keywords: analytics,business,cloud,custom functions,flight,flight rpc,gooddata,headless,headless-bi,intelligence,layer,metrics,native,rpc,semantic,sql
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Database
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Software Development
 Classifier: Typing :: Typed
-Requires-Python: >=3.9.0
-Description-Content-Type: text/markdown
-License-File: LICENSE.txt
+Requires-Python: >=3.10
 Requires-Dist: dynaconf<4.0.0,>=3.1.11
 Requires-Dist: opentelemetry-api<=2.0.0,>=1.24.0
 Requires-Dist: opentelemetry-sdk<=2.0.0,>=1.24.0
@@ -31,17 +28,7 @@ Requires-Dist: prometheus-client~=0.20.0
 Requires-Dist: pyarrow>=16.1.0
 Requires-Dist: readerwriterlock~=1.0.9
 Requires-Dist: structlog<25.0.0,>=24.0.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: keywords
-Dynamic: license
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
+Description-Content-Type: text/markdown
 
 # GoodData Flight Server
 

{gooddata_flight_server-1.34.0.dist-info → gooddata_flight_server-1.56.0.dist-info}/RECORD

@@ -8,7 +8,7 @@ gooddata_flight_server/config/__init__.py,sha256=QL4TGz0QQfsNakkgMkve9k8wiR73dfo
 gooddata_flight_server/config/config.py,sha256=vMdBzGnBrxPCAsfJgpOxQKjxXl5xjEFNu32_4Bk5Qdw,17108
 gooddata_flight_server/errors/__init__.py,sha256=QL4TGz0QQfsNakkgMkve9k8wiR73dfoOF2_18ESSPRE,33
 gooddata_flight_server/errors/error_code.py,sha256=s7wcNlsGKG0PSPXNMBMQMekViZeaoJKWVboEL12kogE,5808
-gooddata_flight_server/errors/error_info.py,sha256=blpMWCABxSj5j4mLuL7PW4mXKmWBeywKNMLp70WAd44,17599
+gooddata_flight_server/errors/error_info.py,sha256=akYJk-QT0O2GQno__yc67YAxG5iBm-Fz4JDG2MQ6y4I,18852
 gooddata_flight_server/health/__init__.py,sha256=QL4TGz0QQfsNakkgMkve9k8wiR73dfoOF2_18ESSPRE,33
 gooddata_flight_server/health/health_check_http_server.py,sha256=A-wJ5L22s4PtJisNSOolrABp703w-0rhvDKKev_iN9I,3393
 gooddata_flight_server/health/server_health_monitor.py,sha256=vDe3Z5u5SPPG3RWGtrexzoJJ4HLFl7A38_hA6LfS71I,2690
@@ -24,26 +24,25 @@ gooddata_flight_server/server/auth/token_verifier_factory.py,sha256=P-ktBUcl7YNa
 gooddata_flight_server/server/auth/token_verifier_impl.py,sha256=F3phLqscCNFKOm3eF7w_KSfmM5_Nf9Kh5R9Rtr63eTU,1626
 gooddata_flight_server/server/flight_rpc/__init__.py,sha256=nfaUa69U3uZAB1udH6tza0se-3fhrsSurgPUQldoGnE,32
 gooddata_flight_server/server/flight_rpc/flight_middleware.py,sha256=Yb44PkgIL8MlYTGSPzHcnII6x9Hxlw5Y88YKeKIFeQI,5738
-gooddata_flight_server/server/flight_rpc/flight_server.py,sha256=mmdFTJwBw3dZhnYpDA9Txy9durOo3WOmlMvznv5-rBQ,8993
-gooddata_flight_server/server/flight_rpc/flight_service.py,sha256=N30rMsSjEqIgTHAARW_oiRz9AYvbm6GwRqR1jq9eP7E,10555
+gooddata_flight_server/server/flight_rpc/flight_server.py,sha256=i1-HByn0_MhRG-d6qOCiwk29LDB4M0e7UZqPoatJYA0,8973
+gooddata_flight_server/server/flight_rpc/flight_service.py,sha256=LqdZPl999_y1BXNB6GBNGEEdEDv0Y6guqa-Daqw_lgg,10522
 gooddata_flight_server/server/flight_rpc/server_methods.py,sha256=fCUndugFTlAUKG7rHpkmjQ_wkXIeRZty65WAeIhS1r0,7708
 gooddata_flight_server/tasks/__init__.py,sha256=QL4TGz0QQfsNakkgMkve9k8wiR73dfoOF2_18ESSPRE,33
 gooddata_flight_server/tasks/base.py,sha256=e_8sV9JROtXo4rhAWXKvaPbKjm5T_0iFQyXnfGtLLwg,593
 gooddata_flight_server/tasks/metrics.py,sha256=PY5FAp0lCEvsDX_xJNJ_soRZeTQ795OVh9CmBWSto7k,4085
 gooddata_flight_server/tasks/task.py,sha256=9Qzhssw53vnlUf5ULL1LsrMrjpG1jAbQBXOc1apHBH4,6485
 gooddata_flight_server/tasks/task_error.py,sha256=1Z2vJfRqRXaQk7fGMo2kGP9wLyxqngutjWyQIYpRNy8,2377
-gooddata_flight_server/tasks/task_executor.py,sha256=W7Ml5W1mIi_5L6C4a0c3YgDR1YGVQn48lXYGj_rRaKw,3728
+gooddata_flight_server/tasks/task_executor.py,sha256=xNO4UcBfdgs8aWr4KF7GNjfpXH5xy305sYyCArqFMu0,4139
 gooddata_flight_server/tasks/task_result.py,sha256=1EqgP6xmhDDiWZYqJEOZxYZzUqAkRYF7ogsqNWAF4O4,12152
 gooddata_flight_server/tasks/temporal_container.py,sha256=biNEUOSmp3D6iVoGBZnHo88tALdIwkb-SzR5besKmNo,8503
-gooddata_flight_server/tasks/thread_task_executor.py,sha256=uWC7PX8FvPHfwkwPxqeUOr_BaHNb1CUXsQZcmjZAy0E,21949
+gooddata_flight_server/tasks/thread_task_executor.py,sha256=QfluNnrm4LivPjfcXBJMyyUa-zX-EN65dnFayInd34A,22209
 gooddata_flight_server/utils/__init__.py,sha256=nfaUa69U3uZAB1udH6tza0se-3fhrsSurgPUQldoGnE,32
 gooddata_flight_server/utils/libc_utils.py,sha256=_ZwKeeB0lk2xh-gSC_7rc9felHu_hq9__GgLKZBI9sI,946
 gooddata_flight_server/utils/logging.py,sha256=3sip4Xte5YHjZNyQEQVo96rCsknxG1JPj7cIbnz0iBo,6081
 gooddata_flight_server/utils/methods_discovery.py,sha256=ZcBJU5dHBsnZD5zDXajE5merwCISfjHGcIlIWCQtNLY,3618
 gooddata_flight_server/utils/otel_tracing.py,sha256=MfKFh0TkVSUmPC6f8Oobp5XSCYxAsQyS0UaSJQF1nCw,4540
-gooddata_flight_server-1.34.0.data/scripts/gooddata-flight-server,sha256=cXl4cF_pNme_0ns-Dap825C_L-0ovWs9hmlNl7kN9kU,239
-gooddata_flight_server-1.34.0.dist-info/LICENSE.txt,sha256=o90MhGat28171lD8r77bLXsSnvDVu9CQAsWxAzx1Z2E,1064
-gooddata_flight_server-1.34.0.dist-info/METADATA,sha256=VLSBELJWAUvyUXZ0vCTj46q7Mt03mnb-6Ac34NruY-k,30237
-gooddata_flight_server-1.34.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-gooddata_flight_server-1.34.0.dist-info/top_level.txt,sha256=fkhUrPUK9CeUzu3qMaEXJbY-HVuY-sqUx8ct7AtZH0s,23
-gooddata_flight_server-1.34.0.dist-info/RECORD,,
+gooddata_flight_server-1.56.0.dist-info/METADATA,sha256=vAil4A3XK7f4mM5D3RX6UkgF1-ip3ziSSw4wwTADV78,29957
+gooddata_flight_server-1.56.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+gooddata_flight_server-1.56.0.dist-info/entry_points.txt,sha256=uew4_TNofSLf7ppUmUNtD0o2KVDr-g_7Hl3D6Dsj71g,81
+gooddata_flight_server-1.56.0.dist-info/licenses/LICENSE.txt,sha256=o90MhGat28171lD8r77bLXsSnvDVu9CQAsWxAzx1Z2E,1064
+gooddata_flight_server-1.56.0.dist-info/RECORD,,

{gooddata_flight_server-1.34.0.dist-info → gooddata_flight_server-1.56.0.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any
-

gooddata_flight_server-1.56.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+gooddata-flight-server = gooddata_flight_server.cli:server_cli

gooddata_flight_server-1.34.0.data/scripts/gooddata-flight-server

@@ -1,10 +0,0 @@
-#!python
-# (C) 2024 GoodData Corporation
-import re
-import sys
-
-import gooddata_flight_server.cli as server
-
-if __name__ == "__main__":
-    sys.argv[0] = re.sub(r"(-script\.pyw?|\.exe)?$", "", sys.argv[0])
-    sys.exit(server.server_cli())

gooddata_flight_server-1.34.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-gooddata_flight_server