From 209b7798d72b3cd34771e9a926ce2fe69a2a07bd Mon Sep 17 00:00:00 2001
From: Daniel Megyesi <daniel.megyesi@gmail.com>
Date: Wed, 14 Oct 2020 00:48:07 +0200
Subject: [PATCH] feat(argo): add PriorityClass support (#470)

---
 charts/argo/Chart.yaml                                    | 2 +-
 charts/argo/templates/server-deployment.yaml              | 3 +++
 charts/argo/templates/workflow-controller-deployment.yaml | 3 +++
 charts/argo/values.yaml                                   | 8 ++++++++
 4 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/charts/argo/Chart.yaml b/charts/argo/Chart.yaml
index 0d42500e..94cad6c1 100644
--- a/charts/argo/Chart.yaml
+++ b/charts/argo/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v1
 appVersion: v2.11.3
 description: A Helm chart for Argo Workflows
 name: argo
-version: 0.13.0
+version: 0.13.1
 icon: https://raw.githubusercontent.com/argoproj/argo/master/docs/assets/argo.png
 home: https://github.com/argoproj/argo-helm
 maintainers:
diff --git a/charts/argo/templates/server-deployment.yaml b/charts/argo/templates/server-deployment.yaml
index 0d5c8af5..bfaa3e97 100644
--- a/charts/argo/templates/server-deployment.yaml
+++ b/charts/argo/templates/server-deployment.yaml
@@ -91,5 +91,8 @@ spec:
       affinity:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- if .Values.server.priorityClassName }}
+      priorityClassName: {{ .Values.server.priorityClassName }}
+      {{- end }}
 
 {{- end -}}
diff --git a/charts/argo/templates/workflow-controller-deployment.yaml b/charts/argo/templates/workflow-controller-deployment.yaml
index ca80c816..74828caa 100644
--- a/charts/argo/templates/workflow-controller-deployment.yaml
+++ b/charts/argo/templates/workflow-controller-deployment.yaml
@@ -83,3 +83,6 @@ spec:
       affinity:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- if .Values.controller.priorityClassName }}
+      priorityClassName: {{ .Values.controller.priorityClassName }}
+      {{- end }}
diff --git a/charts/argo/values.yaml b/charts/argo/values.yaml
index 710f67ce..8ece66d5 100644
--- a/charts/argo/values.yaml
+++ b/charts/argo/values.yaml
@@ -125,6 +125,10 @@ controller:
     kubernetes.io/os: linux
   tolerations: []
   affinity: {}
+  # Leverage a PriorityClass to ensure your pods survive resource shortages
+  # ref: https://kubernetes.io/docs/concepts/configuration/pod-priority-preemption/
+  # PriorityClass: system-cluster-critical
+  priorityClassName: ""
 
 # executor controls how the init and wait container should be customized
 executor:
@@ -185,6 +189,10 @@ server:
     kubernetes.io/os: linux
   tolerations: []
   affinity: {}
+  # Leverage a PriorityClass to ensure your pods survive resource shortages
+  # ref: https://kubernetes.io/docs/concepts/configuration/pod-priority-preemption/
+  # PriorityClass: system-cluster-critical
+  priorityClassName: ""
 
   # Extra arguments to provide to the Argo server binary.
   extraArgs: []