From f72f9150d68b0c282d7748e9e878d8345410bdc2 Mon Sep 17 00:00:00 2001
From: Drew Sirenko <68304519+AndrewSirenko@users.noreply.github.com>
Date: Mon, 12 Aug 2024 12:10:27 -0400
Subject: [PATCH] Add toggle for PodDisruptionBudget in chart

---
 .../templates/poddisruptionbudget-controller.yaml             | 4 ++--
 charts/aws-ebs-csi-driver/values.yaml                         | 3 +++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/charts/aws-ebs-csi-driver/templates/poddisruptionbudget-controller.yaml b/charts/aws-ebs-csi-driver/templates/poddisruptionbudget-controller.yaml
index 36fed9bc3..979a18a11 100644
--- a/charts/aws-ebs-csi-driver/templates/poddisruptionbudget-controller.yaml
+++ b/charts/aws-ebs-csi-driver/templates/poddisruptionbudget-controller.yaml
@@ -1,4 +1,4 @@
-{{- if not .Values.nodeComponentOnly -}}
+{{- if and .Values.controller.podDisruptionBudget.enabled (not .Values.nodeComponentOnly) -}}
 apiVersion: policy/v1
 kind: PodDisruptionBudget
 metadata:
@@ -16,4 +16,4 @@ spec:
   {{- else }}
   minAvailable: 2
   {{- end }}
-{{- end }}
+{{- end -}}
diff --git a/charts/aws-ebs-csi-driver/values.yaml b/charts/aws-ebs-csi-driver/values.yaml
index 6089d8351..f14c652ad 100644
--- a/charts/aws-ebs-csi-driver/values.yaml
+++ b/charts/aws-ebs-csi-driver/values.yaml
@@ -245,6 +245,9 @@ controller:
   deploymentAnnotations: {}
   podAnnotations: {}
   podLabels: {}
+  podDisruptionBudget:
+    # Warning: Disabling PodDisruptionBudget may lead to stateful workloads not starting due to lack of controller pods
+    enabled: true
   priorityClassName: system-cluster-critical
   # AWS region to use. If not specified then the region will be looked up via the AWS EC2 metadata
   # service.