Patch a change to support reset 2dc workflow (cadence-workflow#3778)

zshni · Nov 20, 2020 · 4b75445 · 4b75445
1 parent 0912eef
commit 4b75445
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/service/history/execution/mutable_state_builder.go b/service/history/execution/mutable_state_builder.go
@@ -530,6 +530,11 @@ func (e *mutableStateBuilder) UpdateCurrentVersion(
 
 func (e *mutableStateBuilder) GetCurrentVersion() int64 {
 
+	// TODO: remove this after all 2DC workflows complete
+	if e.replicationState != nil {
+		return e.replicationState.CurrentVersion
+	}
+
 	if e.versionHistories != nil {
 		return e.currentVersion
 	}