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

时间:2015-08-05 06:16:25

标签: powershell parameter-passing

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

我一直在添加for loop以使用ParameterSetName为多个服务器运行该函数。

在添加ParamSets之前,我只需设置一个函数,并在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.

1 个答案:

答案 0 :(得分:0)

制作3个(或更多)不同的功能。参数设置使用方式过于复杂,使您难以调试。

我会制作一个Get-ScheduledRebootClear-ScheduledRebootSet-ScheduledReboot个函数集合。

此脚本中有一些错误导致您遇到问题。第一个是nuke参数集,不包含服务器参数。另一个错误是让你失望的错误。您可以在上面定义$ server参数,但在函数中,您可以从文本文件中更改齿轮和加载。您将$server$servers$comp变量混合在一起导致其失败。

您应该将您的[string]$server更改为[string[]]$server或更好,[string[]$ComputerNameforeach($node in $ComputerName){...}。保持文本文件加载到该函数之外。