omnata-plugin-runtime 0.11.0a302__tar.gz → 0.11.0a304__tar.gz

{omnata_plugin_runtime-0.11.0a302 → omnata_plugin_runtime-0.11.0a304}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.11.0-a302"
+version = "0.11.0-a304"
 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.0a302 → omnata_plugin_runtime-0.11.0a304}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -1185,18 +1185,18 @@ class InboundSyncRequest(SyncRequest):
             query_id: Optional[str] = None
     ) -> str:
         binding_values = []
-        values_clauses = []
+        select_clauses = []
         
         with self._snowflake_query_lock:
             if query_id is None:
                 query_id = self._get_query_id_for_now()
             for stream_name, latest_state in stream_states_for_upload.items():
                 binding_values.extend([stream_name, query_id, json.dumps(latest_state)])
-                values_clauses.append(
-                    f"(?, ?, PARSE_JSON(?))"
+                select_clauses.append(
+                    f"select ?, ?, PARSE_JSON(?)"
                 )
             final_query = f"""INSERT INTO {self.state_register_table_name} (STREAM_NAME, QUERY_ID, STATE_VALUE)
-                VALUES {','.join(values_clauses)}"""
+                {' union all '.join(select_clauses)}"""
             self._session.sql(final_query, binding_values).collect()
     
     def apply_progress_updates(self, ignore_errors:bool = True):

{omnata_plugin_runtime-0.11.0a302 → omnata_plugin_runtime-0.11.0a304}/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