feat: make num of zones configurable (#26)

ouzi-dev · Mar 9, 2020 · dfbe4e9 · dfbe4e9
1 parent 9d05188
commit dfbe4e9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 11 deletions.
diff --git a/main.tf b/main.tf
@@ -34,11 +34,8 @@ module "gke-cluster" {
   project = var.gcloud_project
 
   cluster_name = var.gke_name
-  zones = [
-    data.google_compute_zones.available.names[0],
-    data.google_compute_zones.available.names[1],
-    data.google_compute_zones.available.names[2]
-  ]
+  zones = slice(data.google_compute_zones.available.names, 0, var.gke_num_of_zones)
+
   node_cidr_range    = var.gke_node_cidr_range
   pod_cidr_range     = var.gke_pod_cidr_range
   service_cidr_range = var.gke_service_cidr_range
@@ -79,11 +76,8 @@ module "image-build-workers" {
 
   gke_cluster_name = var.gke_name
   group_name       = local.imagebuilder_worker_group_name
-  zones = [
-    data.google_compute_zones.available.names[0],
-    data.google_compute_zones.available.names[1],
-    data.google_compute_zones.available.names[2]
-  ]
+  zones = slice(data.google_compute_zones.available.names, 0, var.gke_num_of_zones)
+
   gke_node_scopes        = var.gke_node_scopes
   machine_type           = var.imagebuilder_machine_type
   machine_disk_size      = var.imagebuilder_machine_disk_size

diff --git a/variables.tf b/variables.tf
@@ -87,6 +87,11 @@ variable "gke_max_nodes" {
   default     = 4
 }
 
+variable "gke_num_of_zones" {
+  description = "Number of zones for the cluster."
+  default = 3
+}
+
 variable "max_surge" {
   default = 1
 }
@@ -115,7 +120,6 @@ variable "imagebuilder_machine_is_preemptible" {
   default     = true
 }
 
-
 variable "imagebuilder_min_nodes" {
   description = "Min number of workers"
   default     = 0