From 16a263e09b4f20d6b80790bcb17082a558618945 Mon Sep 17 00:00:00 2001
From: Witold Duranek <contact@witalis.net>
Date: Mon, 22 Apr 2024 10:10:29 +0200
Subject: [PATCH] fix(charts): extend pod topology spread constraints

---
 .../emqx-enterprise/templates/StatefulSet.yaml    | 15 +++++++++++++++
 deploy/charts/emqx/templates/StatefulSet.yaml     | 15 +++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml b/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml
index 195bd698e..c3f82c647 100644
--- a/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml
+++ b/deploy/charts/emqx-enterprise/templates/StatefulSet.yaml
@@ -184,5 +184,20 @@ spec:
             matchLabels:
               app.kubernetes.io/name: {{ include "emqx.name" $ }}
               app.kubernetes.io/instance: {{ $.Release.Name }}
+        {{- if .minDomains }}
+          minDomains: {{ .minDomains }}
+        {{- end }}
+        {{- if .matchLabelKeys }}
+          matchLabelKeys:
+            {{- range .matchLabelKeys }}
+            - {{ . }}
+            {{- end }}
+        {{- end }}
+        {{- if .nodeAffinityPolicy }}
+          nodeAffinityPolicy: {{ .nodeAffinityPolicy }}
+        {{- end }}
+        {{- if .nodeTaintsPolicy }}
+          nodeTaintsPolicy: {{ .nodeTaintsPolicy }}
+        {{- end }}
         {{- end }}
     {{- end }}
diff --git a/deploy/charts/emqx/templates/StatefulSet.yaml b/deploy/charts/emqx/templates/StatefulSet.yaml
index 85caa4297..ea7656a95 100644
--- a/deploy/charts/emqx/templates/StatefulSet.yaml
+++ b/deploy/charts/emqx/templates/StatefulSet.yaml
@@ -184,5 +184,20 @@ spec:
             matchLabels:
               app.kubernetes.io/name: {{ include "emqx.name" $ }}
               app.kubernetes.io/instance: {{ $.Release.Name }}
+        {{- if .minDomains }}
+          minDomains: {{ .minDomains }}
+        {{- end }}
+        {{- if .matchLabelKeys }}
+          matchLabelKeys:
+            {{- range .matchLabelKeys }}
+            - {{ . }}
+            {{- end }}
+        {{- end }}
+        {{- if .nodeAffinityPolicy }}
+          nodeAffinityPolicy: {{ .nodeAffinityPolicy }}
+        {{- end }}
+        {{- if .nodeTaintsPolicy }}
+          nodeTaintsPolicy: {{ .nodeTaintsPolicy }}
+        {{- end }}
         {{- end }}
     {{- end }}