From 38cfcde00483a79309475b264136e576dac82894 Mon Sep 17 00:00:00 2001
From: drfaust92 <ilia.lazebnik@gmail.com>
Date: Sun, 16 May 2021 15:37:19 +0300
Subject: [PATCH] node selectors are configureable

---
 README.md    |  2 ++
 daemonset.tf |  6 +++---
 variables.tf | 12 ++++++++++++
 3 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index c9d4c71..692806a 100644
--- a/README.md
+++ b/README.md
@@ -47,8 +47,10 @@ No modules.
 |------|-------------|------|---------|:--------:|
 | <a name="input_annotations"></a> [annotations](#input\_annotations) | Optional annotations to add to EFS CSI driver resources | `map(string)` | `{}` | no |
 | <a name="input_csi_controller_tolerations"></a> [csi\_controller\_tolerations](#input\_csi\_controller\_tolerations) | CSI driver controller tolerations | `list(map(string))` | `[]` | no |
+| <a name="input_extra_node_selectors"></a> [extra\_node\_selectors](#input\_extra\_node\_selectors) | A map of extra node selectors for all components | `map(string)` | `{}` | no |
 | <a name="input_log_level"></a> [log\_level](#input\_log\_level) | The log level for the CSI Driver controller | `number` | `5` | no |
 | <a name="input_namespace"></a> [namespace](#input\_namespace) | Namespace for EFS CSI driver resources | `string` | `"kube-system"` | no |
+| <a name="input_node_extra_node_selectors"></a> [node\_extra\_node\_selectors](#input\_node\_extra\_node\_selectors) | A map of extra node selectors for node pods | `map(string)` | `{}` | no |
 
 ## Outputs
 
diff --git a/daemonset.tf b/daemonset.tf
index 8bef409..698f603 100644
--- a/daemonset.tf
+++ b/daemonset.tf
@@ -24,9 +24,9 @@ resource "kubernetes_daemonset" "efs" {
       }
 
       spec {
-        node_selector = {
-          "beta.kubernetes.io/os" = "linux"
-        }
+        node_selector = merge({
+          "beta.kubernetes.io/os" : "linux",
+        }, var.extra_node_selectors, var.node_extra_node_selectors)
 
         toleration {
           operator = "Exists"
diff --git a/variables.tf b/variables.tf
index 07605f9..df3d6d0 100644
--- a/variables.tf
+++ b/variables.tf
@@ -20,4 +20,16 @@ variable "log_level" {
   description = "The log level for the CSI Driver controller"
   default     = 5
   type        = number
+}
+
+variable "extra_node_selectors" {
+  description = "A map of extra node selectors for all components"
+  default     = {}
+  type        = map(string)
+}
+
+variable "node_extra_node_selectors" {
+  description = "A map of extra node selectors for node pods"
+  default     = {}
+  type        = map(string)
 }
\ No newline at end of file