From 94da9385ec58cf0a00c138be46044b3766a4ad9f Mon Sep 17 00:00:00 2001
From: edwinye <edwinysk@amazon.com>
Date: Fri, 22 Aug 2025 18:19:23 +0000
Subject: [PATCH 1/2] Add podLabels parameter

Original work by bueti in PR 416
---
 charts/aws-fsx-csi-driver/templates/controller-deployment.yaml | 3 +++
 charts/aws-fsx-csi-driver/templates/node-daemonset.yaml        | 3 +++
 charts/aws-fsx-csi-driver/values.yaml                          | 2 ++
 3 files changed, 8 insertions(+)

diff --git a/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml b/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml
index 64707a63..7387eed3 100644
--- a/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml
+++ b/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml
@@ -19,6 +19,9 @@ spec:
       labels:
         app: fsx-csi-controller
         {{- include "aws-fsx-csi-driver.labels" . | nindent 8 }}
+        {{- if .Values.controller.podLabels }}
+          {{- toYaml .Values.controller.podLabels | nindent 8 }}
+        {{- end }}
     spec:
       {{- if .Values.imagePullSecrets }}
       imagePullSecrets:
diff --git a/charts/aws-fsx-csi-driver/templates/node-daemonset.yaml b/charts/aws-fsx-csi-driver/templates/node-daemonset.yaml
index 6bb238b6..01ae67f2 100644
--- a/charts/aws-fsx-csi-driver/templates/node-daemonset.yaml
+++ b/charts/aws-fsx-csi-driver/templates/node-daemonset.yaml
@@ -22,6 +22,9 @@ spec:
       labels:
         app: fsx-csi-node
         {{- include "aws-fsx-csi-driver.labels" . | nindent 8 }}
+        {{- if .Values.node.podLabels }}
+          {{- toYaml .Values.node.podLabels | nindent 8 }}
+        {{- end }}
     spec:
       {{- if .Values.imagePullSecrets }}
       imagePullSecrets:
diff --git a/charts/aws-fsx-csi-driver/values.yaml b/charts/aws-fsx-csi-driver/values.yaml
index 0e949a04..9c3509be 100644
--- a/charts/aws-fsx-csi-driver/values.yaml
+++ b/charts/aws-fsx-csi-driver/values.yaml
@@ -126,6 +126,7 @@ controller:
                 values:
                   - fargate
                   - hybrid
+  podLabels: {}
   # topologySpreadConstraints:
   #  - maxSkew: 1
   #    topologyKey: topology.kubernetes.io/zone
@@ -200,6 +201,7 @@ node:
             values:
             - fargate
             - hybrid
+  podLabels: {}
 
 nameOverride: ""
 fullnameOverride: ""

From fd9e97c052552dc0bbc7f348dbac992de0229924 Mon Sep 17 00:00:00 2001
From: edwinye <edwinysk@amazon.com>
Date: Fri, 22 Aug 2025 18:27:19 +0000
Subject: [PATCH 2/2] Allow the timeout of the CSI driver to be set

Original work by coolstim PR 435
---
 charts/aws-fsx-csi-driver/templates/controller-deployment.yaml | 2 +-
 charts/aws-fsx-csi-driver/values.yaml                          | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml b/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml
index 7387eed3..b93214ce 100644
--- a/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml
+++ b/charts/aws-fsx-csi-driver/templates/controller-deployment.yaml
@@ -99,7 +99,7 @@ spec:
           args:
             - --csi-address=$(ADDRESS)
             - --v={{ .Values.sidecars.provisioner.logLevel }}
-            - --timeout=5m
+            - --timeout={{ .Values.controller.timeout| default "5m" }}
             - --extra-create-metadata
             - --leader-election=true
           env:
diff --git a/charts/aws-fsx-csi-driver/values.yaml b/charts/aws-fsx-csi-driver/values.yaml
index 9c3509be..6ccecc51 100644
--- a/charts/aws-fsx-csi-driver/values.yaml
+++ b/charts/aws-fsx-csi-driver/values.yaml
@@ -103,6 +103,8 @@ controller:
     - effect: NoExecute
       operator: Exists
       tolerationSeconds: 300
+  # Specify the timeout of all calls to the CSI driver.
+  timeout: 5m
   # securityContext on the controller pod
   securityContext:
     runAsNonRoot: false