From 9360cb5b66f2669ba857d1cec7abbc05b54f6c3e Mon Sep 17 00:00:00 2001
From: Adrian Pauli <adrian.pauli@qumea.com>
Date: Fri, 2 Feb 2024 15:53:28 +0100
Subject: [PATCH] feat(helm): possibility to add `priorityClassName` in helm
 chart for the pods

---
 deploy/charts/emqx-enterprise/templates/StatefulSet.yaml | 3 +++
 deploy/charts/emqx-enterprise/values.yaml                | 3 +++
 deploy/charts/emqx/templates/StatefulSet.yaml            | 3 +++
 deploy/charts/emqx/values.yaml                           | 3 +++
 4 files changed, 12 insertions(+)

diff --git a/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml b/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml
index 1eba3d1ba..195bd698e 100644
--- a/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml
+++ b/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml
@@ -56,6 +56,9 @@ spec:
       {{- end }}
     spec:
       serviceAccountName: {{ include "emqx.serviceAccountName" . }}
+      {{- if .Values.priorityClassName }}
+      priorityClassName: {{ .Values.priorityClassName }}
+      {{- end }}
       volumes:
       {{- if .Values.ssl.enabled }}
       - name: ssl-cert
diff --git a/deploy/charts/emqx-enterprise/values.yaml b/deploy/charts/emqx-enterprise/values.yaml
index 9969f23cd..64970de9d 100644
--- a/deploy/charts/emqx-enterprise/values.yaml
+++ b/deploy/charts/emqx-enterprise/values.yaml
@@ -38,6 +38,9 @@ recreatePods: false
 ## Sets the minReadySeconds parameter on the stateful set. This can be used to add delay between restart / updates between the single pods.
 minReadySeconds:
 
+## Sets the priorityClassName parameter on the pods. This can be used to run the pods with increased priority.
+priorityClassName:
+
 clusterDomain: cluster.local
 
 podAnnotations: {}
diff --git a/deploy/charts/emqx/templates/StatefulSet.yaml b/deploy/charts/emqx/templates/StatefulSet.yaml
index 4f3e310c7..85caa4297 100644
--- a/deploy/charts/emqx/templates/StatefulSet.yaml
+++ b/deploy/charts/emqx/templates/StatefulSet.yaml
@@ -56,6 +56,9 @@ spec:
       {{- end }}
     spec:
       serviceAccountName: {{ include "emqx.serviceAccountName" . }}
+      {{- if .Values.priorityClassName }}
+      priorityClassName: {{ .Values.priorityClassName }}
+      {{- end }}
       volumes:
       {{- if .Values.ssl.enabled }}
       - name: ssl-cert
diff --git a/deploy/charts/emqx/values.yaml b/deploy/charts/emqx/values.yaml
index 2a2b91299..f87f426f3 100644
--- a/deploy/charts/emqx/values.yaml
+++ b/deploy/charts/emqx/values.yaml
@@ -38,6 +38,9 @@ recreatePods: false
 ## Sets the minReadySeconds parameter on the stateful set. This can be used to add delay between restart / updates between the single pods.
 minReadySeconds:
 
+## Sets the priorityClassName parameter on the pods. This can be used to run the pods with increased priority.
+priorityClassName:
+
 clusterDomain: cluster.local
 
 podAnnotations: {}