add flag

hashicorp · Nov 15, 2024 · ea74576 · ea74576
1 parent 6df5ec4
commit ea74576
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/services/databricks/registration.go b/internal/services/databricks/registration.go
@@ -4,6 +4,7 @@
 package databricks
 
 import (
+	`github.com/hashicorp/terraform-provider-azurerm/internal/features`
 	"github.com/hashicorp/terraform-provider-azurerm/internal/sdk"
 	"github.com/hashicorp/terraform-provider-azurerm/internal/tf/pluginsdk"
 )
@@ -41,12 +42,17 @@ func (r Registration) SupportedDataSources() map[string]*pluginsdk.Resource {
 
 // SupportedResources returns the supported Resources supported by this Service
 func (r Registration) SupportedResources() map[string]*pluginsdk.Resource {
-	return map[string]*pluginsdk.Resource{
+	resources := map[string]*pluginsdk.Resource{
 		"azurerm_databricks_workspace":                                resourceDatabricksWorkspace(),
 		"azurerm_databricks_workspace_root_dbfs_customer_managed_key": resourceDatabricksWorkspaceRootDbfsCustomerManagedKey(),
 		"azurerm_databricks_virtual_network_peering":                  resourceDatabricksVirtualNetworkPeering(),
-		"azurerm_databricks_workspace_customer_managed_key":           resourceDatabricksWorkspaceCustomerManagedKey(), // TODO: Remove in 5.0
 	}
+
+	if features.FivePointOh() {
+		resources["azurerm_databricks_workspace_customer_managed_key"] = resourceDatabricksWorkspaceCustomerManagedKey()
+	}
+
+	return resources
 }
 
 // DataSources returns the typed DataSources supported by this service