Add New Property Properties to TaskOrchestrationContext (#415)

* initial commit

* sort usings

* update summary and also properties

* update changelog and update comments

* udpate by comment

* pull latest protobuf

* update by comment and also fix some warnings

* remove default settings

* update by reviews

Author: nytian

CHANGELOG.md

@@ -2,7 +2,8 @@
 
 ## (Unreleased)
 
-- Add automatic retry on gateway timeout in `GrpcDurableTaskClient.WaitForInstanceCompletionAsync` in ([#412](https://github.com/microsoft/durabletask-dotnet/pull/412))
+- Add New Property Properties to TaskOrchestrationContext by @nytian in [#415](https://github.com/microsoft/durabletask-dotnet/pull/415)
+- Add automatic retry on gateway timeout in `GrpcDurableTaskClient.WaitForInstanceCompletionAsync` in [#412](https://github.com/microsoft/durabletask-dotnet/pull/412))
 - Add specific logging for NotFound error on worker connection by @halspang in ([#413](https://github.com/microsoft/durabletask-dotnet/pull/413))
 - Add user agent header to gRPC called in ([#417](https://github.com/microsoft/durabletask-dotnet/pull/417))
 - Enrich User-Agent Header in gRPC Metadata to indicate Client or Worker as caller ([#421](https://github.com/microsoft/durabletask-dotnet/pull/421))

src/Abstractions/TaskOrchestrationContext.cs

@@ -65,6 +65,11 @@ public abstract class TaskOrchestrationContext
     /// </summary>
     public virtual string Version => string.Empty;
 
+    /// <summary>
+    /// Gets the configuration settings for the orchestration context.
+    /// </summary>
+    public virtual IReadOnlyDictionary<string, object?> Properties { get; } = new Dictionary<string, object?>();
+
     /// <summary>
     /// Gets the entity feature, for interacting with entities.
     /// </summary>

src/Grpc/orchestrator_service.proto

@@ -11,6 +11,7 @@ import "google/protobuf/timestamp.proto";
 import "google/protobuf/duration.proto";
 import "google/protobuf/wrappers.proto";
 import "google/protobuf/empty.proto";
+import "google/protobuf/struct.proto";
 
 message OrchestrationInstance {
     string instanceId = 1;
@@ -318,6 +319,7 @@ message OrchestratorRequest {
     repeated HistoryEvent newEvents = 4;
     OrchestratorEntityParameters entityParameters = 5;
     bool requiresHistoryStreaming = 6;
+    map<string, google.protobuf.Value> properties = 7;
 }
 
 message OrchestratorResponse {

src/Grpc/versions.txt

@@ -1,2 +1,2 @@
-# The following files were downloaded from branch main at 2025-03-24 23:37:31 UTC
-https://raw.githubusercontent.com/microsoft/durabletask-protobuf/c85ef11430ff8e10e21105abb545b0803bb86c66/protos/orchestrator_service.proto
+# The following files were downloaded from branch main at 2025-04-23 23:27:00 UTC
+https://raw.githubusercontent.com/microsoft/durabletask-protobuf/fbe5bb20835678099fc51a44993ed9b045dee5a6/protos/orchestrator_service.proto

src/ScheduledTasks/Client/ScheduledTaskClient.cs

@@ -5,7 +5,7 @@ namespace Microsoft.DurableTask.ScheduledTasks;
 
 /// <summary>
 /// Client for managing scheduled tasks.
-/// Provides methods to retrieve a ScheduleClient, list all schedules, 
+/// Provides methods to retrieve a ScheduleClient, list all schedules,
 /// and get details of a specific schedule.
 /// </summary>
 public abstract class ScheduledTaskClient

src/ScheduledTasks/Models/ScheduleState.cs

@@ -50,4 +50,4 @@ public void RefreshScheduleRunExecutionToken()
     {
         this.ExecutionToken = Guid.NewGuid().ToString("N");
     }
-}
+}

src/Shared/AzureManaged/GrpcRetryPolicyDefaults.cs

@@ -39,7 +39,6 @@ sealed class GrpcRetryPolicyDefaults
     /// </remarks>
     public static readonly ServiceConfig DefaultServiceConfig;
 
-
     /// <summary>
     /// The default retry policy for gRPC operations.
     /// </summary>

src/Shared/Grpc/ProtoUtils.cs

@@ -904,6 +904,37 @@ internal static T Base64Decode<T>(this MessageParser parser, string encodedMessa
             failureDetails.IsNonRetriable);
     }
 
+    /// <summary>
+    /// Converts a <see cref="Google.Protobuf.WellKnownTypes.Value"/> instance to a corresponding C# object.
+    /// </summary>
+    /// <param name="value">The Protobuf Value to convert.</param>
+    /// <returns>The corresponding C# object.</returns>
+    /// <exception cref="NotSupportedException">
+    /// Thrown when the Protobuf Value.KindCase is not one of the supported types.
+    /// </exception>
+    internal static object? ConvertValueToObject(Google.Protobuf.WellKnownTypes.Value value)
+    {
+        switch (value.KindCase)
+        {
+            case Google.Protobuf.WellKnownTypes.Value.KindOneofCase.NullValue:
+                return null;
+            case Google.Protobuf.WellKnownTypes.Value.KindOneofCase.NumberValue:
+                return value.NumberValue;
+            case Google.Protobuf.WellKnownTypes.Value.KindOneofCase.StringValue:
+                return value.StringValue;
+            case Google.Protobuf.WellKnownTypes.Value.KindOneofCase.BoolValue:
+                return value.BoolValue;
+            case Google.Protobuf.WellKnownTypes.Value.KindOneofCase.StructValue:
+                return value.StructValue.Fields.ToDictionary(
+                    pair => pair.Key,
+                    pair => ConvertValueToObject(pair.Value));
+            case Google.Protobuf.WellKnownTypes.Value.KindOneofCase.ListValue:
+                return value.ListValue.Values.Select(ConvertValueToObject).ToList();
+            default:
+                throw new NotSupportedException($"Unsupported Value kind: {value.KindCase}");
+        }
+    }
+
     /// <summary>
     /// Converts a <see cref="FailureDetails" /> to a grpc <see cref="P.TaskFailureDetails" />.
     /// </summary>

src/Worker/Core/Shims/DurableTaskShimFactory.cs

@@ -89,6 +89,29 @@ public TaskOrchestration CreateOrchestration(
         return new TaskOrchestrationShim(context, orchestrator);
     }
 
+    /// <summary>
+    /// Creates a <see cref="TaskOrchestration" /> from a <see cref="ITaskOrchestrator" />.
+    /// </summary>
+    /// <param name="name">
+    /// The name of the orchestration. This should be the name the orchestration was invoked with.
+    /// </param>
+    /// <param name="orchestrator">The orchestration to wrap.</param>
+    /// <param name ="properties">Configuration for the orchestration.</param>
+    /// <param name="parent">The orchestration parent details or <c>null</c> if no parent.</param>
+    /// <returns>A new <see cref="TaskOrchestration" />.</returns>
+    public TaskOrchestration CreateOrchestration(
+        TaskName name,
+        ITaskOrchestrator orchestrator,
+        IReadOnlyDictionary<string, object?> properties,
+        ParentOrchestrationInstance? parent = null)
+    {
+        Check.NotDefault(name);
+        Check.NotNull(orchestrator);
+        Check.NotNull(properties);
+        OrchestrationInvocationContext context = new(name, this.options, this.loggerFactory, parent);
+        return new TaskOrchestrationShim(context, orchestrator, properties);
+    }
+
     /// <summary>
     /// Creates a <see cref="TaskOrchestration" /> from a <see cref="ITaskOrchestrator" />.
     /// </summary>

src/Worker/Core/Shims/TaskOrchestrationContextWrapper.cs

@@ -38,9 +38,27 @@ public TaskOrchestrationContextWrapper(
         OrchestrationContext innerContext,
         OrchestrationInvocationContext invocationContext,
         object? deserializedInput)
+        : this(innerContext, invocationContext, deserializedInput, new Dictionary<string, object?>())
+    {
+    }
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="TaskOrchestrationContextWrapper"/> class.
+    /// </summary>
+    /// <param name="innerContext">The inner orchestration context.</param>
+    /// <param name="invocationContext">The invocation context.</param>
+    /// <param name="deserializedInput">The deserialized input.</param>
+    /// <param name="properties">The configuration for context.</param>
+    public TaskOrchestrationContextWrapper(
+        OrchestrationContext innerContext,
+        OrchestrationInvocationContext invocationContext,
+        object? deserializedInput,
+        IReadOnlyDictionary<string, object?> properties)
     {
         this.innerContext = Check.NotNull(innerContext);
         this.invocationContext = Check.NotNull(invocationContext);
+        this.Properties = Check.NotNull(properties);
+
         this.logger = this.CreateReplaySafeLogger("Microsoft.DurableTask");
         this.deserializedInput = deserializedInput;
     }
@@ -60,6 +78,11 @@ public TaskOrchestrationContextWrapper(
     /// <inheritdoc/>
     public override DateTime CurrentUtcDateTime => this.innerContext.CurrentUtcDateTime;
 
+    /// <summary>
+    /// Gets the configuration settings for the orchestration.
+    /// </summary>
+    public override IReadOnlyDictionary<string, object?> Properties { get; }
+
     /// <inheritdoc/>
     public override TaskOrchestrationEntityFeature Entities
     {