redo for_each

databricks-default-cluster-policies/groups.tf

@@ -5,14 +5,17 @@ locals {
 
   # Define the policies with existing groups mapped to each policy
   policies = var.policy_map
-  
+
   # Generate full workspace policy names by prefixing policy names
   all_ws_policy_names = flatten([
     for prefix in local.ws_policy_name_prefixes : [
       for policy_map in local.policies : 
         "${prefix}${keys(policy_map)[0]}"
     ]
   ])
+
+  # Create a flat map of policy names to associated groups
+  policy_group_map = merge([for policy_map in local.policies : policy_map]...)
 }
 
 # Create Databricks groups for each policy name
@@ -25,9 +28,9 @@ resource "databricks_group" "ws_policy_groups" {
 
 # Retrieve the existing Databricks groups that need to be assigned
 data "databricks_group" "groups" {
-  for_each = flatten([
-    for policy_map in local.policies : values(policy_map)[0]
-  ])
+  for_each = toset(flatten([
+    for group in local.policy_group_map : group
+  ]))
 
   display_name = each.value
 }
@@ -39,5 +42,5 @@ resource "databricks_group_member" "ws_policy_group_members" {
   group_id = each.value.id
 
   # Assign all existing groups that correspond to this policy group
-  member_id = data.databricks_group.groups[local.policies[*][each.key]].id
+  member_id = data.databricks_group.groups[local.policy_group_map[replace(each.key, var.policy_name_prefix, "")][0]].id
 }