Add DurableTaskClient mediator request extension methods

Author: jviau

src/Abstractions/ActivityRequest.cs

@@ -53,6 +53,21 @@ public static class ActivityRequest
     public static IActivityRequest<TResult> Create<TResult>(TaskName name, object? input = null)
         => new Request<TResult>(name, input);
 
+    /// <summary>
+    /// Gets the activity input from a <see cref="IBaseActivityRequest" />.
+    /// </summary>
+    /// <param name="request">The request to get input for.</param>
+    /// <returns>The input.</returns>
+    internal static object? GetInput(this IBaseActivityRequest request)
+    {
+        if (request is IProvidesInput provider)
+        {
+            return provider.GetInput();
+        }
+
+        return request;
+    }
+
     /// <summary>
     /// Represents an activity request where the input is not the request itself.
     /// </summary>

src/Abstractions/OrchestratorHelper.cs renamed to src/Abstractions/InputHelper.cs

@@ -53,6 +53,21 @@ public static class OrchestrationRequest
     public static IOrchestrationRequest<TResult> Create<TResult>(TaskName name, object? input = null)
         => new Request<TResult>(name, input);
 
+    /// <summary>
+    /// Gets the orchestration input from a <see cref="IBaseOrchestrationRequest" />.
+    /// </summary>
+    /// <param name="request">The request to get input for.</param>
+    /// <returns>The input.</returns>
+    internal static object? GetInput(this IBaseOrchestrationRequest request)
+    {
+        if (request is IProvidesInput provider)
+        {
+            return provider.GetInput();
+        }
+
+        return request;
+    }
+
     /// <summary>
     /// Represents an orchestration request where the input is not the request itself.
     /// </summary>

src/Abstractions/OrchestrationRequest.cs

@@ -23,7 +23,7 @@ public static Task<TResult> RunAsync<TResult>(
         Check.NotNull(context);
         Check.NotNull(request);
         TaskName name = request.GetTaskName();
-        return context.CallSubOrchestratorAsync<TResult>(name, GetInput(request), options);
+        return context.CallSubOrchestratorAsync<TResult>(name, request.GetInput(), options);
     }
 
     /// <summary>
@@ -40,7 +40,7 @@ public static Task RunAsync(
         Check.NotNull(context);
         Check.NotNull(request);
         TaskName name = request.GetTaskName();
-        return context.CallSubOrchestratorAsync(name, GetInput(request), options);
+        return context.CallSubOrchestratorAsync(name, request.GetInput(), options);
     }
 
     /// <summary>
@@ -58,7 +58,7 @@ public static Task<TResult> RunAsync<TResult>(
         Check.NotNull(context);
         Check.NotNull(request);
         TaskName name = request.GetTaskName();
-        return context.CallActivityAsync<TResult>(name, GetInput(request), options);
+        return context.CallActivityAsync<TResult>(name, request.GetInput(), options);
     }
 
     /// <summary>
@@ -75,16 +75,6 @@ public static Task RunAsync(
         Check.NotNull(context);
         Check.NotNull(request);
         TaskName name = request.GetTaskName();
-        return context.CallActivityAsync(name, GetInput(request), options);
-    }
-
-    static object? GetInput(object request)
-    {
-        if (request is IProvidesInput provider)
-        {
-            return provider.GetInput();
-        }
-
-        return request;
+        return context.CallActivityAsync(name, request.GetInput(), options);
     }
 }

src/Abstractions/TaskOrchestrationContextRequestExtensions.cs

@@ -51,4 +51,49 @@ public static Task<PurgeResult> PurgeInstancesAsync(
         DateTimeOffset? createdTo,
         CancellationToken cancellation = default)
         => PurgeInstancesAsync(client, createdFrom, createdTo, null, cancellation);
+
+    /// <summary>
+    /// Starts a new orchestration instance for execution.
+    /// </summary>
+    /// <param name="client">The client to schedule the orchestration with.</param>
+    /// <param name="request">The orchestration request to schedule.</param>
+    /// <param name="cancellation">The cancellation token.</param>
+    /// <returns>
+    /// A task that completes when the orchestration instance is successfully scheduled. The value of this task is the
+    /// instance ID the orchestration was scheduled with.
+    /// </returns>
+    /// <seealso cref="DurableTaskClient.ScheduleNewOrchestrationInstanceAsync(TaskName, object?, CancellationToken)" />
+    public static Task<string> StartNewAsync(
+        this DurableTaskClient client, IBaseOrchestrationRequest request, CancellationToken cancellation)
+    {
+        Check.NotNull(client);
+        Check.NotNull(request);
+        TaskName name = request.GetTaskName();
+        return client.ScheduleNewOrchestrationInstanceAsync(name, request.GetInput(), cancellation);
+    }
+
+    /// <summary>
+    /// Starts a new orchestration instance for execution.
+    /// </summary>
+    /// <param name="client">The client to schedule the orchestration with.</param>
+    /// <param name="request">The orchestration request to schedule.</param>
+    /// <param name="options">The options for starting this orchestration with.</param>
+    /// <param name="cancellation">The cancellation token.</param>
+    /// <returns>
+    /// A task that completes when the orchestration instance is successfully scheduled. The value of this task is
+    /// the instance ID of the scheduled orchestration instance. If a non-null instance ID was provided via
+    /// <paramref name="options" />, the same value will be returned by the completed task.
+    /// </returns>
+    /// <seealso cref="DurableTaskClient.ScheduleNewOrchestrationInstanceAsync(TaskName, object?, StartOrchestrationOptions?, CancellationToken)" />
+    public static Task<string> StartNewAsync(
+        this DurableTaskClient client,
+        IBaseOrchestrationRequest request,
+        StartOrchestrationOptions? options = null,
+        CancellationToken cancellation = default)
+    {
+        Check.NotNull(client);
+        Check.NotNull(request);
+        TaskName name = request.GetTaskName();
+        return client.ScheduleNewOrchestrationInstanceAsync(name, request.GetInput(), options, cancellation);
+    }
 }