Add methods to set and get ChildrenInitializedPostResetPoint in Mutab…

…leState
temporalio · Feb 4, 2025 · 6015fef · 6015fef
1 parent e3ddf93
commit 6015fef
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 2 deletions.
diff --git a/api/persistence/v1/executions.pb.go b/api/persistence/v1/executions.pb.go
diff --git a/proto/internal/temporal/server/api/persistence/v1/executions.proto b/proto/internal/temporal/server/api/persistence/v1/executions.proto
@@ -282,7 +282,7 @@ message WorkflowExecutionInfo {
     VersionedTransition last_transition_history_break_point = 101;
 
     // This is a set of child workflows that were initialized after the reset point in the parent workflow.
-    // The children are identified by the key "workflow_id:workflow_type". When the parent starts to make progress after reset, it uses this data to
+    // The children are identified by the key "workflow_type:workflow_id". When the parent starts to make progress after reset, it uses this data to
     // determine the right start policy to apply to the child.
     map<string, bool> children_initialized_post_reset_point = 102;
 }

diff --git a/service/history/workflow/mutable_state.go b/service/history/workflow/mutable_state.go
@@ -240,6 +240,8 @@ type (
 		UpdateResetRunID(runID string)
 		// IsResetRun returns true if this run is result of a reset operation.
 		IsResetRun() bool
+		SetChildrenInitializedPostResetPoint(children map[string]bool)
+		GetChildrenInitializedPostResetPoint() map[string]bool
 
 		CloneToProto() *persistencespb.WorkflowMutableState
 		RetryActivity(ai *persistencespb.ActivityInfo, failure *failurepb.Failure) (enumspb.RetryState, error)

diff --git a/service/history/workflow/mutable_state_mock.go b/service/history/workflow/mutable_state_mock.go