From e438a5cbcb0f7edd34fd0cb99d92075eed7d8ec3 Mon Sep 17 00:00:00 2001
From: BryanFauble <17128019+BryanFauble@users.noreply.github.com>
Date: Thu, 31 Oct 2024 09:54:37 -0700
Subject: [PATCH] Ensure same zone for deployed nodes

---
 modules/apache-airflow/templates/values.yaml | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/modules/apache-airflow/templates/values.yaml b/modules/apache-airflow/templates/values.yaml
index 6a8f59e2..884f32be 100644
--- a/modules/apache-airflow/templates/values.yaml
+++ b/modules/apache-airflow/templates/values.yaml
@@ -715,7 +715,8 @@ workers:
 
   # Select certain nodes for airflow worker pods.
   nodeSelector: {
-    spotinst.io/node-lifecycle: "od"
+    spotinst.io/node-lifecycle: "od",
+    topology.kubernetes.io/zone: "us-east-1a"
   }
   runtimeClassName: ~
   priorityClassName: ~
@@ -925,7 +926,8 @@ scheduler:
 
   # Select certain nodes for airflow scheduler pods.
   nodeSelector: {
-    spotinst.io/node-lifecycle: "od"
+    spotinst.io/node-lifecycle: "od",
+    topology.kubernetes.io/zone: "us-east-1a"
   }
   affinity:
   # default scheduler affinity is:
@@ -1614,7 +1616,8 @@ triggerer:
 
   # Select certain nodes for airflow triggerer pods.
   nodeSelector: {
-    spotinst.io/node-lifecycle: "od"
+    spotinst.io/node-lifecycle: "od",
+    topology.kubernetes.io/zone: "us-east-1a"
   }
   affinity:
   # default triggerer affinity is:
@@ -2324,7 +2327,8 @@ redis:
 
   # Select certain nodes for redis pods.
   nodeSelector: {
-    spotinst.io/node-lifecycle: "od"
+    spotinst.io/node-lifecycle: "od",
+    topology.kubernetes.io/zone: "us-east-1a"
   }
   affinity: {}
   tolerations: []