From c9b1c399ade2bf940ddb43aeb16ec1e4ad9faed3 Mon Sep 17 00:00:00 2001
From: dntosas <ntosas@gmail.com>
Date: Tue, 9 Feb 2021 12:12:40 +0200
Subject: [PATCH] [chart] Ability to add priorityClassName

CSI components and especially the node one, may be critical for
operators so in this commit we enable them to define priorities for this
kind of Pods.

Signed-off-by: dntosas <ntosas@gmail.com>
---
 charts/aws-ebs-csi-driver/templates/controller.yaml | 1 +
 charts/aws-ebs-csi-driver/templates/node.yaml       | 1 +
 charts/aws-ebs-csi-driver/values.yaml               | 4 ++--
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/charts/aws-ebs-csi-driver/templates/controller.yaml b/charts/aws-ebs-csi-driver/templates/controller.yaml
index 43b5b082cd..160f11bd06 100644
--- a/charts/aws-ebs-csi-driver/templates/controller.yaml
+++ b/charts/aws-ebs-csi-driver/templates/controller.yaml
@@ -21,6 +21,7 @@ spec:
       annotations: {{ toYaml .Values.podAnnotations | nindent 8 }}
       {{- end }}
     spec:
+      priorityClassName: "{{ .Values.priorityClassName }}"
       nodeSelector:
         kubernetes.io/os: linux
         {{- with .Values.nodeSelector }}
diff --git a/charts/aws-ebs-csi-driver/templates/node.yaml b/charts/aws-ebs-csi-driver/templates/node.yaml
index 17dd1ac962..f14c7f13c6 100644
--- a/charts/aws-ebs-csi-driver/templates/node.yaml
+++ b/charts/aws-ebs-csi-driver/templates/node.yaml
@@ -20,6 +20,7 @@ spec:
       annotations: {{ toYaml .Values.node.podAnnotations | nindent 8 }}
       {{- end }}
     spec:
+      priorityClassName: "{{ .Values.node.priorityClassName }}"
       affinity:
         nodeAffinity:
           requiredDuringSchedulingIgnoredDuringExecution:
diff --git a/charts/aws-ebs-csi-driver/values.yaml b/charts/aws-ebs-csi-driver/values.yaml
index adf3d000f0..0017abb363 100644
--- a/charts/aws-ebs-csi-driver/values.yaml
+++ b/charts/aws-ebs-csi-driver/values.yaml
@@ -60,11 +60,10 @@ resources:
   #   cpu: 100m
   #   memory: 128Mi
 
+priorityClassName: ""
 nodeSelector: {}
-
 tolerateAllTaints: true
 tolerations: []
-
 affinity: {}
 
 # Extra volume tags to attach to each dynamically provisioned volume.
@@ -87,6 +86,7 @@ k8sTagClusterId: ""
 region: ""
 
 node:
+  priorityClassName: ""
   nodeSelector: {}
   podAnnotations: {}
   tolerateAllTaints: true