From b0f544489751b8752f8b9e9876e57c7353dc7ed5 Mon Sep 17 00:00:00 2001
From: Ryan Luckie <rtluckie@users.noreply.github.com>
Date: Sun, 15 Jul 2018 12:48:56 -0500
Subject: [PATCH] [incubator/zookeeper] Add support for priorityClasses (#6608)

- add support for priorityClasses
---
 incubator/zookeeper/Chart.yaml                 | 2 +-
 incubator/zookeeper/templates/job-chroots.yaml | 3 +++
 incubator/zookeeper/templates/statefulset.yaml | 3 +++
 incubator/zookeeper/values.yaml                | 2 ++
 4 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/incubator/zookeeper/Chart.yaml b/incubator/zookeeper/Chart.yaml
index 01e81736f6..9959f29ce9 100644
--- a/incubator/zookeeper/Chart.yaml
+++ b/incubator/zookeeper/Chart.yaml
@@ -1,6 +1,6 @@
 name: zookeeper
 home: https://zookeeper.apache.org/
-version: 1.0.2
+version: 1.1.0
 appVersion: 3.4.10
 description: Centralized service for maintaining configuration information, naming,
   providing distributed synchronization, and providing group services.
diff --git a/incubator/zookeeper/templates/job-chroots.yaml b/incubator/zookeeper/templates/job-chroots.yaml
index 6663ddb8ff..29ec3bd063 100644
--- a/incubator/zookeeper/templates/job-chroots.yaml
+++ b/incubator/zookeeper/templates/job-chroots.yaml
@@ -30,6 +30,9 @@ spec:
         job: chroots
     spec:
       restartPolicy: {{ $job.restartPolicy }}
+{{- if .Values.priorityClassName }}
+      priorityClassName: "{{ .Values.priorityClassName }}"
+{{- end }}
       containers:
         - name: main
           image: "{{ .Values.image.repository }}:{{ .Values.image.tag }}"
diff --git a/incubator/zookeeper/templates/statefulset.yaml b/incubator/zookeeper/templates/statefulset.yaml
index bc2d160cdc..14d348fb5b 100644
--- a/incubator/zookeeper/templates/statefulset.yaml
+++ b/incubator/zookeeper/templates/statefulset.yaml
@@ -44,6 +44,9 @@ spec:
 {{- end }}
       securityContext:
 {{ toYaml .Values.securityContext | indent 8 }}
+{{- if .Values.priorityClassName }}
+      priorityClassName: "{{ .Values.priorityClassName }}"
+{{- end }}
       containers:
 
         - name: zookeeper
diff --git a/incubator/zookeeper/values.yaml b/incubator/zookeeper/values.yaml
index 35f5e98cec..660b033dbd 100644
--- a/incubator/zookeeper/values.yaml
+++ b/incubator/zookeeper/values.yaml
@@ -62,6 +62,8 @@ resources: {}  # Optionally specify how much CPU and memory (RAM) each zookeeper
   #  cpu: 100m
   #  memory: 128Mi
 
+priorityClassName: ""
+
 nodeSelector: {}  # Node label-values required to run zookeeper pods.
 
 tolerations: []  # Node taint overrides for zookeeper pods.