Invoke-xRDS_SessionHostMaintenance.ps1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
function Invoke-xRDS_SessionHostMaintenance {

<#
    .DESCRIPTION
    Sets RDS Session Hosts Maintenance mode.
  
    .PARAMETER Broker
    -ConnectionBroker - FQDN of RDS ConnectionBroker.
 
    .PARAMETER Maintenance
    -Maintenance - Enable (no new logons), Disable (Allow new logons), Reboot (no new logons until reboot).
 
    .PARAMETER Credential
    -Credential [Optional] - Query RDS Connection Broker resources under provided credentials, the same credentials will be used to set maintenance mode too.
 
    .EXAMPLE
    # Sets RDS Session host for selected servers not allow new connections until the next reboot of the hosts:
    Invoke-xRDS_SessionHostMaintenance -ConnectionBroker ardscbl01.adatum.labnet -Maintenance Reboot
#>



    [CmdletBinding()]
    Param(
        [Parameter(Mandatory=$true)][string]$ConnectionBroker,    
        [Parameter(Mandatory=$false)]
        [ValidateSet('Enable','Disable','Reboot')]
        [string]$Maintenance = 'Disable',
        [PSCredential]$Credential
    )      

    $Collection = Get-xRDS_CollectionsList -ConnectionBroker $ConnectionBroker -Credential $Credential

    #Attribute values for maintenance mode
    switch ($Maintenance) {
                'Enable'           {$attributeSyntax = 2}
                'Disable'          {$attributeSyntax = 0}
                'Reboot'           {$attributeSyntax = 1}
                Default {}
            }    


    Try {

    $ActiveManagementServer =  (Invoke-WmiMethod -ComputerName $ConnectionBroker -Class "Win32_RDMSEnvironment" -Namespace "root\CIMV2\rdms" -Name "GetActiveServer").ServerName

    $SelectedServers = Get-WmiObject -ComputerName $ActiveManagementServer -Credential $Credential -Query "SELECT ServerName,LoadIndicator,NumberOfSessions,NumPendRedir,ServerWeight,SingleSessionMode FROM Win32_SessionDirectoryServer" | select `
    @{Label="Host";Expression={$_.ServerName}}, @{Label="Collections";Expression={$Collection[$_.ServerName]}},@{Label="Sessions";Expression={$_.NumberOfSessions}}, `
    @{Label="DrainMode";Expression={Convert-xRDS_DrainMode((Get-WmiObject -Credential $Credential -Class "Win32_TerminalServiceSetting" -Namespace "root\CIMV2\terminalservices" -ComputerName $_.ServerName).SessionBrokerDrainMode)}},`
    @{Label="LoadIndicator";Expression={$_.LoadIndicator}},@{Label="ServerWeight";Expression={$_.ServerWeight}},`
    @{Label="NumPendRedir";Expression={$_.NumPendRedir}},@{Label="SingleSessionMode";Expression={$_.SingleSessionMode}}    | sort Host | Out-GridView -PassThru -Title "Host List (Select Servers)"    

    If ($SelectedServers.Host.Count -le 0) {Write-Host "No hosts have been selected." -ForegroundColor Cyan }
    else { Write-Host "Maintenance has been set: $Maintenance for the following hosts" -ForegroundColor Cyan ; $SelectedServers.Host | foreach {Write-Host "$_" -ForegroundColor Cyan ; Set-WmiInstance -Credential $Credential -Class "Win32_TerminalServiceSetting" -Namespace "root\CIMV2\terminalservices" -Arguments @{SessionBrokerDrainMode=$attributeSyntax} -ComputerName $_ | Out-Null }}   
    
    } Catch {Write-host $_.Exception.message -ForegroundColor Red}
}