From 7823dfbfd2f21e366c9bf3c6261fe8a8bc4090e5 Mon Sep 17 00:00:00 2001
From: Peng Liu <pengliu12@gmail.com>
Date: Wed, 9 Dec 2020 17:04:20 -0500
Subject: [PATCH] Add ability to customize node daemonset nodeselector

---
 charts/aws-ebs-csi-driver/Chart.yaml          | 2 +-
 charts/aws-ebs-csi-driver/templates/node.yaml | 3 +++
 charts/aws-ebs-csi-driver/values.yaml         | 1 +
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/charts/aws-ebs-csi-driver/Chart.yaml b/charts/aws-ebs-csi-driver/Chart.yaml
index 539c8c5b29..ab24d3ffc1 100644
--- a/charts/aws-ebs-csi-driver/Chart.yaml
+++ b/charts/aws-ebs-csi-driver/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v1
 appVersion: "0.8.0"
 name: aws-ebs-csi-driver
 description: A Helm chart for AWS EBS CSI Driver
-version: 0.7.0
+version: 0.7.1
 kubeVersion: ">=1.14.0-0"
 home: https://github.com/kubernetes-sigs/aws-ebs-csi-driver
 sources:
diff --git a/charts/aws-ebs-csi-driver/templates/node.yaml b/charts/aws-ebs-csi-driver/templates/node.yaml
index b808342ea9..b5407ca659 100644
--- a/charts/aws-ebs-csi-driver/templates/node.yaml
+++ b/charts/aws-ebs-csi-driver/templates/node.yaml
@@ -31,6 +31,9 @@ spec:
                 - fargate
       nodeSelector:
         kubernetes.io/os: linux
+        {{- with .Values.node.nodeSelector }}
+{{ toYaml . | indent 8 }}
+        {{- end }}
       hostNetwork: true
       priorityClassName: system-node-critical
       tolerations:
diff --git a/charts/aws-ebs-csi-driver/values.yaml b/charts/aws-ebs-csi-driver/values.yaml
index d1d14e7eac..07e6c878c1 100644
--- a/charts/aws-ebs-csi-driver/values.yaml
+++ b/charts/aws-ebs-csi-driver/values.yaml
@@ -84,6 +84,7 @@ k8sTagClusterId: ""
 region: ""
 
 node:
+  nodeSelector: {}
   podAnnotations: {}
   tolerateAllTaints: true
   tolerations: []