PowerShell 5.1下并行执行SQL文件Trunk位置检索脚本优化需求

现有PowerShell脚本用于检查每个SQL文件是否存在于20个Trunk位置之一（遍历SQL名称的所有组合$res），但SQL文件数量较多时执行速度太慢。需要做两个优化：

• 一旦找到SQL文件所在的Trunk位置，立刻终止当前所有循环并记录该Trunk名称
• 在PowerShell 5.1环境下实现并行执行提速（无法升级到PowerShell 7）

原串行脚本如下：

write-host "Starting to search sqls in all trunk locations"
foreach ($sql in $sqls) {
    Write-Host "Working on $($sql.name)" -ForegroundColor Yellow
    $results = ConvertTo-DecodedPath $sql.Name
    foreach ($trunk in $TrunkList) {
        Write-Host "Working on $trunk" -ForegroundColor Cyan
        foreach ($res in $results) {
            
            $searchedText = ("File Name" + "=" + '"' + $res + '"').Replace("/", "\")
            write-host "Working on $searchedText"
            $xmlPathInTfs = $trunk + '/' + "InstallationFiles.xml"
            $checkFileContentTfsResult = CheckFileExistsTfs $xmlPathInTfs $searchedText
            if ($checkFileContentTfsResult.SearchedTextFound) {
                write-host "[$($sql.name)] was found in [$trunk]" -ForegroundColor green
                break
            }
        } 
    }
}

1. 先解决提前终止多层循环的问题 #

原脚本里的break只能终止最内层的$res循环，就算找到匹配项，还是会继续检查其他Trunk。用**标签（Label）**可以直接跳出多层循环，同时记录找到的Trunk：

write-host "开始在所有Trunk位置搜索SQL文件" -ForegroundColor Cyan
foreach ($sql in $sqls) {
    Write-Host "正在处理SQL文件: $($sql.name)" -ForegroundColor Yellow
    $results = ConvertTo-DecodedPath $sql.Name
    $foundTrunk = $null
    # 定义外层循环标签，用于直接跳出所有循环
    :trunkLoop foreach ($trunk in $TrunkList) {
        Write-Host "检查Trunk: $trunk" -ForegroundColor Cyan
        :resLoop foreach ($res in $results) {
            
            $searchedText = ("File Name" + "=" + '"' + $res + '"').Replace("/", "\")
            write-host "搜索文本: $searchedText"
            $xmlPathInTfs = $trunk + '/' + "InstallationFiles.xml"
            $checkFileContentTfsResult = CheckFileExistsTfs $xmlPathInTfs $searchedText
            if ($checkFileContentTfsResult.SearchedTextFound) {
                $foundTrunk = $trunk
                Write-Host "[$($sql.name)] 在 [$trunk] 中找到" -ForegroundColor Green
                # 直接跳出trunkLoop标签对应的外层循环，停止后续检查
                break trunkLoop
            }
        } 
    }
    # 把结果写入文件（可根据需求改成其他存储方式）
    if ($foundTrunk) {
        Add-Content -Path "SQL_Trunk_Mapping.txt" -Value "$($sql.name),$foundTrunk"
    } else {
        Add-Content -Path "SQL_Trunk_Mapping.txt" -Value "$($sql.name),未找到匹配Trunk"
    }
}

2. PowerShell 5.1 并行执行实现 #

PowerShell 5.1没有ForEach-Object -Parallel这种原生并行命令，得用Runspace池来实现多线程并行任务，避免单线程逐个等待的耗时。以下是完整的并行优化脚本：

# 初始化Runspace池：最小1线程，最大10线程（可根据服务器CPU核心数调整，比如4核设6-8）
$runspacePool = [RunspaceFactory]::CreateRunspacePool(1, 10)
$runspacePool.Open()
$jobs = @()

write-host "开始并行搜索SQL文件在所有Trunk位置" -ForegroundColor Cyan

foreach ($sql in $sqls) {
    # 复制变量解决闭包引用问题，否则所有并行任务会共用同一个$sql变量
    $sqlCopy = $sql
    # 创建并行任务脚本块
    $job = [PowerShell]::Create().AddScript({
        param($sql, $TrunkList)
        $results = ConvertTo-DecodedPath $sql.Name
        $foundTrunk = $null
        :trunkLoop foreach ($trunk in $TrunkList) {
            :resLoop foreach ($res in $results) {
                $searchedText = ("File Name" + "=" + '"' + $res + '"').Replace("/", "\")
                $xmlPathInTfs = $trunk + '/' + "InstallationFiles.xml"
                $checkFileContentTfsResult = CheckFileExistsTfs $xmlPathInTfs $searchedText
                if ($checkFileContentTfsResult.SearchedTextFound) {
                    $foundTrunk = $trunk
                    break trunkLoop
                }
            }
        }
        # 返回结构化结果
        [PSCustomObject]@{
            SQLName = $sql.Name
            FoundTrunk = $foundTrunk ?? "未找到匹配Trunk"
        }
    }).AddArgument($sqlCopy).AddArgument($TrunkList)
    
    $job.RunspacePool = $runspacePool
    $jobs += @{ Job = $job; Handle = $job.BeginInvoke() }
}

# 等待所有并行任务完成，获取结果并处理
foreach ($jobItem in $jobs) {
    $result = $jobItem.Job.EndInvoke($jobItem.Handle)
    $jobItem.Job.Dispose()
    Write-Host "[$($result.SQLName)] 匹配结果: $($result.FoundTrunk)" -ForegroundColor Green
    # 写入结果文件
    Add-Content -Path "SQL_Trunk_Mapping.txt" -Value "$($result.SQLName),$($result.FoundTrunk)"
}

# 清理资源
$runspacePool.Close()
$runspacePool.Dispose()

注意事项 #

• 线程数调整：CreateRunspacePool(1,10)的第二个参数是最大线程数，不要设得比CPU核心数太多，否则会导致资源竞争反而变慢。
• 变量复制：必须把$sql复制成$sqlCopy，否则并行任务会因为闭包引用问题拿到错误的变量值。
• 结果持久化：脚本里用Add-Content写入文本文件，你可以根据需求改成写入数据库或者其他存储。

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