Bugfix: fix continue as new history size (uber#2295)

dankohn · Jul 27, 2019 · 66c0c5d · 66c0c5d
1 parent 5dcef3f
commit 66c0c5d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/service/history/workflowExecutionContext.go b/service/history/workflowExecutionContext.go
@@ -460,6 +460,7 @@ func (c *workflowExecutionContextImpl) updateWorkflowExecutionWithNew(
 
 	var newWorkflow *persistence.WorkflowSnapshot
 	if newContext != nil && newMutableState != nil && newWorkflowTransactionPolicy != nil {
+
 		defer func() {
 			if retError != nil {
 				newContext.clear()
@@ -470,17 +471,15 @@ func (c *workflowExecutionContextImpl) updateWorkflowExecutionWithNew(
 		if err != nil {
 			return err
 		}
-
 		newWorkflowSizeSize := newContext.getHistorySize()
 		eventsSize, err := c.persistFirstWorkflowEvents(workflowEventsSeq[0])
 		if err != nil {
 			return err
 		}
 		newWorkflowSizeSize += eventsSize
-
-		newContext.setHistorySize(currentWorkflowSize)
+		newContext.setHistorySize(newWorkflowSizeSize)
 		newWorkflow.ExecutionStats = &persistence.ExecutionStats{
-			HistorySize: currentWorkflowSize,
+			HistorySize: newWorkflowSizeSize,
 		}
 	}