Copy new state variables to original variables after saving workflow …

…instance
NitorCreations · Aug 1, 2014 · d8b88fd · d8b88fd
1 parent e3f749e
commit d8b88fd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/...ine/src/main/java/com/nitorcreations/nflow/engine/internal/executor/WorkflowExecutor.java b/...ine/src/main/java/com/nitorcreations/nflow/engine/internal/executor/WorkflowExecutor.java
@@ -129,9 +129,8 @@ private WorkflowInstance saveWorkflowInstanceState(StateExecutionImpl execution,
       builder.setState(execution.getNextState()).setStateText(execution.getNextStateReason()).setRetries(0);
     }
     actionBuilder.setExecutionEnd(now()).setStateText(execution.getNextStateReason());
-    WorkflowInstance newInstance = builder.build();
-    workflowInstances.updateWorkflowInstance(newInstance, actionBuilder.build());
-    return newInstance;
+    workflowInstances.updateWorkflowInstance(builder.build(), actionBuilder.build());
+    return builder.setOriginalStateVariables(instance.stateVariables).build();
   }
 
   private boolean isNextActivationImmediately(StateExecutionImpl execution) {

diff --git a/...ine/src/main/java/com/nitorcreations/nflow/engine/workflow/instance/WorkflowInstance.java b/...ine/src/main/java/com/nitorcreations/nflow/engine/workflow/instance/WorkflowInstance.java
@@ -128,6 +128,12 @@ public Builder setProcessing(boolean processing) {
       return this;
     }
 
+    public Builder setOriginalStateVariables(Map<String, String> originalStateVariables) {
+      this.originalStateVariables.clear();
+      this.originalStateVariables.putAll(originalStateVariables);
+      return this;
+    }
+
     public Builder setStateVariables(Map<String, String> stateVariables) {
       this.stateVariables.clear();
       this.stateVariables.putAll(stateVariables);