IWF-512: Add end timestamp to events

indeedeng · Feb 6, 2025 · 2d93ad0 · 2d93ad0
1 parent 2175081
commit 2d93ad0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/service/interpreter/activityImpl.go b/service/interpreter/activityImpl.go
@@ -64,6 +64,7 @@ func StateApiWaitUntil(
 			StateId:            ptr.Any(input.Request.WorkflowStateId),
 			StateExecutionId:   ptr.Any(input.Request.Context.GetStateExecutionId()),
 			StartTimestampInMs: ptr.Any(stateApiWaitUntilStartTime),
+			EndTimestampInMs:   ptr.Any(time.Now().UnixMilli()),
 			SearchAttributes:   searchAttributes,
 		})
 		return nil, composeHttpError(
@@ -80,6 +81,7 @@ func StateApiWaitUntil(
 			StateId:            ptr.Any(input.Request.WorkflowStateId),
 			StateExecutionId:   ptr.Any(input.Request.Context.GetStateExecutionId()),
 			StartTimestampInMs: ptr.Any(stateApiWaitUntilStartTime),
+			EndTimestampInMs:   ptr.Any(time.Now().UnixMilli()),
 			SearchAttributes:   searchAttributes,
 		})
 		return nil, composeStartApiRespError(provider, err, resp)
@@ -156,6 +158,7 @@ func StateApiExecute(
 			StateId:            ptr.Any(input.Request.WorkflowStateId),
 			StateExecutionId:   input.Request.Context.StateExecutionId,
 			StartTimestampInMs: ptr.Any(stateApiExecuteStartTime),
+			EndTimestampInMs:   ptr.Any(time.Now().UnixMilli()),
 			SearchAttributes:   searchAttributes,
 		})
 		return nil, composeHttpError(
@@ -172,6 +175,7 @@ func StateApiExecute(
 			StateId:            ptr.Any(input.Request.WorkflowStateId),
 			StateExecutionId:   input.Request.Context.StateExecutionId,
 			StartTimestampInMs: ptr.Any(stateApiExecuteStartTime),
+			EndTimestampInMs:   ptr.Any(time.Now().UnixMilli()),
 			SearchAttributes:   searchAttributes,
 		})
 		return nil, composeExecuteApiRespError(provider, err, resp)

diff --git a/service/interpreter/workflowImpl.go b/service/interpreter/workflowImpl.go
@@ -51,6 +51,7 @@ func InterpreterImpl(
 					WorkflowRunId:      provider.GetWorkflowInfo(ctx).WorkflowExecution.RunID,
 					SearchAttributes:   sas,
 					StartTimestampInMs: ptr.Any(provider.GetWorkflowInfo(ctx).WorkflowStartTime.UnixMilli()),
+					EndTimestampInMs:   ptr.Any(provider.Now(ctx).UnixMilli()),
 				})
 			}
 		}
@@ -646,6 +647,7 @@ func processStateExecution(
 					StateId:            ptr.Any(state.StateId),
 					StateExecutionId:   ptr.Any(stateExeId),
 					StartTimestampInMs: ptr.Any(stateWaitUntilApiStartTime),
+					EndTimestampInMs:   ptr.Any(provider.Now(ctx).UnixMilli()),
 					SearchAttributes:   persistenceManager.GetAllSearchAttributes(),
 				})
 			}
@@ -924,6 +926,7 @@ func invokeStateExecute(
 				WorkflowRunId:      provider.GetWorkflowInfo(ctx).WorkflowExecution.RunID,
 				StateId:            ptr.Any(state.StateId),
 				StartTimestampInMs: ptr.Any(stateExecuteApiStartTime),
+				EndTimestampInMs:   ptr.Any(provider.Now(ctx).UnixMilli()),
 				StateExecutionId:   ptr.Any(stateExeId),
 				SearchAttributes:   persistenceManager.GetAllSearchAttributes(),
 			})