如何优化PowerShell IP校验函数以适配超5万行IP的大体积IP池场景

原有实现的核心问题 #

• 时间复杂度高：每次校验对全量IP池字符串执行Contains匹配，复杂度为O(n)，IP数量增长时耗时线性上升
• 匹配结果不准确：字符串模糊匹配会出现误判，例如待校验IP为192.168.1.1时，IP池存在192.168.1.10也会返回true
• 若每次调用函数都重复读取IP池文件，额外增加了大量IO开销，进一步放大性能问题

优化后实现 #

核心思路是首次调用时一次性将IP池的所有IP加载到HashSet<string>结构中，后续所有查询的时间复杂度为O(1)，同时支持精确匹配，避免误判。

# 脚本作用域变量缓存IP池哈希集，避免重复加载，也可替换为$global:前缀声明全局可用
$script:IPPoolHashSet = $null
# 可选：记录IP池文件最后修改时间，文件更新时自动重载
$script:IPPoolLastWriteTime = $null

function LocateIP {
    param(
        # IP池文件路径
        [Parameter(Mandatory=$true)]
        [String]$IPPoolPath,
        # 待校验的目标IP
        [Parameter(Mandatory=$true)]
        [String]$IP
    )

    $currentFileTime = (Get-Item $IPPoolPath).LastWriteTimeUtc
    # 哈希集未初始化 或 IP池文件已更新时，重新加载IP池
    if ($null -eq $script:IPPoolHashSet -or $currentFileTime -ne $script:IPPoolLastWriteTime) {
        # 读取IP池所有行，过滤空行和空白字符后生成哈希集，忽略IP大小写（实际IP无大小写，适配特殊格式场景）
        $ipList = Get-Content -Path $IPPoolPath | ForEach-Object { $_.Trim() } | Where-Object { $_ -ne '' }
        $script:IPPoolHashSet = [System.Collections.Generic.HashSet[string]]::new($ipList, [System.StringComparer]::OrdinalIgnoreCase)
        $script:IPPoolLastWriteTime = $currentFileTime
    }

    return $script:IPPoolHashSet.Contains($IP)
}

额外性能提升建议 #

• 单次需要校验大量IP时，不要循环调用LocateIP，可直接复用已加载的$script:IPPoolHashSet批量比对，性能可再提升30%以上
• 若IP池存储的是CIDR网段而非单个IP，可将网段转成IP整数范围存储，查询时转成整数做范围匹配即可

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