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

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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.11.0-a304"
+version = "0.11.0-a306"
 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.0a304 → omnata_plugin_runtime-0.11.0a306}/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):
         """
@@ -1220,6 +1222,14 @@ class InboundSyncRequest(SyncRequest):
                 if result is None:
                     return False
                 self._last_stream_progress_update = new_progress_update
+            else:
+                completed_streams_awaiting_results_upload = [
+                    s for s in self._completed_streams if s in self._apply_results and self._apply_results[s] is not None
+                ]
+                for completed_stream_awaiting_results_upload in completed_streams_awaiting_results_upload:
+                    logger.debug(
+                        f"Stream {completed_stream_awaiting_results_upload} is marked as completed but awaiting upload of {len(self._apply_results[completed_stream_awaiting_results_upload])} results"
+                    )
         return True
 
     def apply_cancellation(self):