使用带有 for 循环的 ParamSet 的 PowerShell 函数

Question

我整理了一个用于安排服务器重启的 powershell 函数。

我坚持要添加 for loop 以使用 ParameterSetName 为多个服务器运行该函数。

在添加 ParamSet 之前，我只是设置了一个函数，并在 for 循环内的同一脚本中调用了该函数，效果很好。

Function ScheduleReboot{
[CmdletBinding(SupportsShouldProcess=$true,DefaultParameterSetName="ViewOnly")]
Param(
    [Parameter(Mandatory=$false,ParameterSetName="KillSwitch")]
    [Switch]$NukeJobs,

    [Parameter(Mandatory=$true,
    HelpMessage="ServerName goes here",ParameterSetName="FullRun")]
    [string]$server,

    [Parameter(Mandatory=$true,
        HelpMessage="Enter a Date/Time 07-28-15 16:00 For July 28th, 2015 at 4:00 PM",
        ParameterSetName="FullRun"
    )]
    [ValidatePattern('(?#Enter MM-dd-YY HH:MM 24hr clock)\d{2}-\d{2}-\d{2}\s\d{2}[:]\d{2}')]
    $date,

    [Parameter(Mandatory=$false,ParameterSetName="ViewOnly")]
    [switch]$ViewOnly

) #"$server will reboot on $date"

    Switch ($PSCmdlet.ParameterSetName){
        "KillSwitch"{
            if($env:PROCESSOR_ARCHITECTURE -eq "x86")
            {
                set-alias ps64 "$env:windir\sysnative\WindowsPowerShell\v1.0\powershell.exe"
                $script2 = [ScriptBlock]::Create("IF(Test-Path -Path C:\Windows\System32\Tasks\Microsoft\Windows\PowerShell\ScheduledJobs\$server) 
                {Remove-Item -Force -Path C:\Windows\System32\Tasks\Microsoft\Windows\PowerShell\ScheduledJobs\$server}")
                "Removing previous scheduled server reboot data"
                ps64 -command $script2
            } else {
                Get-ScheduledJob | Unregister-ScheduledJob
            }
        }#End killswitch block.

        "FullRun" {
            [array]$servers = gc D:\Scripts\servers.txt
            [array]$dates = gc D:\Scripts\dates.txt
            if($dates.Length -eq $servers.Length)
            {
                "Input count matches"
                for($i=0;$i -lt $servers.Length;$i++)
                {
                    $day =  $dates[$i]
                    $comp = $servers[$i]
                    $user = Get-Credential -UserName $env:USERNAME -Message "UserName/password for scheduled Reboot"
                    $trigger = New-JobTrigger -once -at $day
                    $script = [ScriptBlock]::Create("D:\Scripts\Scheduled-Reboot-Single.ps1 -server $server | Out-File -Force \\SysLogSvr\d$\scripts\$server-Reboot.log")
                    Register-ScheduledJob -Name $comp -Credential $user -Trigger $trigger -ScriptBlock $script
                }#end for loop
            } else {
                $warn = "Server list contains " + $servers.Count + " items and Date list contains " + $dates.Count + " items, Please re-check!!!"
                Write-Warning $warn
            }
        }#end fullrun block.

        "ViewOnly" {
            Get-ScheduledJob | Get-JobTrigger
            Get-ScheduledJob | Select Name,Command | FT -AutoSize
        }#end viewonly block.
    }#End param switch
}#end function.

Answer 1

制作 3 个（或更多）不同的功能。该参数设置您使用它的方式过于复杂，并且难以调试。

我会创建一个Get-ScheduledReboot、Clear-ScheduledReboot、Set-ScheduledReboot 函数集合。

发布的此脚本中有一些错误会导致您出现问题。首先是nuke参数集不包含server参数。另一个错误是让您失望的错误。您在上面定义了一个 $server 参数，但在函数中，您更改齿轮并从文本文件加载。您正在混合 $server、$servers 和 $comp 变量导致它失败。

您应该将收藏中的[string]$server 更改为[string[]]$server 或更好的[string[]$ComputerName 和foreach($node in $ComputerName){...}。将文本文件加载保持在该函数之外。