Include WorkflowInstanceId and TaskId in Payload

Signed-off-by: Victor Chang <[email protected]>

Author: mocsharp

src/Api/Storage/Payload.cs

@@ -64,6 +64,10 @@ public enum PayloadState
 
         public string CorrelationId { get; init; }
 
+        public string? WorkflowInstanceId { get; init; }
+
+        public string? TaskId { get; init; }
+
         public int Count { get => Files.Count; }
 
         public bool HasTimedOut { get => ElapsedTime().TotalSeconds >= Timeout; }
@@ -80,13 +84,20 @@ public TimeSpan Elapsed
         public int FilesFailedToUpload { get => Files.Count(p => p.IsUploadFailed); }
 
         public Payload(string key, string correlationId, uint timeout)
+            : this(key, correlationId, null, null, timeout)
+        {
+        }
+
+        public Payload(string key, string correlationId, string? workflowInstanceId, string? taskId, uint timeout)
         {
             Guard.Against.NullOrWhiteSpace(key, nameof(key));
 
             Files = new List<FileStorageMetadata>();
             _lastReceived = new Stopwatch();
 
             CorrelationId = correlationId;
+            WorkflowInstanceId = workflowInstanceId;
+            TaskId = taskId;
             MachineName = Environment.MachineName;
             DateTimeCreated = DateTime.UtcNow;
             PayloadId = Guid.NewGuid();