[Durable] Custom orchestration status (#592)

Author: AnatoliB

examples/durable/DurableApp/CustomStatusOrchestrator/function.json

@@ -0,0 +1,9 @@
+{
+  "bindings": [
+    {
+      "name": "Context",
+      "type": "orchestrationTrigger",
+      "direction": "in"
+    }
+  ]
+}

examples/durable/DurableApp/CustomStatusOrchestrator/run.ps1

@@ -0,0 +1,22 @@
+using namespace System.Net
+
+param($Context)
+
+Write-Host 'CustomStatusOrchestrator: started.'
+
+$output = @()
+
+Set-DurableCustomStatus -CustomStatus 'Processing Tokyo'
+$output += Invoke-ActivityFunction -FunctionName 'SayHello' -Input 'Tokyo'
+
+Set-DurableCustomStatus -CustomStatus @{ ProgressMessage = 'Processing Seattle'; Stage = 2 }
+$output += Invoke-ActivityFunction -FunctionName 'SayHello' -Input 'Seattle'
+
+Set-DurableCustomStatus -CustomStatus @('Processing London', 'Last stage')
+$output += Invoke-ActivityFunction -FunctionName 'SayHello' -Input 'London'
+
+Set-DurableCustomStatus 'Processing completed'
+
+Write-Host 'CustomStatusOrchestrator: finished.'
+
+$output

examples/durable/DurableApp/CustomStatusStart/function.json

@@ -0,0 +1,24 @@
+{
+  "bindings": [
+    {
+      "authLevel": "function",
+      "type": "httpTrigger",
+      "direction": "in",
+      "name": "Request",
+      "methods": [
+        "get",
+        "post"
+      ]
+    },
+    {
+      "type": "http",
+      "direction": "out",
+      "name": "Response"
+    },
+    {
+      "name": "starter",
+      "type": "durableClient",
+      "direction": "in"
+    }    
+  ]
+}

examples/durable/DurableApp/CustomStatusStart/run.ps1

@@ -0,0 +1,13 @@
+using namespace System.Net
+
+param($Request, $TriggerMetadata)
+
+Write-Host 'CustomStatusStart started'
+
+$InstanceId = Start-NewOrchestration -FunctionName 'CustomStatusOrchestrator' -InputObject 'Hello'
+Write-Host "Started orchestration with ID = '$InstanceId'"
+
+$Response = New-OrchestrationCheckStatusResponse -Request $Request -InstanceId $InstanceId
+Push-OutputBinding -Name Response -Value $Response
+
+Write-Host 'CustomStatusStart completed'

src/Durable/Commands/SetDurableCustomStatusCommand.cs

@@ -0,0 +1,29 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+#pragma warning disable 1591 // Missing XML comment for publicly visible type or member 'member'
+
+namespace Microsoft.Azure.Functions.PowerShellWorker.Durable.Commands
+{
+    using System.Collections;
+    using System.Management.Automation;
+
+    [Cmdlet("Set", "DurableCustomStatus")]
+    public class SetDurableCustomStatusCommand : PSCmdlet
+    {
+        [Parameter(
+            Mandatory = true,
+            Position = 0,
+            ValueFromPipeline = true)]
+        public object CustomStatus { get; set; }
+
+        protected override void EndProcessing()
+        {
+            var privateData = (Hashtable)MyInvocation.MyCommand.Module.PrivateData;
+            var context = (OrchestrationContext)privateData[SetFunctionInvocationContextCommand.ContextKey];
+            context.CustomStatus = CustomStatus;
+        }
+    }
+}

src/Durable/OrchestrationContext.cs

@@ -31,8 +31,10 @@ public class OrchestrationContext
         [DataMember]
         internal HistoryEvent[] History { get; set; }
 
-        public DateTime CurrentUtcDateTime {get; internal set; }
+        public DateTime CurrentUtcDateTime { get; internal set; }
 
         internal OrchestrationActionCollector OrchestrationActionCollector { get; } = new OrchestrationActionCollector();
+
+        internal object CustomStatus { get; set; }
     }
 }

src/Durable/OrchestrationFailureException.cs

@@ -24,17 +24,23 @@ public OrchestrationFailureException()
         {
         }
 
-        public OrchestrationFailureException(List<List<OrchestrationAction>> actions, Exception innerException)
-            : base(FormatOrchestrationFailureMessage(actions, innerException), innerException)
+        public OrchestrationFailureException(
+            List<List<OrchestrationAction>> actions,
+            object customStatus,
+            Exception innerException)
+            : base(FormatOrchestrationFailureMessage(actions, customStatus, innerException), innerException)
         {
         }
 
-        private static string FormatOrchestrationFailureMessage(List<List<OrchestrationAction>> actions, Exception exception)
+        private static string FormatOrchestrationFailureMessage(
+            List<List<OrchestrationAction>> actions,
+            object customStatus,
+            Exception exception)
         {
             // For more details on why this message looks like this, see:
             // - https://github.com/Azure/azure-functions-durable-js/pull/145
             // - https://github.com/Azure/azure-functions-durable-extension/pull/1171
-            var orchestrationMessage = new OrchestrationMessage(isDone: false, actions, output: null, exception.Message);
+            var orchestrationMessage = new OrchestrationMessage(isDone: false, actions, output: null, customStatus, exception.Message);
             var message = $"{exception.Message}{OutOfProcDataLabel}{JsonConvert.SerializeObject(orchestrationMessage)}";
             return message;
         }

src/Durable/OrchestrationInvoker.cs

@@ -40,7 +40,7 @@ public Hashtable Invoke(OrchestrationBindingInfo orchestrationBindingInfo, IPowe
                 {
                     // The orchestration function should be stopped and restarted
                     pwsh.StopInvoke();
-                    return CreateOrchestrationResult(isDone: false, actions, output: null);
+                    return CreateOrchestrationResult(isDone: false, actions, output: null, context.CustomStatus);
                 }
                 else
                 {
@@ -49,13 +49,13 @@ public Hashtable Invoke(OrchestrationBindingInfo orchestrationBindingInfo, IPowe
                         // The orchestration function completed
                         pwsh.EndInvoke(asyncResult);
                         var result = FunctionReturnValueBuilder.CreateReturnValueFromFunctionOutput(outputBuffer);
-                        return CreateOrchestrationResult(isDone: true, actions, output: result);
+                        return CreateOrchestrationResult(isDone: true, actions, output: result, context.CustomStatus);
                     }
                     catch (Exception e)
                     {
                         // The orchestrator code has thrown an unhandled exception:
                         // this should be treated as an entire orchestration failure
-                        throw new OrchestrationFailureException(actions, e);
+                        throw new OrchestrationFailureException(actions, context.CustomStatus, e);
                     }
                 }
             }
@@ -68,9 +68,10 @@ public Hashtable Invoke(OrchestrationBindingInfo orchestrationBindingInfo, IPowe
         private static Hashtable CreateOrchestrationResult(
             bool isDone,
             List<List<OrchestrationAction>> actions,
-            object output)
+            object output,
+            object customStatus)
         {
-            var orchestrationMessage = new OrchestrationMessage(isDone, actions, output);
+            var orchestrationMessage = new OrchestrationMessage(isDone, actions, output, customStatus);
             return new Hashtable { { AzFunctionInfo.DollarReturn, orchestrationMessage } };
         }
     }

src/Durable/OrchestrationMessage.cs

@@ -14,12 +14,18 @@ namespace Microsoft.Azure.Functions.PowerShellWorker.Durable
     /// </summary>
     internal class OrchestrationMessage
     {
-        public OrchestrationMessage(bool isDone, List<List<OrchestrationAction>> actions, object output, string error = null)
+        public OrchestrationMessage(
+            bool isDone,
+            List<List<OrchestrationAction>> actions,
+            object output,
+            object customStatus,
+            string error = null)
         {
             IsDone = isDone;
             Actions = actions;
             Output = output;
             Error = error;
+            CustomStatus = customStatus;
         }
 
         /// <summary>
@@ -41,5 +47,10 @@ public OrchestrationMessage(bool isDone, List<List<OrchestrationAction>> actions
         /// The orchestration error message.
         /// </summary>
         public readonly string Error;
+
+        /// <summary>
+        /// Custom orchestration status.
+        /// </summary>
+        public readonly object CustomStatus;
     }
 }

src/Modules/Microsoft.Azure.Functions.PowerShellWorker/Microsoft.Azure.Functions.PowerShellWorker.psd1

@@ -59,6 +59,7 @@ CmdletsToExport = @(
     'Get-OutputBinding',
     'Invoke-ActivityFunction',
     'Push-OutputBinding',
+    'Set-DurableCustomStatus',
     'Set-FunctionInvocationContext',
     'Start-DurableExternalEventListener',
     'Start-DurableTimer',