Wait all handlers finished before CAN

temporalio · Jul 24, 2024 · d934ee1 · d934ee1
1 parent e3eba6e
commit d934ee1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/updates_and_signals/safe_message_handlers/workflow.py b/updates_and_signals/safe_message_handlers/workflow.py
@@ -200,6 +200,7 @@ async def run(self, input: ClusterManagerInput) -> ClusterManagerResult:
             if self.state.cluster_shutdown:
                 break
             if self.should_continue_as_new():
+                await workflow.wait_condition(lambda: workflow.all_handlers_finished())
                 workflow.logger.info("Continuing as new")
                 workflow.continue_as_new(
                     ClusterManagerInput(