feat: add pdb support

Author: fe80

charts/mageai/templates/pdb-scheduler.yaml

@@ -0,0 +1,20 @@
+{{- if .Values.scheduler.PodDisruptionBudget.create }}
+---
+apiVersion: policy/v1
+kind: PodDisruptionBudget
+metadata:
+  name: {{ include "mageai.fullname" . }}
+  labels:
+    {{- include "mageai.labels" . | nindent 4 }}
+spec:
+   {{- if .Values.scheduler.PodDisruptionBudget.minAvailable }}
+  minAvailable: {{ .Values.scheduler.PodDisruptionBudget.minAvailable }}
+  {{- end }}
+  {{- if or .Values.scheduler.PodDisruptionBudget.maxUnavailable ( not .Values.scheduler.PodDisruptionBudget.minAvailable ) }}
+  maxUnavailable: {{ .Values.scheduler.PodDisruptionBudget.maxUnavailable | default 1 }}
+  {{- end }}
+  selector:
+    matchLabels:
+      {{- include "mageai.schedulerSelectorLabels" . | nindent 6 }}
+...
+{{- end }}

charts/mageai/templates/pdb.yaml

@@ -0,0 +1,20 @@
+{{- if .Values.webServer.PodDisruptionBudget.create }}
+---
+apiVersion: policy/v1
+kind: PodDisruptionBudget
+metadata:
+  name: {{ include "mageai.fullname" . }}
+  labels:
+    {{- include "mageai.labels" . | nindent 4 }}
+spec:
+   {{- if .Values.webServer.PodDisruptionBudget.minAvailable }}
+  minAvailable: {{ .Values.webServer.PodDisruptionBudget.minAvailable }}
+  {{- end }}
+  {{- if or .Values.webServer.PodDisruptionBudget.maxUnavailable ( not .Values.webServer.PodDisruptionBudget.minAvailable ) }}
+  maxUnavailable: {{ .Values.webServer.PodDisruptionBudget.maxUnavailable | default 1 }}
+  {{- end }}
+  selector:
+    matchLabels:
+      {{- include "mageai.selectorLabels" . | nindent 6 }}
+...
+{{- end }}

charts/mageai/values.yaml

@@ -24,6 +24,9 @@ scheduler:
   #   minReplicas: 1
   #   maxReplicas: 10
   #   targetCPUUtilizationPercentage: 50
+  PodDisruptionBudget:
+    create: false
+    maxUnavailable: "50%"
 
 # Effective if standaloneScheduler is true
 webServer:
@@ -44,6 +47,9 @@ webServer:
   #   minReplicas: 1
   #   maxReplicas: 10
   #   targetCPUUtilizationPercentage: 50
+  PodDisruptionBudget:
+    create: false
+    maxUnavailable: 50%
 
 # Enable Postgres as the DB
 postgresql: