fix: workspace assignment

Author: owlleg6

main.tf

@@ -1,5 +1,5 @@
 locals {
-  # This optional suffix is added to the end of resource names. 
+  # This optional suffix is added to the end of resource names.
   suffix = length(var.suffix) == 0 ? "" : "-${var.suffix}"
 
   databricks_metastore_name = var.custom_databricks_metastore_name == null ? "meta-${var.project}-${var.env}-${var.location}${local.suffix}" : var.custom_databricks_metastore_name
@@ -56,6 +56,14 @@ resource "databricks_metastore_data_access" "this" {
   is_default = true
 }
 
+resource "databricks_metastore_assignment" "this" {
+  count = anytrue([!var.create_metastore, length(var.external_metastore_id) == 0]) ? 0 : 1
+
+  workspace_id         = var.workspace_id
+  metastore_id         = length(var.external_metastore_id) == 0 ? databricks_metastore.this[0].id : var.external_metastore_id
+  default_catalog_name = "hive_metastore"
+}
+
 # Catalog
 resource "databricks_catalog" "this" {
   for_each = alltrue([!var.create_metastore, length(var.external_metastore_id) == 0]) ? {} : var.catalog

variables.tf

@@ -19,6 +19,11 @@ variable "suffix" {
   default     = ""
 }
 
+variable "workspace_id" {
+  type        = string
+  description = "Id of Azure Databricks workspace"
+}
+
 # Unity Catalog variables
 variable "create_metastore" {
   type        = bool