Add PE Location Variable for Databricks Workspace

PerfectThymeTech · Nov 19, 2024 · c9c46bd · c9c46bd
1 parent 696e468
commit c9c46bd
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/modules/databricksworkspace/README.md b/modules/databricksworkspace/README.md
@@ -152,6 +152,14 @@ list(object({
 
 Default: `[]`
 
+### <a name="input_location_private_endpoint"></a> [location\_private\_endpoint](#input\_location\_private\_endpoint)
+
+Description: Specifies the location of the private endpoint. Use this variables only if the private endpoint(s) should reside in a different location than the service itself.
+
+Type: `string`
+
+Default: `null`
+
 ### <a name="input_private_dns_zone_id_databricks"></a> [private\_dns\_zone\_id\_databricks](#input\_private\_dns\_zone\_id\_databricks)
 
 Description: Specifies the resource ID of the private DNS zone for Azure Databricks. Not required if DNS A-records get created via Azure Policy.

diff --git a/modules/databricksworkspace/connectivity.tf b/modules/databricksworkspace/connectivity.tf
@@ -1,6 +1,6 @@
 resource "azurerm_private_endpoint" "private_endpoint_databricks_workspace_databricks_ui_api" {
   name                = "${azurerm_databricks_workspace.databricks_workspace.name}-uiapi-pe"
-  location            = var.location
+  location            = var.location_private_endpoint != null ? var.location_private_endpoint : var.location
   resource_group_name = azurerm_databricks_workspace.databricks_workspace.resource_group_name
   tags                = var.tags
 
@@ -33,7 +33,7 @@ resource "azurerm_private_endpoint" "private_endpoint_databricks_workspace_brows
   count = var.databricks_workspace_browser_authentication_private_endpoint_enabled ? 1 : 0
 
   name                = "${azurerm_databricks_workspace.databricks_workspace.name}-auth-pe"
-  location            = var.location
+  location            = var.location_private_endpoint != null ? var.location_private_endpoint : var.location
   resource_group_name = azurerm_databricks_workspace.databricks_workspace.resource_group_name
   tags                = var.tags
 

diff --git a/modules/databricksworkspace/tests/test.tftest.hcl b/modules/databricksworkspace/tests/test.tftest.hcl
@@ -68,6 +68,7 @@ run "setup" {
 
   variables {
     location                                 = var.location
+    location_private_endpoint                = var.location
     environment                              = "int"
     prefix                                   = "tfmdladb"
     virtual_network_id                       = var.virtual_network_id

diff --git a/modules/databricksworkspace/variables.tf b/modules/databricksworkspace/variables.tf
@@ -6,6 +6,14 @@ variable "location" {
   nullable    = false
 }
 
+variable "location_private_endpoint" {
+  description = "Specifies the location of the private endpoint. Use this variables only if the private endpoint(s) should reside in a different location than the service itself."
+  type        = string
+  sensitive   = false
+  nullable    = true
+  default     = null
+}
+
 variable "resource_group_name" {
   description = "Specifies the resource group name in which all resources will get deployed."
   type        = string