Merge pull request #94 from sclasen/snapshot-continuation-id

Add ContinuedExecutionRunID to FSMSnapshot
sclasen · Sep 30, 2015 · 422913e · 422913e
2 parents 5656897 + 237684c
commit 422913e
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/fsm/snapshots.go b/fsm/snapshots.go
@@ -18,9 +18,11 @@ import (
 )
 
 type FSMSnapshot struct {
-	State      *FSMSnapshotState
-	Correlator *EventCorrelator
-	Events     []*FSMSnapshotEvent
+	State                   *FSMSnapshotState
+	Correlator              *EventCorrelator
+	Events                  []*FSMSnapshotEvent
+	WorkflowID              *string
+	ContinuedExecutionRunID *string
 }
 
 type FSMSnapshotState struct {
@@ -130,6 +132,12 @@ func (s *snapshotter) FromHistoryEventIterator(itr HistoryEventIterator) ([]FSMS
 				break
 			}
 
+			snapshot.WorkflowID = &state.WorkflowID
+
+			if event.WorkflowExecutionStartedEventAttributes != nil {
+				snapshot.ContinuedExecutionRunID = event.WorkflowExecutionStartedEventAttributes.ContinuedExecutionRunID
+			}
+
 			snapshot.Correlator = nextCorrelator
 			nextCorrelator = nil