From dd144d3c477dc038d61be9ead38fe670ac1b9565 Mon Sep 17 00:00:00 2001
From: Wes Carroll <wcarroll@oceanbeachlife.com>
Date: Sun, 14 Jul 2019 20:35:47 -0400
Subject: [PATCH] Update to use New-Map helper function

---
 ZertoApiWrapper/Public/Export-ZertoVmNicSetting.ps1 | 10 ++--------
 ZertoApiWrapper/Public/Import-ZertoVmNicSetting.ps1 | 10 ++--------
 2 files changed, 4 insertions(+), 16 deletions(-)

diff --git a/ZertoApiWrapper/Public/Export-ZertoVmNicSetting.ps1 b/ZertoApiWrapper/Public/Export-ZertoVmNicSetting.ps1
index 6846145..6de8852 100644
--- a/ZertoApiWrapper/Public/Export-ZertoVmNicSetting.ps1
+++ b/ZertoApiWrapper/Public/Export-ZertoVmNicSetting.ps1
@@ -38,18 +38,12 @@ function Export-ZertoVmNicSetting {
         }
         $nicSettings = foreach ($group in $vpgs) {
             $protectedVms = Get-ZertoProtectedVm -vpgName ($group.vpgname)
-            $vmMap = @{ }
-            foreach ($vm in $protectedVms) {
-                $vmMap["$($vm.vmIdentifier)"] = $vm.vmName
-            }
+            $vmMap = New-Map -InputObject $protectedVms -key "vmIdentifier" -value "vmName"
             $settingsId = New-ZertoVpgSettingsIdentifier -vpgIdentifier $group.vpgIdentifier
             $vmSettings = Get-ZertoVpgSetting -vpgSettingsIdentifier $settingsId -vms
             $networks = Get-ZertoVirtualizationSite -siteIdentifier $group.RecoverySite.identifier -networks
             $null = Remove-ZertoVpgSettingsIdentifier -vpgSettingsIdentifier $settingsId
-            $networkMap = @{ }
-            foreach ($network in $networks) {
-                $networkMap[$network.NetworkIdentifier] = $network.VirtualizationNetworkName
-            }
+            $networkMap = New-Map -InputObject $networks -key "NetworkIdentifier" -value "VirtualizationNetworkName"
             foreach ($vm in $vmSettings) {
                 $nicInfo = [PSCustomObject]@{
                     VPGName              = $group.VPGName
diff --git a/ZertoApiWrapper/Public/Import-ZertoVmNicSetting.ps1 b/ZertoApiWrapper/Public/Import-ZertoVmNicSetting.ps1
index 8a24dec..3cdf766 100644
--- a/ZertoApiWrapper/Public/Import-ZertoVmNicSetting.ps1
+++ b/ZertoApiWrapper/Public/Import-ZertoVmNicSetting.ps1
@@ -31,15 +31,9 @@ function Import-ZertoVmNicSetting {
             $VpgInfo = Get-ZertoVpg -vpgName $Vpg
             $VpgIdentifier = $VpgInfo.VpgIdentifier
             $RecoveryNetworks = Get-ZertoVirtualizationSite -siteIdentifier $VpgInfo.RecoverySite.Identifier -networks
-            $NetworkMap = @{ }
-            foreach ($Network in $RecoveryNetworks) {
-                $NetworkMap[$Network.VirtualizationNetworkName] = $Network.NetworkIdentifier
-            }
+            $NetworkMap = New-Map -InputObject $RecoveryNetworks -key "VirtualizationNetworkName" -value "NetworkIdentifier"
             $VpgVms = Get-ZertoProtectedVm -vpgName $Vpg
-            $VmMap = @{ }
-            foreach ($Vm in $VpgVms) {
-                $VmMap[$Vm.vmName] = $Vm.vmIdentifier
-            }
+            $VmMap = New-Map -InputObject $VpgVms -key "vmName" -value "vmIdentifier"
             $VpgSettingsId = New-ZertoVpgSettingsIdentifier -vpgIdentifier $VpgIdentifier
             $VmsToUpdate = $ImportData | Where-Object { $_.VPGName -eq $Vpg }
             foreach ($vm in $VmsToUpdate) {