feat: add topologySpreadConstraints

Signed-off-by: fe80 <steffyfort@gmail.com>

Author: fe80

charts/mageai/templates/deployment.yaml

@@ -139,6 +139,17 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.topologySpreadConstraints }}
+      topologySpreadConstraints:
+        {{- range $constraint := . }}
+        - {{ toYaml $constraint | nindent 10 | trim }}
+            {{- if not $constraint.labelSelector }}
+          labelSelector:
+            matchLabels:
+              {{- include "mageai.selectorLabels" $ | nindent 14 }}
+            {{- end }}
+        {{- end }}
+      {{- end }}
       volumes:
       {{- if .Values.volumes }}
         {{- toYaml .Values.volumes | nindent 8 }}

charts/mageai/templates/scheduler.yaml

@@ -110,6 +110,17 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.scheduler.topologySpreadConstraints }}
+      topologySpreadConstraints:
+        {{- range $constraint := . }}
+        - {{ toYaml $constraint | nindent 10 | trim }}
+            {{- if not $constraint.labelSelector }}
+          labelSelector:
+            matchLabels:
+              {{- include "mageai.schedulerSelectorLabels" $ | nindent 14 }}
+            {{- end }}
+        {{- end }}
+      {{- end }}
       volumes:
       {{- if .Values.volumes }}
         {{- toYaml .Values.volumes | nindent 8 }}

charts/mageai/templates/webservice.yaml

@@ -138,6 +138,17 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.webServer.topologySpreadConstraints }}
+      topologySpreadConstraints:
+        {{- range $constraint := . }}
+        - {{ toYaml $constraint | nindent 10 | trim }}
+            {{- if not $constraint.labelSelector }}
+          labelSelector:
+            matchLabels:
+              {{- include "mageai.selectorLabels" $ | nindent 14 }}
+            {{- end }}
+        {{- end }}
+      {{- end }}
       volumes:
       {{- if .Values.volumes }}
         {{- toYaml .Values.volumes | nindent 8 }}

charts/mageai/values.yaml

@@ -25,6 +25,18 @@ scheduler:
   #   maxReplicas: 10
   #   targetCPUUtilizationPercentage: 50
 
+  topologySpreadConstraints: []
+  #   - maxSkew: 1
+  #     topologyKey: topology.kubernetes.io/zone
+  #     whenUnsatisfiable: ScheduleAnyway
+  #     matchLabelKeys:
+  #       - pod-template-hash
+  #   - maxSkew: 1
+  #     topologyKey: kubernetes.io/hostname
+  #     whenUnsatisfiable: DoNotSchedule
+  #     matchLabelKeys:
+  #       - pod-template-hash
+
 # Effective if standaloneScheduler is true
 webServer:
   replicaCount: 1
@@ -44,6 +56,17 @@ webServer:
   #   minReplicas: 1
   #   maxReplicas: 10
   #   targetCPUUtilizationPercentage: 50
+  topologySpreadConstraints: []
+  #   - maxSkew: 1
+  #     topologyKey: topology.kubernetes.io/zone
+  #     whenUnsatisfiable: ScheduleAnyway
+  #     matchLabelKeys:
+  #       - pod-template-hash
+  #   - maxSkew: 1
+  #     topologyKey: kubernetes.io/hostname
+  #     whenUnsatisfiable: DoNotSchedule
+  #     matchLabelKeys:
+  #       - pod-template-hash
 
 # Enable Postgres as the DB
 postgresql:
@@ -174,6 +197,18 @@ nodeSelector: {}
 
 tolerations: []
 
+topologySpreadConstraints:
+  - maxSkew: 1
+    topologyKey: topology.kubernetes.io/zone
+    whenUnsatisfiable: ScheduleAnyway
+    matchLabelKeys:
+      - pod-template-hash
+  - maxSkew: 1
+    topologyKey: kubernetes.io/hostname
+    whenUnsatisfiable: DoNotSchedule
+    matchLabelKeys:
+      - pod-template-hash
+
 affinity: {}
 
 extraVolumeMounts: