Private/Set-LoadBalancerConfiguration.ps1
function Global:Set-LoadBalancerConfiguration { [CmdletBinding()] <# .SYNOPSIS ... .DESCRIPTION ... #> param( [Parameter(Mandatory = $true)] [string] $ResourceGroupName, [Parameter(Mandatory = $true)] [string] $ResourceLocation, [Parameter(Mandatory = $true)] [string] $LoadBalancerName, [Parameter(Mandatory = $true)] [string] $StorageAccountName, [Parameter(Mandatory = $true)] [string] $TableNameEnvironments, [Parameter(Mandatory = $true)] [string] $EnvironmentTypeFilter, [HashTable] $Tags ) process { Write-CustomHost -Message "Configuring load balancer $LoadBalancerName..." $loadBalancer = Get-AzLoadBalancer -ResourceGroupName $ResourceGroupName -Name $LoadBalancerName -ErrorAction Stop $frontEndConfig = Get-AzLoadBalancerFrontendIpConfig -LoadBalancer $loadBalancer $backendPool = Get-AzLoadBalancerBackendAddressPoolConfig -LoadBalancer $loadBalancer <# # Test $VMScaleSet = Get-AzVmss -ResourceGroupName $ResourceGroupName -VMScaleSetName "ApplicationServerScaleSet" #$VMScaleSet Remove-AzVmssNetworkInterfaceConfiguration -VirtualMachineScaleSet $VMScaleSet -Name "Test" Remove-AzLoadBalancerBackendAddressPoolConfig -LoadBalancer $loadBalancer -Name $backendPool.Name foreach ($rule in Get-AzLoadBalancerRuleConfig -Name $loadBalancingRuleName -LoadBalancer $loadBalancer) { Write-CustomHost -Message "Removing rule $($rule.Name)..." Remove-AzLoadBalancerRuleConfig -LoadBalancer $loadBalancer -Name $rule.Name | Out-Null } foreach ($probe in Get-AzLoadBalancerProbeConfig -Name $loadBalancingRuleName -LoadBalancer $loadBalancer) { Write-CustomHost -Message "Removing probe $($probe.Name)..." Remove-AzLoadBalancerProbeConfig -LoadBalancer $loadBalancer -Name $probe.Name | Out-Null } Set-AzLoadBalancer -LoadBalancer $loadBalancer | Out-Null return #> $storageAccount = Get-AzStorageAccount -ResourceGroupName $ResourceGroupName -Name $StorageAccountName $storageAccountContext = $storageAccount.Context $environments = Get-EnvironmentsFromStorage -StorageAccountContext $storageAccountContext -TableNameEnvironments $TableNameEnvironments -TableNameDefaults "" -TypeFilter $EnvironmentTypeFilter -EnvironmentsOnly $counter = 0 Write-CustomHost -Message "Checking probes for $LoadBalancerName..." foreach ($environment in $environments) { $counter += 1 foreach ($property in $environment.PSObject.Properties | Where-Object { $_.Name -like '*Port*' }) { $probeName = "$($property.Name)HP$counter" if (-not(Get-AzLoadBalancerProbeConfig -Name $probeName -LoadBalancer $loadBalancer -ErrorAction SilentlyContinue)) { Write-CustomHost -Message "Adding probe $probeName..." $loadBalancer | Add-AzLoadBalancerProbeConfig -Name $probeName -Protocol Tcp -Port $property.Value -IntervalInSeconds 15 -ProbeCount 2 | Out-Null $loadBalancer | Set-AzLoadBalancerProbeConfig -Name $probeName -Protocol Tcp -Port $property.Value -IntervalInSeconds 15 -ProbeCount 2 | Out-Null } } } Write-CustomHost -Message "Saving changes to $LoadBalancerName..." $loadBalancer = Set-AzLoadBalancer -LoadBalancer $loadBalancer # Needs to be separated, otherwise Get-AzLoadBalancerProbeConfig will return an empty probe objects $counter = 0 Write-CustomHost -Message "Checking rules for $LoadBalancerName..." foreach ($environment in $environments) { $counter += 1 foreach ($property in $environment.PSObject.Properties | Where-Object { $_.Name -like '*Port*' }) { $probeName = "$($property.Name)HP$counter" $loadBalancingRuleName = "$($property.Name)Rule$counter" if (-not(Get-AzLoadBalancerRuleConfig -Name $loadBalancingRuleName -LoadBalancer $loadBalancer -ErrorAction SilentlyContinue)) { $probe = Get-AzLoadBalancerProbeConfig -Name $probeName -LoadBalancer $loadBalancer Write-CustomHost -Message "Adding rule $loadBalancingRuleName..." $loadBalancer | Add-AzLoadBalancerRuleConfig -Name $loadBalancingRuleName -Protocol Tcp -FrontendIpConfiguration $frontEndConfig ` -BackendPort $property.Value -FrontendPort $property.Value ` -Probe $probe -BackendAddressPool $backendPool -LoadDistribution SourceIPProtocol | Out-Null } } } Write-CustomHost -Message "Saving changes to $LoadBalancerName..." $loadBalancer = Set-AzLoadBalancer -LoadBalancer $loadBalancer } } |