add param to Set-AzVmssStorageProfile (Azure#19862)

* set-azvmssStorageProfile

* set-azvm

* Update Set-AzVM.md

* Update ChangeLog.md

Co-authored-by: Theodore Chang <[email protected]>
Co-authored-by: Yabo Hu <[email protected]>

Author: 3 people

src/Compute/Compute/ChangeLog.md

@@ -25,6 +25,8 @@
 * Added Disk Delete Optional parameters `OsDisk Deletion Option` and `Delete Option` to the `Set-AzVmssStorageProfile` (OS Disk) and `Add-AzVmssDataDisk` (Data Disk)
 * Improved printed output for `Get-AzComputeResourceSku`
 * Updated `Get-AzHost` cmdlet logic to return Host for `-ResourceId` parameterset.
+* Added `-OSDiskSizeGB` optional parameter for `Set-AzVmssStorageProfile`.
+* Improved cmdlet description for `Set-AzVM` and added examples.
 * Updated property mapping for parameter `Encryption` of `New-AzGalleryImageVersion`
 * Updated list format to display all VmssVmRunCommand properties for `Get-AzVmssVmRunCommand`
 

src/Compute/Compute/Generated/VirtualMachineScaleSet/Config/SetAzureRmVmssStorageProfileCommand.cs

@@ -150,6 +150,12 @@ public partial class SetAzureRmVmssStorageProfileCommand : Microsoft.Azure.Comma
             ValueFromPipelineByPropertyName = true)]
         public VirtualMachineScaleSetDataDisk[] DataDisk { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "Specifies the size of an empty data disk in gigabytes. This element can be used to overwrite the size of the disk in a virtual machine image.")]
+        public int OSDiskSizeGB { get; set; }
+
         [Parameter(
             Mandatory = false,
             HelpMessage = "Specifies the disk controller type configured for the VM and VirtualMachineScaleSet. This property is only supported for virtual machines whose operating system disk and VM sku supports Generation 2 (https://docs.microsoft.com/en-us/azure/virtual-machines/generation-2), please check the HyperVGenerations capability returned as part of VM sku capabilities in the response of Microsoft.Compute SKUs api for the region contains V2 (https://docs.microsoft.com/rest/api/compute/resourceskus/list) . <br> For more information about Disk Controller Types supported please refer to https://aka.ms/azure-diskcontrollertypes.")]
@@ -552,6 +558,26 @@ private void Run()
                 this.VirtualMachineScaleSet.VirtualMachineProfile.StorageProfile.DiskControllerType = this.DiskControllerType;
             }
 
+            if (this.IsParameterBound(c => c.OSDiskSizeGB))
+            {
+                // VirtualMachineProfile
+                if (this.VirtualMachineScaleSet.VirtualMachineProfile == null)
+                {
+                    this.VirtualMachineScaleSet.VirtualMachineProfile = new PSVirtualMachineScaleSetVMProfile();
+                }
+                // StorageProfile
+                if (this.VirtualMachineScaleSet.VirtualMachineProfile.StorageProfile == null)
+                {
+                    this.VirtualMachineScaleSet.VirtualMachineProfile.StorageProfile = new VirtualMachineScaleSetStorageProfile();
+                }
+                // OsDisk
+                if (this.VirtualMachineScaleSet.VirtualMachineProfile.StorageProfile.OsDisk == null)
+                {
+                    this.VirtualMachineScaleSet.VirtualMachineProfile.StorageProfile.OsDisk = new VirtualMachineScaleSetOSDisk();
+                }
+                this.VirtualMachineScaleSet.VirtualMachineProfile.StorageProfile.OsDisk.DiskSizeGB = this.OSDiskSizeGB;
+            }
+
             WriteObject(this.VirtualMachineScaleSet);
         }
     }

src/Compute/Compute/help/Set-AzVM.md

@@ -9,7 +9,7 @@ schema: 2.0.0
 # Set-AzVM
 
 ## SYNOPSIS
-Marks a virtual machine as generalized.
+This cmdlet can be used to do the followings: reapply or redeploy a virtual machine, mark a virtual machine as generalized, simulate eviction to a spot virtual machine.
 
 ## SYNTAX
 
@@ -61,8 +61,8 @@ Set-AzVM [-Id] <String> [-SimulateEviction] [-AsJob] [-DefaultProfile <IAzureCon
 ```
 
 ## DESCRIPTION
-The **Set-AzVM** cmdlet marks a virtual machine as generalized.
-Before you run this cmdlet, log on to the virtual machine and use Sysprep to prepare the hard disk.
+The **Set-AzVM** cmdlet can be used to redeploy or reapply a virtual machine. It can also be used to mark a virtual machine as generalized or simulate eviction to a spot virtual machine.<br>
+Before you mark a virtual machine as generalized, log on to the virtual machine and use Sysprep to prepare the hard disk. See [here](https://learn.microsoft.com/en-us/azure/virtual-machines/generalize#windows) for more information.
 
 ## EXAMPLES
 
@@ -71,7 +71,30 @@ Before you run this cmdlet, log on to the virtual machine and use Sysprep to pre
 Set-AzVM -ResourceGroupName "ResourceGroup11" -Name "VirtualMachine07" -Generalized
 ```
 
-This command marks the virtual machine named VirtualMachine07 as generalized.
+This command marks the virtual machine named "VirtualMachine07" as generalized.
+
+### Example 2: Simulate eviction to a spot virtual machine using resource ID
+```powershell
+$vm = Get-AzVM -ResourceGroupName "ResourceGroup11" -Name "VirtualMachine07"
+Set-AzVM -Id $vm.id -SimulateEviction
+```
+
+This command simulates eviction to the virtual machine named "VirtualMachine07" using the resource ID.
+
+### Example 3: Reapply a virtual machine
+```powershell
+Set-AzVM -ResourceGroupName "ResourceGroup11" -Name "VirtualMachine07" -Reapply
+```
+
+This command reapplies the virtual machine named "VirtualMachine07".
+
+### Example 4: Redeploy a virtual machine using resource ID
+```powershell
+$vm = Get-AzVM -ResourceGroupName "ResourceGroup11" -Name "VirtualMachine07"
+Set-AzVM -Id $vm.id -Redeploy
+```
+
+This command deploys the virtual machine named "VirtualMachine07" using the resource ID.
 
 ## PARAMETERS
 

src/Compute/Compute/help/Set-AzVmssStorageProfile.md

@@ -20,8 +20,9 @@ Set-AzVmssStorageProfile [-VirtualMachineScaleSet] <PSVirtualMachineScaleSet>
  [[-OsDiskCreateOption] <String>] [-OsDiskDeleteOption <String>] [[-OsDiskOsType] <OperatingSystemTypes>]
  [[-Image] <String>] [[-VhdContainer] <String[]>] [-ImageReferenceId <String>] [-OsDiskWriteAccelerator]
  [-DiffDiskSetting <String>] [-DiffDiskPlacement <String>] [-ManagedDisk <String>]
- [-DiskEncryptionSetId <String>] [-DataDisk <VirtualMachineScaleSetDataDisk[]>] [-DiskControllerType <String>]
- [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
+ [-DiskEncryptionSetId <String>] [-DataDisk <VirtualMachineScaleSetDataDisk[]>] [-OSDiskSizeGB <Int32>]
+ [-DiskControllerType <String>] [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm]
+ [<CommonParameters>]
 ```
 
 ## DESCRIPTION
@@ -332,6 +333,21 @@ Accept pipeline input: True (ByPropertyName)
 Accept wildcard characters: False
 ```
 
+### -OSDiskSizeGB
+Specifies the size of an empty data disk in gigabytes. This element can be used to overwrite the size of the disk in a virtual machine image.
+
+```yaml
+Type: System.Int32
+Parameter Sets: (All)
+Aliases:
+
+Required: False
+Position: Named
+Default value: None
+Accept pipeline input: True (ByPropertyName)
+Accept wildcard characters: False
+```
+
 ### -OsDiskWriteAccelerator
 Specifies whether WriteAccelerator should be enabled or disabled on the OS disk.