Read TaskPriorityClass from config

Signed-off-by: Victor Chang <[email protected]>
Project-MONAI · Sep 6, 2023 · 556f373 · 556f373
1 parent 0aca28d
commit 556f373
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/src/Common/Configuration/TaskManagerConfiguration.cs b/src/Common/Configuration/TaskManagerConfiguration.cs
@@ -74,5 +74,8 @@ public class ArgoPluginArguments
 
         [ConfigurationKeyName("messageSenderContainerMemoryLimit")]
         public string MessageSenderContainerMemoryLimit { get; set; } = "500Mi";
+
+        [ConfigurationKeyName("taskPriorityClass")]
+        public string TaskPriorityClass { get; set; } = string.Empty;
     }
 }
diff --git a/src/TaskManager/Plug-ins/Argo/ArgoPlugin.cs b/src/TaskManager/Plug-ins/Argo/ArgoPlugin.cs
@@ -432,7 +432,7 @@ private async Task<Workflow> BuildWorkflowWrapper(CancellationToken cancellation
         private void ProcessTaskPluginArguments(Workflow workflow)
         {
             Guard.Against.Null(workflow, nameof(workflow));
-            var priorityClassName = Event.GetTaskPluginArgumentsParameter(ArgoParameters.TaskPriorityClassName) ?? "standard";
+            var priorityClassName = Event.GetTaskPluginArgumentsParameter(ArgoParameters.TaskPriorityClassName) ?? _options.Value.TaskManager.ArgoPluginArguments.TaskPriorityClass;
 
             foreach (var template in workflow.Spec.Templates)
             {

diff --git a/src/TaskManager/TaskManager/appsettings.json b/src/TaskManager/TaskManager/appsettings.json
@@ -59,7 +59,8 @@
         "messageGeneratorContainerCpuLimit": "1",
         "messageGeneratorContainerMemoryLimit": "500Mi",
         "messageSenderContainerCpuLimit": "1",
-        "messageSenderContainerMemoryLimit": "500Mi"
+        "messageSenderContainerMemoryLimit": "500Mi",
+        "taskPriorityClass":  ""
       },
       "argoExitHookSendMessageContainerImage": "ghcr.io/project-monai/monai-deploy-task-manager-callback:0.2.0-beta.211"
     },