PowerShell Core如何基于前置参数生成动态ValidateSet验证集

我们正在开发一个用于处理数百条结构化条目的PowerShell函数，数据条目结构与样例如下：

原始数据通过Import-Csv cmdlet从定期更新的CSV文件中获取。

我们尝试通过PowerShell类实现参数联动效果：用户选择State参数值后，City参数可动态加载对应州的城市作为可选验证值，当前已实现的最小可复现代码如下：

$Global:StateData = Import-Csv \\path\to\city-state-population.csv

class State : System.Management.Automation.IValidateSetValuesGenerator {
    [string[]] GetValidValues() {
        return (($Global:StateData).State | Select-Object -Unique)
    }
}
class City : System.Management.Automation.IValidateSetValuesGenerator {
    [string[]] GetValidValues($State) {
        return ($Global:StateData | Where-Object State -eq $State).City
    }
}
function Get-Population {
    param (
        # Name of the state
        [Parameter(Mandatory, Position = 0)]
        [ValidateSet([State])]
        $State,

        # Name of the city
        [Parameter(Mandatory, Position = 1)]
        [ValidateSet([City])]
        $City
    )
    
    $City | ForEach-Object {
        $TargetCity = $City | Where-Object City -match $PSItem
        "The population of $($TargetCity.City), $($TargetCity.State) is $($TargetCity.Population)."
    }
}

根据官方文档说明，GetValidValues()方法不支持传入参数。现有公开的参数依赖验证方案处理当前规模的数据时性能无法满足需求，需要可行的实现方案。

运行环境说明：当前使用环境为PowerShell Core，而非Windows PowerShell，后者不支持IValidateSetValuesGenerator接口。

核心优化点是提前构建O(1)时间复杂度的哈希索引，避免每次参数验证/补全时全表遍历数据，同时使用原生参数补全接口读取已绑定的参数值，不需要给验证方法传参，性能远高于传统的脚本块验证方案。

• 第一步：加载CSV时一次性构建两级索引，后续所有查找操作直接走哈希表，无遍历开销
• 第二步：州参数沿用IValidateSetValuesGenerator实现静态验证集
• 第三步：城市参数使用IArgumentCompleter接口实现动态补全，直接从参数绑定上下文读取用户已选择的州值，返回对应城市列表
• 第四步：城市参数增加轻量验证逻辑，同样基于预构建的哈希表做合法性校验，无性能损耗

可直接运行的实现代码如下：

# 加载CSV并预构建索引，仅执行一次
$Global:CityPopulationData = Import-Csv \\path\to\city-state-population.csv
$Global:StateCityMap = @{}
$Global:CityEntryMap = @{}
foreach ($entry in $Global:CityPopulationData) {
    # 构建州-城市集合索引
    if (-not $Global:StateCityMap.ContainsKey($entry.State)) {
        $Global:StateCityMap[$entry.State] = [System.Collections.Generic.HashSet[string]]::new()
    }
    $Global:StateCityMap[$entry.State].Add($entry.City) | Out-Null
    # 构建城市全信息索引
    $Global:CityEntryMap[($entry.State + '|' + $entry.City)] = $entry
}

# 州参数验证集生成
class StateValidSet : System.Management.Automation.IValidateSetValuesGenerator {
    [string[]] GetValidValues() {
        return $Global:StateCityMap.Keys
    }
}

# 城市参数动态补全器
class CityArgumentCompleter : System.Management.Automation.IArgumentCompleter {
    [System.Collections.Generic.IEnumerable[System.Management.Automation.CompletionResult]] CompleteArgument(
        [string]$commandName,
        [string]$parameterName,
        [string]$wordToComplete,
        [System.Management.Automation.Language.CommandAst]$commandAst,
        [System.Collections.IDictionary]$fakeBoundParameters
    ) {
        $completionResults = [System.Collections.Generic.List[System.Management.Automation.CompletionResult]]::new()
        $selectedState = $fakeBoundParameters['State']
        # 未选择州时不返回补全结果
        if ([string]::IsNullOrWhiteSpace($selectedState)) { return $completionResults }
        # 直接从预构建索引取对应州的城市列表，无全表遍历
        $matchedCities = $Global:StateCityMap[$selectedState].Where({ $_ -like "$wordToComplete*" })
        foreach ($city in $matchedCities) {
            $completionResults.Add([System.Management.Automation.CompletionResult]::new(
                $city, $city, [System.Management.Automation.CompletionResultType]::ParameterValue, $city
            ))
        }
        return $completionResults
    }
}

function Get-Population {
    param (
        [Parameter(Mandatory, Position = 0)]
        [ValidateSet([StateValidSet])]
        [string]$State,

        [Parameter(Mandatory, Position = 1)]
        [ArgumentCompleter([CityArgumentCompleter])]
        [ValidateScript({
            $selectedState = $PSBoundParameters['State']
            # 基于哈希集合做合法性校验，性能远高于Where-Object遍历
            if (-not $Global:StateCityMap[$selectedState].Contains($_)) {
                throw "输入的城市 $_ 不属于所选州 $selectedState，请重新输入"
            }
            $true
        })]
        [string]$City
    )

    $targetEntry = $Global:CityEntryMap[($State + '|' + $City)]
    return "The population of $($targetEntry.City), $($targetEntry.State) is $($targetEntry.Population)."
}

该方案即使数据量上涨到十万条级别，参数补全和验证也不会出现可感知的延迟，完全满足性能要求。

内容的提问来源于stack exchange，提问作者Ram Iyer