ac-2247 (#822)

Signed-off-by: Lillie Dae <[email protected]>
Project-MONAI · Jun 5, 2023 · 9439e8c · 9439e8c
1 parent 4d143b7
commit 9439e8c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 13 deletions.
diff --git a/src/WorkflowManager/Common/Services/WorkflowService.cs b/src/WorkflowManager/Common/Services/WorkflowService.cs
@@ -54,11 +54,6 @@ public async Task<string> CreateAsync(Workflow workflow)
         {
             Guard.Against.Null(workflow);
 
-            foreach (var task in workflow.Tasks)
-            {
-                task.Args["workflow_name"] = workflow.Name;
-            }
-
             var id = await _workflowRepository.CreateAsync(workflow);
             _logger.WorkflowCreated(id, workflow.Name);
             return id;
@@ -76,14 +71,6 @@ public async Task<string> CreateAsync(Workflow workflow)
                 return null;
             }
 
-            if (isUpdateToWorkflowName)
-            {
-                foreach (var task in workflow.Tasks)
-                {
-                    task.Args["workflow_name"] = workflow.Name;
-                }
-            }
-
             var result = await _workflowRepository.UpdateAsync(workflow, existingWorkflow);
             _logger.WorkflowUpdated(id, workflow.Name);
             return result;

diff --git a/src/WorkflowManager/WorkflowExecuter/Services/WorkflowExecuterService.cs b/src/WorkflowManager/WorkflowExecuter/Services/WorkflowExecuterService.cs
@@ -743,6 +743,7 @@ private async Task<bool> DispatchTask(WorkflowInstance workflowInstance, Workflo
                     AttachPatientMetaData(taskExec, payload.PatientDetails);
                 }
 
+                taskExec.TaskPluginArguments["workflow_name"] = workflow!.Workflow!.Name;
                 _logger.LogGeneralTaskDispatchInformation(workflowInstance.PayloadId, taskExec.TaskId, workflowInstance.Id, workflow?.Id, JsonConvert.SerializeObject(pathOutputArtifacts));
                 var taskDispatchEvent = EventMapper.ToTaskDispatchEvent(taskExec, workflowInstance, pathOutputArtifacts, correlationId, _storageConfiguration);
                 var jsonMesssage = new JsonMessage<TaskDispatchEvent>(taskDispatchEvent, MessageBrokerConfiguration.WorkflowManagerApplicationId, taskDispatchEvent.CorrelationId, Guid.NewGuid().ToString());