From c6ab1d18e3a6368a9db59d97c4eaad169e6eb468 Mon Sep 17 00:00:00 2001
From: Marvin Cai <zxc@streamnative.io>
Date: Mon, 20 Jun 2022 07:59:43 -0700
Subject: [PATCH] Support defining extra env for broker and proxy statefulsset.
 (#273)

---
 charts/pulsar/templates/broker-statefulset.yaml |  4 ++++
 charts/pulsar/templates/proxy-statefulset.yaml  |  4 ++++
 charts/pulsar/values.yaml                       | 14 +++++++++++++-
 3 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/charts/pulsar/templates/broker-statefulset.yaml b/charts/pulsar/templates/broker-statefulset.yaml
index 1b56d95..333ac54 100644
--- a/charts/pulsar/templates/broker-statefulset.yaml
+++ b/charts/pulsar/templates/broker-statefulset.yaml
@@ -227,6 +227,10 @@ spec:
         - name: "{{ .Values.tlsPrefix }}pulsarssl"
           containerPort: {{ .Values.broker.ports.pulsarssl }}
         {{- end }}
+{{- if .Values.broker.extreEnvs }}
+        env:
+{{ toYaml .Values.broker.extreEnvs | indent 8 }}
+{{- end }}
         envFrom:
         - configMapRef:
             name: "{{ template "pulsar.fullname" . }}-{{ .Values.broker.component }}"
diff --git a/charts/pulsar/templates/proxy-statefulset.yaml b/charts/pulsar/templates/proxy-statefulset.yaml
index 1dc739b..a70811c 100644
--- a/charts/pulsar/templates/proxy-statefulset.yaml
+++ b/charts/pulsar/templates/proxy-statefulset.yaml
@@ -202,6 +202,10 @@ spec:
         securityContext:
           readOnlyRootFilesystem: false
       {{- end }}
+{{- if .Values.proxy.extreEnvs }}
+        env:
+{{ toYaml .Values.proxy.extreEnvs | indent 8 }}
+{{- end }}
         envFrom:
         - configMapRef:
             name: "{{ template "pulsar.fullname" . }}-{{ .Values.proxy.component }}"
diff --git a/charts/pulsar/values.yaml b/charts/pulsar/values.yaml
index 4c6b5c4..ce6cce0 100644
--- a/charts/pulsar/values.yaml
+++ b/charts/pulsar/values.yaml
@@ -740,6 +740,12 @@ broker:
   #     readOnly: true
   extraVolumes: []
   extraVolumeMounts: []
+  extreEnvs: []
+#    - name: POD_NAME
+#      valueFrom:
+#        fieldRef:
+#          apiVersion: v1
+#          fieldPath: metadata.name
   ## Broker configmap
   ## templates/broker-configmap.yaml
   ##
@@ -856,6 +862,12 @@ proxy:
   #     readOnly: true
   extraVolumes: []
   extraVolumeMounts: []
+  extreEnvs: []
+#    - name: POD_IP
+#      valueFrom:
+#        fieldRef:
+#          apiVersion: v1
+#          fieldPath: status.podIP
   ## Proxy configmap
   ## templates/proxy-configmap.yaml
   ##
@@ -1162,4 +1174,4 @@ pulsar_manager:
 job:
   ttl:
     enabled: false
-    secondsAfterFinished: 3600
\ No newline at end of file
+    secondsAfterFinished: 3600