Read TaskPriorityClass from config (#872)

* Read TaskPriorityClass from config

Signed-off-by: Victor Chang <vicchang@nvidia.com>

Author: mocsharp

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;
     }
 }

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)
             {

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"
     },

tests/UnitTests/TaskManager.Argo.Tests/ArgoPluginTestBase.cs

@@ -100,6 +100,7 @@ public ArgoPluginTestBase()
             Options.Value.TaskManager.ArgoPluginArguments.MessageGeneratorContainerMemoryLimit = MessageGeneratorContainerMemoryLimit;
             Options.Value.TaskManager.ArgoPluginArguments.MessageSenderContainerCpuLimit = MessageSenderContainerCpuLimit;
             Options.Value.TaskManager.ArgoPluginArguments.MessageSenderContainerMemoryLimit = MessageSenderContainerMemoryLimit;
+            Options.Value.TaskManager.ArgoPluginArguments.TaskPriorityClass = "standard";
         }
     }
 }