Merge pull request #283 from conductor-oss/sub_workflow_priority

Support priority override for sub workflows
conductor-oss · Oct 14, 2024 · 76ed504 · 76ed504
2 parents a28e0e9 + 16a7c07
commit 76ed504
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/common/src/main/java/com/netflix/conductor/common/metadata/workflow/SubWorkflowParams.java b/common/src/main/java/com/netflix/conductor/common/metadata/workflow/SubWorkflowParams.java
@@ -45,6 +45,9 @@ public class SubWorkflowParams {
 
     private IdempotencyStrategy idempotencyStrategy;
 
+    // Priority of the sub workflow, not set inherits from the parent
+    private Integer priority;
+
     public String getIdempotencyKey() {
         return idempotencyKey;
     }
@@ -61,6 +64,14 @@ public void setIdempotencyStrategy(IdempotencyStrategy idempotencyStrategy) {
         this.idempotencyStrategy = idempotencyStrategy;
     }
 
+    public Integer getPriority() {
+        return priority;
+    }
+
+    public void setPriority(Integer priority) {
+        this.priority = priority;
+    }
+
     /**
      * @return the name
      */

diff --git a/core/src/main/java/com/netflix/conductor/core/execution/mapper/SubWorkflowTaskMapper.java b/core/src/main/java/com/netflix/conductor/core/execution/mapper/SubWorkflowTaskMapper.java
@@ -83,6 +83,9 @@ public List<TaskModel> getMappedTasks(TaskMapperContext taskMapperContext) {
         subWorkflowTask.addInput("workflowInput", taskMapperContext.getTaskInput());
         subWorkflowTask.setStatus(TaskModel.Status.SCHEDULED);
         subWorkflowTask.setCallbackAfterSeconds(workflowTask.getStartDelay());
+        if (subWorkflowParams.getPriority() != null) {
+            subWorkflowTask.setWorkflowPriority(subWorkflowParams.getPriority());
+        }
         LOGGER.debug("SubWorkflowTask {} created to be Scheduled", subWorkflowTask);
         return List.of(subWorkflowTask);
     }