powershell - 使用ParamSets和for循环的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，$servers和$comp变量，导致其失败。

您应该将[string]$server更改为[string[]]$server或更好，集合上的[string[]$ComputerName和foreach($node in $ComputerName){...}。将文本文件加载到该函数之外。