From 52563b7e8d9ef4ede4c00fada5e68a00daa2122a Mon Sep 17 00:00:00 2001
From: zjhe <lonegunmanb@hotmail.com>
Date: Fri, 4 Oct 2024 16:29:40 +0800
Subject: [PATCH] fix example

---
 .../301-machine-learning-hub-spoke-secure/dsvm.tf     | 11 ++++++++++-
 .../variables.tf                                      |  1 +
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/quickstart/301-machine-learning-hub-spoke-secure/dsvm.tf b/quickstart/301-machine-learning-hub-spoke-secure/dsvm.tf
index 204e9b6e..a821de08 100644
--- a/quickstart/301-machine-learning-hub-spoke-secure/dsvm.tf
+++ b/quickstart/301-machine-learning-hub-spoke-secure/dsvm.tf
@@ -10,6 +10,15 @@ resource "azurerm_network_interface" "dsvm" {
   }
 }
 
+resource "random_password" "dsvm_host_password" {
+  count  = var.dsvm_host_password == null ? 1 : 0
+  length = 20
+}
+
+locals {
+  dsvm_host_password = try(random_password.dsvm_host_password[0].result, var.dsvm_host_password)
+}
+
 resource "azurerm_windows_virtual_machine" "dsvm" {
   name                = var.dsvm_name
   location            = azurerm_resource_group.default.location
@@ -37,7 +46,7 @@ resource "azurerm_windows_virtual_machine" "dsvm" {
   }
   computer_name  = var.dsvm_name
   admin_username = var.dsvm_admin_username
-  admin_password = var.dsvm_host_password
+  admin_password = local.dsvm_host_password
 
   provision_vm_agent = true
 
diff --git a/quickstart/301-machine-learning-hub-spoke-secure/variables.tf b/quickstart/301-machine-learning-hub-spoke-secure/variables.tf
index f2a29cd8..2d3ba0aa 100644
--- a/quickstart/301-machine-learning-hub-spoke-secure/variables.tf
+++ b/quickstart/301-machine-learning-hub-spoke-secure/variables.tf
@@ -91,4 +91,5 @@ variable "dsvm_host_password" {
   type        = string
   description = "Password for the admin username of the Data Science VM"
   sensitive   = true
+  default     = null
 }
\ No newline at end of file