PowerShell 7中如何退出ForEach-Object Parallel并更新运行空间外变量？

问题描述 #

需求：遍历$loopMe集合时，一旦检测到大于3的数字，立即终止所有并行运行的ForEach-Object循环，并将numberOverLimit变量设为true。但现有代码无法实现预期效果。

原代码 #

$loopMe = [System.Collections.ArrayList]@()
for($index = 0; $index -lt 5; $index++){
  $loopMe.Add($index)>$null;
}
$global:numberOverLimit=$false
$addToMe= [System.Collections.Concurrent.ConcurrentBag[psobject]]::new() 
$loopMe | Foreach-Object -ThrottleLimit 6 -Parallel{
  $localAddToMe=$using:addToMe
  Write-Host $_
  if($_ -gt 3){
    $global:numberOverLimit=$true
    break
  }
  $localAddToMe.Add($_)
}
write-Host $addToMe
Write-Host $numberOverLimit
if($numberOverLimit){
    Write-Host "A number was over the limit"
    exit
}
else{
    Write-Host "All is good"
}

预期结果 #

0
1
2
3
3 2 1 0
True
A number was over the limit

实际结果 #

0
1
4
2
3
3 2 1 0
False
All is good

问题原因 #

1. 全局变量不共享：$global:numberOverLimit在每个并行运行空间中是独立副本，修改的并非主运行空间的变量，因此主空间中该变量始终为false。
2. break作用范围有限：break只能终止当前运行空间内的循环，无法影响其他并行执行的任务。
3. 并行执行无序：ForEach-Object -Parallel的执行顺序不固定，因此输出顺序与预期存在差异。

解决方案 #

使用线程安全的同步事件（ManualResetEventSlim）实现跨运行空间的终止信号，确保所有并行任务能及时响应终止指令，同时主运行空间可准确检测状态。

修改后的代码 #

$loopMe = [System.Collections.ArrayList]@()
for($index = 0; $index -lt 5; $index++){
  $loopMe.Add($index) > $null
}

# 初始化线程安全的终止信号，初始状态为未触发
$terminateEvent = [System.Threading.ManualResetEventSlim]::new($false)
$numberOverLimit = $false
$addToMe = [System.Collections.Concurrent.ConcurrentBag[psobject]]::new() 

$loopMe | Foreach-Object -ThrottleLimit 6 -Parallel {
  $localAddToMe = $using:addToMe
  $localTerminateEvent = $using:terminateEvent

  # 先检查终止信号，若已触发则直接退出当前任务
  if ($localTerminateEvent.IsSet) { return }

  Write-Host $_
  if ($_ -gt 3) {
    # 设置终止信号，通知所有并行任务停止执行
    $localTerminateEvent.Set()
    # 可选：是否将超限数字加入集合，根据需求调整
    $localAddToMe.Add($_) > $null
    return
  }

  # 再次检查信号，避免在判断数字后、添加集合前触发终止
  if (-not $localTerminateEvent.IsSet) {
    $localAddToMe.Add($_) > $null
  }
}

# 主运行空间检测终止信号状态，确定是否存在超限数字
$numberOverLimit = $terminateEvent.IsSet
# 释放同步事件资源
$terminateEvent.Dispose()

write-Host $addToMe
Write-Host $numberOverLimit
if ($numberOverLimit) {
    Write-Host "存在超过限制的数字"
    exit
} else {
    Write-Host "所有数字均符合要求"
}

方案说明 #

• 同步信号跨空间生效：ManualResetEventSlim是线程安全的同步原语，所有并行运行空间共享同一实例，触发后所有任务都能检测到状态变化。
• 提前终止无效任务：每个并行任务执行前先检查终止信号，避免不必要的计算。
• 状态准确同步：主运行空间通过信号的IsSet属性直接获取状态，无需依赖变量副本同步。

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