Adding function to change multipath policy

Microsoft.AVS.VMFS/Microsoft.AVS.VMFS.psd1

@@ -85,7 +85,8 @@
         "Get-StorageAdapters",
         "Get-VmKernelAdapters",
         "Set-NVMeTCP",
-        "New-NVMeTCPAdapter"
+        "New-NVMeTCPAdapter",
+        "Set-PureStorageDatastoreMultipathingPolicy"
     )
 
     # Cmdlets to export from this module, for best performance, do not use wildcards and do not delete the entry, use an empty array if there are no cmdlets to export.

Microsoft.AVS.VMFS/Microsoft.AVS.VMFS.psm1

@@ -110,8 +110,11 @@ function Set-VmfsIscsi {
     .PARAMETER Size
      Datastore capacity size in bytes
 
+     .PARAMETER MultipathPolicy
+     Set Multipath Policy (optional). If not provided default VMFS Policy "VMW_PSP_MRU" is used.
+
     .EXAMPLE
-     New-VmfsDatastore -ClusterName "myCluster" -DatastoreName "myDatastore" -DeviceNaaId $DeviceNaaId -Size <size-in-bytes>
+     New-VmfsDatastore -ClusterName "myCluster" -DatastoreName "myDatastore" -DeviceNaaId $DeviceNaaId -Size <size-in-bytes> -MultipathPolicy "myMultipathPolicy"
 
     .INPUTS
      vCenter cluster name, datastore name, device NAA ID and datastore size.
@@ -150,6 +153,13 @@ function New-VmfsDatastore {
         [ValidateNotNull()]
         [String]
         $Size
+
+        [Parameter(
+            Mandatory=$false,
+            HelpMessage = 'Set Multipath policy')]
+        [ValidateNotNull()]
+        [String]
+        $MultipathPolicy,
     )
 
     try {
@@ -199,6 +209,10 @@ function New-VmfsDatastore {
         $VmfsDatastoreCreateSpec.vmfs.MajorVersion = $DatastoreCreateOptions[0].Spec.Vmfs.MajorVersion
 
         $DatastoreSystem.CreateVmfsDatastore($VmfsDatastoreCreateSpec)
+
+        if (-not [string]::IsNullOrEmpty($MultipathPolicy)) {
+            Set-VMFSDatastoreMultipathPolicy -DatastoreName $DatastoreName -MultipathPolicy $MultipathPolicy
+        }
     } catch {
         Write-Error $Global:Error[0]
     }
@@ -1544,3 +1558,55 @@ function New-NVMeTCPAdapter {
     Write-Host ""
 
 }  
+
+
+<#
+    .DESCRIPTION
+     Set default Multipath Policy to Round Robin for Pure Storage Datastore.
+
+    .PARAMETER DatastoreName
+     Datastore name
+
+    .PARAMETER MultipathPolicy
+     Multipath Policy
+
+    .EXAMPLE
+     Set-VMFSDatastoreMultipathPolicy -DatastoreName "myDatastore" -MultipathPolicy "RoundRobin"
+
+    .INPUTS
+     vCenter datastore name.
+
+    .OUTPUTS
+     None.
+#>
+
+function Set-VMFSDatastoreMultipathPolicy {
+    [CmdletBinding()]
+    [AVSAttribute(10, UpdatesSDDC = $false, AutomationOnly = $true)]
+
+    param
+    (
+        [Parameter(Mandatory = $true,
+            HelpMessage = 'Datastore Name')]
+        [string]$DatastoreName,
+
+        [Parameter(Mandatory = $true,
+            HelpMessage = 'Multipath Policy')]
+        [string]$MultipathPolicy
+    )
+
+    try {
+        $Datastore = Get-Datastore -Name $DatastoreName
+        $ValidMultipathPolicies = Get-ScsiLun -VmHost $Datastore.VMHost | Select-Object -ExpandProperty MultipathPolicy | Sort-Object -Unique
+        if ($ValidMultipathPolicies -notcontains $MultipathPolicy) {
+            throw Write-Error "Invalid multipath policy '$MultipathPolicy' specified. Valid policies are: $($ValidMultipathPolicies -join ', ')."
+        }
+
+        Set-Datastore -Datastore $Datastore -DefaultMultipathPolicy $MultipathPolicy
+        Write-Host "Successfully set Multipath Policy to $MultipathPolicy for datastore $DatastoreName."
+    }
+    catch {
+        throw "Failed to set Multipath Policy to $MultipathPolicy for datastore $DatastoreName. Error: $($_.Exception)"
+    }
+
+}