omnata-plugin-runtime 0.11.0a303__tar.gz → 0.11.0a305__tar.gz

{omnata_plugin_runtime-0.11.0a303 → omnata_plugin_runtime-0.11.0a305}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-runtime
-Version: 0.11.0a303
+Version: 0.11.0a305
 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.11.0a303 → omnata_plugin_runtime-0.11.0a305}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.11.0-a303"
+version = "0.11.0-a305"
 description = "Classes and common runtime components for building and running Omnata Plugins"
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_runtime-0.11.0a303 → omnata_plugin_runtime-0.11.0a305}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -1198,6 +1198,8 @@ class InboundSyncRequest(SyncRequest):
             final_query = f"""INSERT INTO {self.state_register_table_name} (STREAM_NAME, QUERY_ID, STATE_VALUE)
                 {' union all '.join(select_clauses)}"""
             self._session.sql(final_query, binding_values).collect()
+            streams_included = list(stream_states_for_upload.keys())
+            logger.debug(f"Inserted state for streams: {streams_included} with query ID {query_id}")
     
     def apply_progress_updates(self, ignore_errors:bool = True):
         """

{omnata_plugin_runtime-0.11.0a303 → omnata_plugin_runtime-0.11.0a305}/src/omnata_plugin_runtime/plugin_entrypoints.py

@@ -282,6 +282,11 @@ class PluginEntrypoint:
                     # token is set. We throw it here as an error since that's currently how it flows back to the engine with a DELAYED state
                     raise DeadlineReachedException()
             finally:
+                # try to upload any remaining results
+                try:
+                    inbound_sync_request.apply_results_queue()
+                except Exception as e:
+                    logger.warning(f"Error uploading remaining results: {str(e)}", exc_info=True)
                 # cancel the thread so we don't leave anything hanging around and cop a nasty error
                 try:
                     inbound_sync_request._thread_cancellation_token.set()  # pylint: disable=protected-access