enhance

Author: YunchuWang

src/Abstractions/Entities/TaskEntityContext.cs

@@ -60,9 +60,12 @@ public abstract string ScheduleNewOrchestration(
     /// <param name="name">The name of the orchestration to start.</param>
     /// <param name="input">The input for the orchestration.</param>
     /// <param name="options">The options for starting the orchestration.</param>
-    /// <returns>The instance id for the new orchestration.</returns>
-    public virtual async Task<string> ScheduleNewOrchestrationAsync(TaskName name, object? input = null, StartOrchestrationOptions? options = null)
+    /// <returns>A task representing the asynchronous operation.</returns>
+    public virtual Task<string> ScheduleNewOrchestrationAsync(
+        TaskName name,
+        object? input = null,
+        StartOrchestrationOptions? options = null)
     {
-        return await Task.FromResult(this.ScheduleNewOrchestration(name, input, options));
+        return Task.FromResult(this.ScheduleNewOrchestration(name, input, options));
     }
 }

src/Abstractions/Entities/TaskEntityState.cs

@@ -59,4 +59,15 @@ public abstract class TaskEntityState
     /// </summary>
     /// <param name="state">The state to set.</param>
     public abstract void SetState(object? state);
+
+    /// <summary>
+    /// Asynchronously sets the entity state. Setting of <c>null</c> will delete entity state.
+    /// </summary>
+    /// <param name="state">The state to set.</param>
+    /// <returns>A task representing the asynchronous operation.</returns>
+    public virtual Task SetStateAsync(object? state)
+    {
+        this.SetState(state);
+        return Task.CompletedTask;
+    }
 }

src/Worker/Core/Shims/TaskEntityShim.cs

@@ -190,7 +190,13 @@ public void Reset()
             return this.cachedValue;
         }
 
-        public override async void SetState(object? state)
+        public override void SetState(object? state)
+        {
+            this.value = this.dataConverter.Serialize(state);
+            this.cachedValue = state;
+        }
+
+        public override async Task SetStateAsync(object? state)
         {
             this.value = this.enableLargePayloadSupport
                 ? await this.dataConverter.SerializeAsync(state)
@@ -281,7 +287,10 @@ public override string ScheduleNewOrchestration(TaskName name, object? input = n
             return instanceId;
         }
 
-        public override async Task<string> ScheduleNewOrchestrationAsync(TaskName name, object? input = null, StartOrchestrationOptions? options = null)
+        public override async Task<string> ScheduleNewOrchestrationAsync(
+            TaskName name,
+            object? input = null,
+            StartOrchestrationOptions? options = null)
         {
             Check.NotEntity(true, options?.InstanceId);
 

test/Abstractions.Tests/Entities/Mocks/TestEntityState.cs

@@ -28,4 +28,10 @@ public override void SetState(object? state)
     {
         this.State = state;
     }
+
+    public override Task SetStateAsync(object? state)
+    {
+        this.SetState(state);
+        return Task.CompletedTask;
+    }
 }