omnata-plugin-runtime 0.8.0a195__py3-none-any.whl → 0.8.0a198__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -48,6 +48,7 @@ from snowflake.snowpark.functions import col
 from tenacity import Retrying, stop_after_attempt, wait_fixed, retry_if_exception_message
 
 from .logging import OmnataPluginLogHandler, logger, tracer
+from opentelemetry import context
 
 from .api import (
     PluginMessage,
@@ -316,6 +317,8 @@ class SyncRequest(ABC):
         self._cancel_checking_task = None
         self._rate_limit_update_task = None
         self._last_stream_progress_update = None
+        # store the opentelemetry context so that it can be attached inside threads
+        self.opentelemetry_context = context.get_current()
 
         threading.excepthook = self.thread_exception_hook
         if self.development_mode is False:
@@ -387,6 +390,7 @@ class SyncRequest(ABC):
         """
         Designed to be run in a thread, this method polls the results every 20 seconds and sends them back to Snowflake.
         """
+        context.attach(self.opentelemetry_context)
         while not cancellation_token.is_set():
             logger.debug("apply results worker checking for results")
             self.apply_results_queue()
@@ -399,6 +403,7 @@ class SyncRequest(ABC):
         It also gives us the latest rate limit state from Snowflake, so that activity on other syncs/branches can
         impact rate limiting on this one.
         """
+        context.attach(self.opentelemetry_context)
         while not cancellation_token.is_set():
             try:
                 self.apply_rate_limit_state()
@@ -408,6 +413,7 @@ class SyncRequest(ABC):
         logger.info("rate limit update worker exiting")
 
     def __cancel_checking_worker(self, cancellation_token:threading.Event):
+        context.attach(self.opentelemetry_context)
         """
         Designed to be run in a thread, this method checks to see if the sync run has been cancelled
         or if the deadline has been reached.
@@ -1637,6 +1643,8 @@ class OmnataPlugin(ABC):
         take care of various status updates and loading enqueued results.
         Only set this to True if you plan to do all of this yourself (e.g. in a Java stored proc)
         """
+        # store the opentelemetry context so that it can be attached inside threads
+        self.opentelemetry_context = context.get_current()
     
 
     @abstractmethod
@@ -1946,6 +1954,7 @@ def __managed_outbound_processing_worker(
     Consumes a fixed sized set of records by passing them to the wrapped function,
     while adhering to the defined API constraints.
     """
+    context.attach(plugin_class_obj.opentelemetry_context)
     logger.debug(
         f"worker {worker_index} processing. Cancelled: {cancellation_token.is_set()}"
     )
@@ -2128,6 +2137,7 @@ def __managed_inbound_processing_worker(
     A worker thread for the managed_inbound_processing annotation.
     Passes single streams at a time to the wrapped function, adhering to concurrency constraints.
     """
+    context.attach(plugin_class_obj.opentelemetry_context)
     while not cancellation_token.is_set():
         # Get our generator object out of the queue
         logger.debug(

omnata_plugin_runtime/rate_limiting.py

@@ -15,13 +15,10 @@ import logging
 from pydantic import Field, root_validator, PrivateAttr, field_serializer
 from pydantic_core import to_jsonable_python
 from .configuration import SubscriptableBaseModel
-from .logging import logger
+from .logging import logger, tracer
 import pytz
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
-from opentelemetry import trace
-
-tracer = trace.get_tracer(__name__)
 
 TimeUnitType = Literal["second", "minute", "hour", "day"]
 

{omnata_plugin_runtime-0.8.0a195.dist-info → omnata_plugin_runtime-0.8.0a198.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.8.0a195
+Version: 0.8.0a198
 Summary: Classes and common runtime components for building and running Omnata Plugins
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_runtime-0.8.0a195.dist-info → omnata_plugin_runtime-0.8.0a198.dist-info}/RECORD

@@ -3,10 +3,10 @@ omnata_plugin_runtime/api.py,sha256=tVi4KLL0v5N3yz3Ie0kSyFemryu572gCbtSRfWN6wBU,
 omnata_plugin_runtime/configuration.py,sha256=6JmgE4SL3F5cGlDYqt17A1vTFu6nB74yWgEpQ5qV9ho,38380
 omnata_plugin_runtime/forms.py,sha256=ueodN2GIMS5N9fqebpY4uNGJnjEb9HcuaVQVfWH-cGg,19838
 omnata_plugin_runtime/logging.py,sha256=WBuZt8lF9E5oFWM4KYQbE8dDJ_HctJ1pN3BHwU6rcd0,4461
-omnata_plugin_runtime/omnata_plugin.py,sha256=ZMoF5ef9GT0ri8nof_3JaJT4dTQ9oDRf22zw-5UryjI,130600
+omnata_plugin_runtime/omnata_plugin.py,sha256=GAvFRnx02_beTuw1LhPgOBxS_cTJmjDXc4EwVk69ZY8,131191
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=3nqIWOlUdNVkBu7YBqC73Gb5G_xvMUzmlZf838N0JSg,32728
-omnata_plugin_runtime/rate_limiting.py,sha256=6fn_h2vxcHbqqiW-OZ6FKfNYv_XlNvorsrCknVce2PA,25929
-omnata_plugin_runtime-0.8.0a195.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
-omnata_plugin_runtime-0.8.0a195.dist-info/METADATA,sha256=JW6TXPXIu60cr3BLAydTSegpjml5vFKFJQuRWZbmP8M,2148
-omnata_plugin_runtime-0.8.0a195.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-omnata_plugin_runtime-0.8.0a195.dist-info/RECORD,,
+omnata_plugin_runtime/rate_limiting.py,sha256=eOWVRYWiqPlVeYzmB1exVXfXbrcpmYb7vtTi9B-4zkQ,25868
+omnata_plugin_runtime-0.8.0a198.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
+omnata_plugin_runtime-0.8.0a198.dist-info/METADATA,sha256=tnVsgQAi4ZfgczB0MjUQD-Sbezcm3jxi4TfYO124OG8,2148
+omnata_plugin_runtime-0.8.0a198.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+omnata_plugin_runtime-0.8.0a198.dist-info/RECORD,,