PowerShell 7.3.3多线程脚本内存无法释放问题求助

问题背景 #

在PowerShell 7.3.3中编写了与Microsoft ExchangeOnline交互的多线程脚本，采用带ReuseThread选项的RunspacePool复用工作线程上下文。脚本执行完毕后，已执行关闭释放资源、移除变量、调用垃圾回收等操作，但PowerShell进程仍占用7GB内存，需排查遗漏的操作。

脚本代码 #

$outputStream = New-Object -Typename  System.Management.Automation.PSDataCollection[PSObject]
$initialState = [InitialSessionState]::CreateDefault();
$initialState.ImportPSModule("ExchangeOnlineManagement")
$maxThreads = 20
$RunspacePool = [runspacefactory]::CreateRunspacePool($initialState)
$RunspacePool.SetMaxRunspaces($maxThreads);
$RunspacePool.SetMinRunspaces($maxThreads);
$RunspacePool.ThreadOptions = 2 # ReuseThread
$RunspacePool.Open()
[Collections.Arraylist]$traitements = @()
# Create contacts ExchangeOnline
Import-csv $ImportFile -Delimiter ";" | ForEach-Object { 
    $traitements += New-Object -TypeName PSObject -Property @{
        Runspace   = $null
        PowerShell = [powershell]::Create()
    }
    $traitements[-1].PowerShell.RunspacePool = $RunspacePool
    $traitements[-1].PowerShell.AddScript($blockTraitement).AddParameters(@{
            "name"                         = $_.cn
            "email"                        = $_.mail
            "firstName"                    = $_.givenName
            "lastName"                     = $_.surName
            "extensionAttribute4"          = $_.extensionAttribute4
            "msExchSenderHintTranslations" = $_.msExchSenderHintTranslations
            "description"                  = $_.Description
            "title"                        = $_.title
            "company"                      = $_.company
            "postalcode"                   = $_.postalcode
            "city"                         = $_.l
            "departement"                  = $_.departement
        })
    $traitements[-1].Runspace = $traitements[-1].PowerShell.BeginInvoke()
}

# EndInvoke and dispose
while ($traitements) {
    foreach ($runspace in $traitements.ToArray()) {
        if ($runspace.Runspace.IsCompleted) {
            Write-Host "RUNSPACE is completed " + $runspace.Runspace.IsCompleted
            $outputStream = $runspace.PowerShell.EndInvoke($runspace.Runspace)
            $outputStream | ForEach-Object {
                Write-Host $_
            }
            $runspace.PowerShell.Stop()
            $runspace.PowerShell.Dispose()
            $traitements.Remove($runspace)
        }
        Start-Sleep -Milliseconds 100
    }
}

$outputStream.Dispose()
$RunspacePool.Close()
$RunspacePool.Dispose()
Clear-Variable -Name initialState,RunspacePool,traitements,outputStream -Confirm:$false
Remove-Variable -Name initialState,RunspacePool,traitements,outputStream -Confirm:$false
[System.GC]::Collect()
[System.GC]::WaitForPendingFinalizers()
[System.GC]::Collect()

遗漏的操作及解决方案 #

• 显式断开ExchangeOnline连接：
  每个Runspace执行的$blockTraitement中，完成联系人创建后未断开ExchangeOnline会话。由于启用了ReuseThread，线程上下文会保留连接会话，导致内存无法释放。需在$blockTraitement末尾添加：

  Disconnect-ExchangeOnline -Confirm:$false -ErrorAction SilentlyContinue
  

  确保每个线程的Exchange连接被主动断开。

• 正确处理EndInvoke返回的PSDataCollection：
  原代码中用$outputStream = $runspace.PowerShell.EndInvoke(...)覆盖初始创建的$outputStream，导致之前的PSDataCollection实例未被释放。应单独处理每个EndInvoke返回的输出集合，处理完成后立即Dispose：

  # 替换原输出处理代码
  $currentOutput = $runspace.PowerShell.EndInvoke($runspace.Runspace)
  $currentOutput | ForEach-Object { Write-Host $_ }
  $currentOutput.Dispose() # 释放当前输出集合
  

• 优化InitialSessionState的模块导入：
  原代码导入整个ExchangeOnlineManagement模块到Runspace，会加载大量未使用的命令和资源，增加内存占用。改为只导入所需命令：

  # 替换原模块导入代码
  $initialState.ImportPSModule(@{
      ModuleName = "ExchangeOnlineManagement"
      Cmdlets = @("New-MailContact", "Disconnect-ExchangeOnline") # 仅导入需要的命令
  })
  

• 清理Runspace对象的引用：
  在移除$runspace对象前，需清空其属性引用，避免残留对象被隐式持有：

  # 在$traitements.Remove($runspace)前添加
  $runspace.Runspace = $null
  $runspace.PowerShell = $null
  

• 强化垃圾回收操作：
  原垃圾回收调用可指定回收代并等待完全回收完成，确保内存被彻底释放：

  # 替换原垃圾回收代码
  [System.GC]::Collect(2, [System.GCCollectionMode]::Forced)
  [System.GC]::WaitForPendingFinalizers()
  [System.GC]::Collect(2, [System.GCCollectionMode]::Forced)
  

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