fix configuration

microsoft · Jan 29, 2025 · 7f3fc0a · 7f3fc0a
1 parent d2400cd
commit 7f3fc0a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 9 deletions.
diff --git a/src/Client/AzureManaged/DurableTaskSchedulerClientExtensions.cs b/src/Client/AzureManaged/DurableTaskSchedulerClientExtensions.cs
@@ -85,12 +85,15 @@ static void ConfigureSchedulerOptions(
             .Configure(additionalConfig ?? (_ => { }))
             .ValidateDataAnnotations();
 
+        builder.Services.AddOptions<DurableTaskClientOptions>(builder.Name)
+            .Configure(options =>
+            {
+                options.EnableEntitySupport = true;
+            });
+
         builder.Services.TryAddEnumerable(
             ServiceDescriptor.Singleton<IConfigureOptions<GrpcDurableTaskClientOptions>, ConfigureGrpcChannel>());
-        builder.UseGrpc((GrpcDurableTaskClientOptions options) =>
-        {
-            options.EnableEntitySupport = true;
-        });
+        builder.UseGrpc(_ => { });
     }
 
     /// <summary>

diff --git a/src/Worker/AzureManaged/DurableTaskSchedulerWorkerExtensions.cs b/src/Worker/AzureManaged/DurableTaskSchedulerWorkerExtensions.cs
@@ -85,13 +85,15 @@ static void ConfigureSchedulerOptions(
             .Configure(additionalConfig ?? (_ => { }))
             .ValidateDataAnnotations();
 
+        builder.Services.AddOptions<DurableTaskWorkerOptions>(builder.Name)
+           .Configure(options =>
+           {
+               options.EnableEntitySupport = true;
+           });
+
         builder.Services.TryAddEnumerable(
             ServiceDescriptor.Singleton<IConfigureOptions<GrpcDurableTaskWorkerOptions>, ConfigureGrpcChannel>());
-        builder.UseGrpc((GrpcDurableTaskWorkerOptions options) =>
-        {
-            options.EnableEntitySupport = true;
-            options.ConvertOrchestrationEntityEvents = true;
-        });
+        builder.UseGrpc(_ => { });
     }
 
     /// <summary>
@@ -117,6 +119,9 @@ public void Configure(string? name, GrpcDurableTaskWorkerOptions options)
         {
             DurableTaskSchedulerWorkerOptions source = schedulerOptions.Get(name ?? Options.DefaultName);
             options.Channel = source.CreateChannel();
+
+            options.ConvertOrchestrationEntityEvents = true;
+            options.InsertEntityUnlocksOnCompletion = true;
         }
     }
 }