feat(Workflow): add context and wfAction to Execute method

src/Dtmworkflow/Workflow.cs

@@ -17,10 +17,13 @@ public partial class Workflow
 
         public virtual WorkflowImp WorkflowImp { get; set; } = new WorkflowImp();
 
+        public Dictionary<string, object> Context => _context ??= new();
+
         private readonly IDtmClient _httpClient;
         private readonly IDtmgRPCClient _grpcClient;
         private readonly Dtmcli.IBranchBarrierFactory _bbFactory;
         private readonly ILogger _logger;
+        private Dictionary<string, object> _context;
 
         public Workflow(IDtmClient httpClient, IDtmgRPCClient grpcClient, Dtmcli.IBranchBarrierFactory bbFactory, ILogger logger)
         {

src/Dtmworkflow/WorkflowGlobalTransaction.cs

@@ -20,13 +20,20 @@
         }
 
         public async Task<byte[]> Execute(string name, string gid, byte[] data, bool isHttp = true)
+        {
+            return await this.Execute(name, gid, data, null, isHttp);
+        }
+
+        public async Task<byte[]> Execute(string name, string gid, byte[] data, Action<Workflow> wfAction, bool isHttp = true)
         {
             if (!this._handlers.TryGetValue(name, out var handler))
             {
                 throw new DtmCommon.DtmException($"workflow '{name}' not registered. please register at startup");
             }
 
             var wf = _workflowFactory.NewWorkflow(name, gid, data, isHttp);
+            if (wfAction != null)
+                wfAction(wf);
 
             foreach (var fn in handler.Custom)
             {

tests/Dtmgrpc.IntegrationTests/WorkflowGrpcTest.cs

@@ -94,5 +94,32 @@ public async Task Execute_Success()
             status = await ITTestHelper.GetTranStatus(gid);
             Assert.Equal("succeed", status);
         }
+
+
+        [Fact]
+        public async Task ExecuteWithWfAction()
+        {
+            var provider = ITTestHelper.AddDtmGrpc();
+            var workflowFactory = provider.GetRequiredService<IWorkflowFactory>();
+            var loggerFactory = provider.GetRequiredService<ILoggerFactory>();
+            WorkflowGlobalTransaction workflowGlobalTransaction = new WorkflowGlobalTransaction(workflowFactory, loggerFactory);
+            string wfName = Guid.NewGuid().ToString();
+            workflowGlobalTransaction.Register(wfName, (workflow, data) =>
+            {
+                Assert.NotNull(workflow.Context);
+                Assert.Equal(2, workflow.Context.Count);
+                Assert.Equal("value1", workflow.Context["key1-string"]);
+                Assert.Equal(7, workflow.Context["key2-int"]);
+
+                Assert.Equal("input", Encoding.UTF8.GetString(data));
+
+                return Task.FromResult("output"u8.ToArray());
+            });
+            await workflowGlobalTransaction.Execute(wfName, Guid.NewGuid().ToString(), "input"u8.ToArray(), workflow =>
+            {
+                workflow.Context.Add("key1-string", "value1");
+                workflow.Context.Add("key2-int", 7);
+            });
+        }
     }
 }