implement entity support for DTS (squashed commit and rebased) (#369)

Author: sebastianburckhardt

src/Client/AzureManaged/DurableTaskSchedulerClientExtensions.cs

@@ -85,6 +85,11 @@ static void ConfigureSchedulerOptions(
             .Configure(additionalConfig ?? (_ => { }))
             .ValidateDataAnnotations();
 
+        builder.Configure(options =>
+        {
+            options.EnableEntitySupport = true;
+        });
+
         builder.Services.TryAddEnumerable(
             ServiceDescriptor.Singleton<IConfigureOptions<GrpcDurableTaskClientOptions>, ConfigureGrpcChannel>());
         builder.UseGrpc(_ => { });

src/Grpc/orchestrator_service.proto

@@ -288,6 +288,15 @@ message TerminateOrchestrationAction {
     bool recurse = 3;
 }
 
+message SendEntityMessageAction {
+    oneof EntityMessageType {
+        EntityOperationSignaledEvent entityOperationSignaled = 1;
+        EntityOperationCalledEvent entityOperationCalled = 2;
+        EntityLockRequestedEvent entityLockRequested = 3;
+        EntityUnlockSentEvent entityUnlockSent = 4;
+    }
+}
+
 message OrchestratorAction {
     int32 id = 1;
     oneof orchestratorActionType {
@@ -297,6 +306,7 @@ message OrchestratorAction {
         SendEventAction sendEvent = 5;
         CompleteOrchestrationAction completeOrchestration = 6;
         TerminateOrchestrationAction terminateOrchestration = 7;
+        SendEntityMessageAction sendEntityMessage = 8;
     }
 }
 
@@ -329,6 +339,7 @@ message CreateInstanceRequest {
     OrchestrationIdReusePolicy orchestrationIdReusePolicy = 6;
     google.protobuf.StringValue executionId = 7;
     map<string, string> tags = 8;
+    TraceContext parentTraceContext = 9;
 }
 
 message OrchestrationIdReusePolicy {
@@ -550,6 +561,8 @@ message EntityBatchResult {
     repeated OperationAction actions = 2;
     google.protobuf.StringValue entityState = 3;
     TaskFailureDetails failureDetails = 4;
+    string completionToken = 5;
+    repeated OperationInfo operationInfos = 6; // used only with DTS
 }
 
 message EntityRequest {
@@ -572,6 +585,11 @@ message OperationResult {
     }
 }
 
+message OperationInfo {
+    string requestId = 1;
+    OrchestrationInstance responseDestination = 2; // null for signals
+}
+
 message OperationResultSuccess {
     google.protobuf.StringValue result = 1;
 }

src/Grpc/versions.txt

@@ -1,2 +1,2 @@
-# The following files were downloaded from branch main at 2025-01-30 00:06:14 UTC
-https://raw.githubusercontent.com/microsoft/durabletask-protobuf/c672a0dc97c06587d7399ee12f1c5b0b9fc492a7/protos/orchestrator_service.proto
+# The following files were downloaded from branch main at 2025-02-11 19:40:58 UTC
+https://raw.githubusercontent.com/microsoft/durabletask-protobuf/589cb5ecd9dd4b1fe463750defa3e2c84276b079/protos/orchestrator_service.proto