improve logic for the cluster autoscaler extension

module-extensions.tf

@@ -43,6 +43,7 @@ module "extensions" {
   cluster_autoscaler_helm_version      = var.cluster_autoscaler_helm_version
   cluster_autoscaler_helm_values       = var.cluster_autoscaler_helm_values
   cluster_autoscaler_helm_values_files = var.cluster_autoscaler_helm_values_files
+  expected_autoscale_worker_pools      = coalesce(one(module.workers[*].worker_pool_autoscale_expected), 0)
 
   # Gatekeeper
   gatekeeper_install           = var.gatekeeper_install

modules/extensions/autoscaler.tf

@@ -12,7 +12,7 @@ locals {
   cluster_autoscaler_enabled = alltrue([
     var.cluster_autoscaler_install,
     var.expected_node_count > 0,
-    length(local.worker_pools_autoscaling) > 0,
+    var.expected_autoscale_worker_pools > 0,
   ])
 
   # Templated Helm manifest values

modules/extensions/variables.tf

@@ -63,6 +63,7 @@ variable "cluster_autoscaler_namespace" { type = string }
 variable "cluster_autoscaler_helm_version" { type = string }
 variable "cluster_autoscaler_helm_values" { type = map(string) }
 variable "cluster_autoscaler_helm_values_files" { type = list(string) }
+variable "expected_autoscale_worker_pools" { type = number }
 
 # Prometheus
 variable "prometheus_install" { type = bool }

modules/workers/locals.tf

@@ -168,6 +168,11 @@ locals {
     for k, v in local.enabled_worker_pools : tobool(v.drain) ? lookup(v, "size", var.worker_pool_size) : 0
   ])
 
+   # Number of work pools in the worker pools with autoscale enabled
+  expected_autoscale_worker_pools = length(local.enabled_worker_pools) == 0 ? 0 : sum([
+    for k, v in local.enabled_worker_pools : tobool(v.autoscale) ? 1 : 0
+  ])
+
   # Enabled worker_pool map entries for node pools
   enabled_node_pools = {
     for k, v in local.enabled_worker_pools : k => v

modules/workers/outputs.tf

@@ -30,3 +30,8 @@ output "worker_drain_expected" {
   description = "# of nodes expected to be draining in worker pools"
   value       = local.expected_drain_count
 }
+
+output "worker_pool_autoscale_expected" {
+  description = "# of worker pools expected with autoscale enabled from created worker pools"
+  value       = local.expected_autoscale_worker_pools
+}