Public/Wait-RSJob.ps1
|
Function Wait-RSJob { <# .SYNOPSIS Waits until all RSJobs are in one of the following states: .DESCRIPTION Waits until all RSJobs are in one of the following states: .PARAMETER Name The name of the jobs to query for. .PARAMETER ID The ID of the jobs that you want to wait for. .PARAMETER InstanceID The GUID of the jobs that you want to wait for. .PARAMETER State The State of the job that you want to wait for. Accepted values are: NotStarted Running Completed Failed Stopping Stopped Disconnected .PARAMETER Batch Name tof the set of RSJobs .PARAMETER HasMoreData Waits for jobs that have data being outputted. You can specify -HasMoreData:$False to wait for jobs that have no data to output. .PARAMETER Timeout Timeout after specified number of seconds. This is a global timeout meaning that it is not a per job timeout. .PARAMETER ShowProgress Displays a progress bar .NOTES Name: Wait-RSJob Author: Ryan Bushe/Boe Prox Notes: This function is a slightly modified version of Get-RSJob by Boe Prox.(~10 lines of code changed) .EXAMPLE Get-RSJob | Wait-RSJob Description ----------- Waits for jobs which have to be completed. #> [cmdletbinding( DefaultParameterSetName='All' )] Param ( [parameter(ValueFromPipeline=$True,ParameterSetName='Job')] [RSJob[]]$Job, [parameter(ValueFromPipelineByPropertyName=$True, ParameterSetName='Name')] [string[]]$Name, [parameter(ValueFromPipelineByPropertyName=$True, ParameterSetName='Id')] [int[]]$Id, [parameter(ValueFromPipelineByPropertyName=$True, ParameterSetName='Guid')] [guid[]]$InstanceID, [parameter(ValueFromPipelineByPropertyName=$True, ParameterSetName='Batch')] [string[]]$Batch, [parameter(ParameterSetName='Batch')] [parameter(ParameterSetName='Name')] [parameter(ParameterSetName='Id')] [parameter(ParameterSetName='Guid')] [parameter(ParameterSetName='All')] [ValidateSet('NotStarted','Running','Completed','Failed','Stopping','Stopped','Disconnected')] [string[]]$State, [parameter(ParameterSetName='Batch')] [parameter(ParameterSetName='Name')] [parameter(ParameterSetName='Id')] [parameter(ParameterSetName='Guid')] [parameter(ParameterSetName='All')] [Switch]$HasMoreData, [int]$Timeout, [switch]$ShowProgress ) Begin { If ($PSBoundParameters['Debug']) { $DebugPreference = 'Continue' } Write-Verbose "ParameterSet: $($PSCmdlet.parametersetname)" $List = New-Object System.Collections.ArrayList $WhereList = New-Object System.Collections.ArrayList #Take care of bound parameters If ($PSBoundParameters['Name']) { [void]$list.AddRange($Name) } If ($PSBoundParameters['Batch']) { [void]$list.AddRange($Batch) } If ($PSBoundParameters['Id']) { [void]$list.AddRange($Id) } If ($PSBoundParameters['InstanceId']) { [void]$list.AddRange($InstanceId) } If ($PSBoundParameters['Job']){ [void]$list.AddRange($Job) } } Process { If ($PSCmdlet.ParameterSetName -ne 'All') { Write-Verbose "Adding $($_)" [void]$List.Add($_) } } End { If ($List.count -eq 0) { return } Switch ($PSCmdlet.parametersetname) { 'Name' { $Items = '"{0}"' -f (($list | ForEach-Object {"^{0}$" -f $_}) -join '|') -replace '\*','.*' [void]$WhereList.Add("`$_.Name -match $Items") } 'Id' { $Items = '"{0}"' -f (($list | ForEach-Object {"^{0}$" -f $_}) -join '|') [void]$WhereList.Add("`$_.Id -match $Items") } 'Guid' { $Items = '"{0}"' -f (($list | ForEach-Object {"^{0}$" -f $_}) -join '|') [void]$WhereList.Add("`$_.InstanceId -match $Items") } 'Job' { $Items = '"{0}"' -f (($list | Select-Object -ExpandProperty Id | ForEach-Object {"^{0}$" -f $_}) -join '|') [void]$WhereList.Add("`$_.id -match $Items") } 'Batch' { $Items = '"{0}"' -f (($list | ForEach-Object {"^{0}$" -f $_}) -join '|') [void]$WhereList.Add("`$_.batch -match $Items") } } If ($PSBoundParameters['State']) { [void]$WhereList.Add("`$_.State -match `"$($State -join '|')`"") } If ($PSBoundParameters.ContainsKey('HasMoreData')) { [void]$WhereList.Add("`$_.HasMoreData -eq `$$HasMoreData") } If ($WhereList.count -gt 0) { $WhereString = $WhereList -join ' -AND ' $ScriptBlock = [scriptblock]::Create($WhereString) } If ($ScriptBlock) { Write-Verbose "WhereString: $($WhereString)" Write-Verbose "Using scriptblock" $FilteredJobs = @($list | Where-Object $ScriptBlock) $WaitJobs = $FilteredJobs $TotalJobs = $FilteredJobs.Count Write-Verbose "$($FilteredJobs.Count)" $Date = Get-Date $Completed = 0 Do{ #only ever check $WaitJobs State once per loop, and do all operations based on that snapshot to avoid bugs where the state of a job may have changed mid loop $JustFinishedJobs = New-Object System.Collections.ArrayList $RunningJobs = New-Object System.Collections.ArrayList ForEach ($WaitJob in $WaitJobs) { If($WaitJob.State -match 'Completed|Failed|Stopped|Suspended|Disconnected' -and $WaitJob.Completed) { [void]$JustFinishedJobs.Add($WaitJob) } Else { [void]$RunningJobs.Add($WaitJob) } } $WaitJobs = $RunningJobs $JustFinishedJobs $Completed += $JustFinishedJobs.Count Write-Verbose "Wait: $($Waitjobs.Count)" Write-Verbose "Completed: ($Completed)" Write-Verbose "Total: ($Totaljobs)" Write-Verbose "Status: $($Completed/$TotalJobs)" If ($PSBoundParameters.ContainsKey('ShowProgress')) { Write-Progress -Activity "RSJobs Tracker" -Status ("Remaining Jobs: {0}" -f $Waitjobs.Count) -PercentComplete (($Completed/$TotalJobs)*100) } if($Timeout){ if((New-Timespan $Date).TotalSeconds -ge $Timeout){ $TimedOut = $True break } } } While($Waitjobs.Count -ne 0) } } } |