Powershell 使用 arg 启动/停止

Question

我在 test.ps1 中编写了以下函数，我想在运行 thsi 脚本启动/停止/..时做出选择：

function getState($SeviceName) { 
    $server = @('host_1', 'host_2')

    # get status 
    $server | % {Write-Host "verify: $_"; Get-Service -ComputerName $_ -Name SeviceName
}

1. 我想提供$ServiceName 作为参数（带标准输入）我该怎么做？ => 类似选择 1 开始 2 停止 ...

2. 在 Powershell 中使用 switch/case

   $doAction = {"Stop-Service", "Start-service"}
   $server | % {Write-Host "verify: $_"; Get-Service -ComputerName $_ -Name SeviceName | $doAction}
   

如何使用开关选择开始或停止？

Answer 1

这是一个可以满足您要求的功能：

function Get-State {
    [CmdletBinding()]
    [OutputType('System.ServiceProcess.ServiceController')]
    param(
        [Parameter(Position = 0, Mandatory)]
        [ValidateSet('Start', 'Stop', 'Get')]
        [string] $Action,

        [Parameter(Position = 1, ValueFromPipeline, Mandatory)]
        [ValidateNotNullOrEmpty()]
        [string] $ServiceName
    )

    begin {
        $serverList = @('host_1', 'host_2')
    }

    process {
        foreach ($server in $serverList) {
            try {
                $svc = Get-Service -ComputerName $server -Name $ServiceName -ErrorAction Stop
            } catch {
                throw "Failed to find service $ServiceName on $server! $PSItem"
            }

            switch ($Action) {
                'Start' { $svc | Start-Service -PassThru }
                'Stop'  { $svc | Stop-Service -Force -PassThru }
                default { $svc }
            }
        }
    }
}

它利用高级功能特性和属性来获取管道输入（用您的话来说是stdin）。我建议reading this documentation.

Answer 2

您可以通过向脚本添加参数来向脚本添加参数。 在脚本文件的顶部放置：

Param
(
    [parameter()]
    [String[]]$YourArgumentVariable

    [parameter()]
    [switch] $MySwitch
)

对于函数，它紧跟在函数定义之后。所以在你的情况下：

function getState($SeviceName) { 
    Param
    (
        [parameter()]
        [String[]]$server

        [parameter()]
        [switch] $MySwitch
    )
    # get status 
    $server | % {Write-Host "verify: $_"; Get-Service -ComputerName $_ -Name SeviceName
}

开关基本上将布尔值设置为真或假。 因此，如果您使用 -MySwitch 调用脚本，它会将变量 $MySwitch 设置为 true。否则它将保持错误。

Don Jones 写了一封 good getting started article on paramters，我建议您查看一下。

请注意，您可以在参数中定义很多东西。就像如果你想确保它总是被填满，你可以设置

 [parameter(Mandatory=$true)]

这只是您可以使用参数执行的众多示例之一。