Adds topologySpreadConstraints to the statefulset (#58)

* Adds topologySpreadConstraints to the statefulset. Adds topologySpreadConstraints to the statefulset so that users can ensure that keydb instances are placed in different AZs. * Fixes label selector * Fixes indentation
Enapter · Oct 17, 2022 · 39d816b · 39d816b
1 parent a5e2363
commit 39d816b
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/keydb/templates/sts.yaml b/keydb/templates/sts.yaml
@@ -14,6 +14,26 @@ spec:
   selector:
     matchLabels:
       {{- include "keydb.selectorLabels" . | nindent 6 }}
+  {{- if .Values.topologySpreadConstraints }}
+  topologySpreadConstraints:
+    {{- range .Values.topologySpreadConstraints }}
+    - labelSelector:
+        matchLabels:
+          {{- include "keydb.selectorLabels" $ | nindent 10 }}
+      topologyKey: {{ default "topology.kubernetes.io/zone" .topologyKey }}
+      maxSkew: {{ .maxSkew }}
+      {{- if .minDomains }}
+      minDomains: {{ .minDomains }}
+      {{- end }}
+      whenUnsatisfiable: {{ default "DoNotSchedule" .whenUnsatisfiable }}
+      {{- if .nodeAffinityPolicy }}
+      nodeAffinityPolicy: {{ .nodeAffinityPolicy }}
+      {{- end }}
+      {{- if .nodeTaintsPolicy }}
+      nodeTaintsPolicy: {{ .nodeTaintsPolicy }}
+      {{- end }}
+    {{- end }}
+  {{- end }}
   template:
     metadata:
       annotations:

diff --git a/keydb/values.yaml b/keydb/values.yaml
@@ -40,6 +40,15 @@ tolerations: {}
 nodeSelector: {}
   # topology.kubernetes.io/region: some-region
 
+topologySpreadConstraints: []
+# - maxSkew: 1
+#   ## Optional keys
+#   # whenUnsatisfiable: DoNotSchedule
+#   # topologyKey: "topology.kubernetes.io/zone"
+#   # minDomains: 0
+#   # nodeAffinityPolicy: Honor
+#   # nodeTaintsPolicy: Honor
+
 affinity:
   podAntiAffinity:
     preferredDuringSchedulingIgnoredDuringExecution: