PowerShell 向上/向下提示

看来你们中的许多人都对 PowerShell 提示功能的可能性着迷。在之前的文章中，我提到了这样一个事实：您可以在提示函数中执行几乎任何操作。今天我有一个例子来说明我正在谈论的内容。创建功能更多的提示函数的挑战性部分是您希望它运行得非常非常快。您最不想看到的就是等待几秒钟提示返回。考虑到这一点，这里有一个提示功能，将显示一小部分服务器正在运行或已关闭。

当我第一次开始处理这个问题时，我打电话给 Test-WSMan 来验证 4 台计算机是否已启动并运行。然而，每台服务器需要花费几秒钟的时间，这使得整个提示功能慢得令人沮丧。我考虑过在后台作业中运行测试并显示上一个作业的结果，但这也非常慢。然后我想到了使用运行空间和同步哈希表的想法。这是一个高级主题，所以如果您没有遵循所有内容，请不要感到难过。

将运行空间视为一个单独且隐藏的 PowerShell 窗口。在运行空间中，我运行代码，每 5 秒使用 Test-WSMan “ping”一次服务器。这似乎是一个合理的时间。我的提示功能并非旨在作为高可用性解决方案。运行空间更新与我的控制台会话同步的哈希表变量。这意味着我可以访问哈希表并更新我的提示函数。该函数的代码可以在 Github 上找到，也可以从这里复制。

UpDownPrompt.ps1:

function prompt {

    $up = 0x25b2 -as [char]
    $down = 0x25bc -as [char]

    Try {
        Get-Variable -Name testHash -Scope global -ErrorAction Stop | Out-Null
    }
    catch {
         #create the runspace and synchronized hashtable
        $global:testHash = [hashtable]::Synchronized(@{Computername = $env:computername;results = ""; date= (Get-Date)})
        $newRunspace = [runspacefactory]::CreateRunspace()
        #set apartment state if available
        if ($newRunspace.ApartmentState) {
            $newRunspace.ApartmentState = "STA"
        }
        $newRunspace.ThreadOptions = "ReuseThread"         
        $newRunspace.Open()
        $newRunspace.SessionStateProxy.SetVariable("testHash", $testHash)  

        $pscmd = [PowerShell]::Create().AddScript( {  
                #define the list of computers to test
                $computers = "dom1", "srv1", "srv2", "srv3"

                do {
                    $results = $computers | ForEach-Object {
                        [pscustomobject]@{
                            Computername = $_.toupper()
                            Responding   = Test-WSMan -ComputerName $_
                        }
                    }

                    $global:testHash.results = $results
                    $global:testHash.date = Get-Date
                    #set a sleep interval between tests
                    Start-Sleep -Seconds 5
                } while ($True)

            })

        $pscmd.runspace = $newrunspace
        [void]$psCmd.BeginInvoke()

    } #catch
    Write-Host "[" -NoNewline

    $global:testHash.results.foreach( {
            Write-Host $_.Computername -NoNewline
            if ($_.responding) {
                Write-Host $up -ForegroundColor green -NoNewline
            }
            else {
                Write-Host $down -ForegroundColor red -NoNewline
            }
        })

    Write-Host "]" -NoNewline

    "PS $($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) "

}

因为我使用的是 Test-WSMan，所以这不适用于非 Windows 平台。但它将在 Windows 上的 PowerShell Core 上运行。我已经对我的服务器名称进行了硬编码，因此您需要编辑该函数。您还可以修改代码以将名称作为运行空间脚本的参数传递。

首次运行提示功能时，可能需要 5-10 秒才能得到结果。

同步哈希的好处是我可以看到结果。这是 PowerShell Core 上的提示。

如果您开始遇到问题，可以从会话中删除 $TestHash 变量。下次按 Enter 时，提示功能将重新创建它。或者启动一个新的PowerShell会话并加载提示功能。

本周晚些时候我将提供一些更有趣且可能有用的提示功能。享受！