如何在PowerShell脚本的所有退出条件下运行清理代码？

类 Unix 系统的 Shell 中，trap 命令可以捕获各类终止信号（比如 Ctrl+C、kill 指令发送的信号）并执行清理逻辑，但 PowerShell 中的 trap 关键字仅能捕获脚本内的错误，更像是 try/catch 的简化语法。而用于全局事件处理的 Register-EngineEvent 存在诸多局限：仅支持全局引擎事件、清理依赖不稳定的 try/finally、外部终止信号会导致事件注册悬空，后续脚本运行报错。

以下是几种摆脱这一困境的方法：

1. 处理 Ctrl+C 信号的专用逻辑 #

通过设置 [Console]::TreatControlCAsInput 将 Ctrl+C 转为输入而非直接终止进程，配合循环监听输入实现自定义清理：

[Console]::TreatControlCAsInput = $true
try {
    # 注册需要清理的事件
    $eventRegistration = Register-CimIndicationEvent ...

    while ($true) {
        # 脚本核心逻辑
        Start-Sleep -Seconds 1

        # 监听Ctrl+C输入
        if ([Console]::KeyAvailable) {
            $key = [Console]::ReadKey($true)
            if ($key.Modifiers -eq 'Control' -and $key.Key -eq 'C') {
                Write-Host "捕获到Ctrl+C，执行清理..."
                break
            }
        }
    }
}
finally {
    # 确保清理逻辑执行
    if ($eventRegistration) {
        Unregister-Event -SubscriptionId $eventRegistration.Id
        Remove-Job -Id $eventRegistration.JobId -Force
    }
    [Console]::TreatControlCAsInput = $false
}

2. 绑定进程退出事件 #

利用 .NET 的 ProcessExit 事件，确保在进程终止（包括外部 kill 信号）时执行清理：

# 注册进程退出事件
$exitEvent = Register-ObjectEvent -InputObject ([System.AppDomain]::CurrentDomain) `
    -EventName ProcessExit `
    -Action {
        # 此处放置清理逻辑，比如注销事件、释放资源
        Write-Host "进程即将退出，执行全局清理..."
        Get-EventSubscriber | Where-Object SourceObject -eq [System.AppDomain]::CurrentDomain | Unregister-Event
    }

try {
    # 脚本核心任务，比如注册CIM事件
    $cimEvent = Register-CimIndicationEvent ...
    # 保持脚本运行
    Wait-Event -Timeout -1
}
finally {
    # 主动清理时先注销事件
    if ($cimEvent) { Unregister-Event -SubscriptionId $cimEvent.Id }
    if ($exitEvent) { Unregister-Event -SubscriptionId $exitEvent.Id }
}

这种方式能覆盖大部分外部终止场景，包括 kill 指令触发的进程退出。

3. PowerShell 7+ 中的信号处理增强 #

PowerShell 7 及以上版本支持通过 .NET 类直接监听终止信号，结合事件注册实现类 Unix trap 的效果：

using namespace System.Runtime.InteropServices

# 定义信号处理委托
$sigHandler = [ConsoleCancelEventHandler]{
    param($sender, $e)
    Write-Host "捕获到终止信号，执行清理..."
    # 执行清理逻辑
    Get-EventSubscriber | Unregister-Event
    Remove-Job * -Force
    $e.Cancel = $true # 阻止默认终止，确保清理完成
}

# 注册信号处理
[Console]::CancelKeyPress.Add($sigHandler)

try {
    # 脚本核心逻辑
    while ($true) { Start-Sleep 1 }
}
finally {
    [Console]::CancelKeyPress.Remove($sigHandler)
}

4. 封装清理逻辑为自定义函数 #

将清理逻辑封装为独立函数，结合脚本生命周期绑定，确保无论正常退出还是异常终止都能触发：

function Invoke-Cleanup {
    param($EventSubscriptions)
    foreach ($sub in $EventSubscriptions) {
        if (Get-EventSubscriber -SubscriptionId $sub.Id -ErrorAction SilentlyContinue) {
            Unregister-Event -SubscriptionId $sub.Id
        }
    }
    Write-Host "清理完成"
}

# 收集需要清理的事件订阅
$subscriptions = @()
try {
    $subscriptions += Register-CimIndicationEvent ...
    # 脚本运行逻辑
    Wait-Event
}
finally {
    Invoke-Cleanup -EventSubscriptions $subscriptions
}

# 额外绑定进程退出事件，覆盖外部终止场景
Register-ObjectEvent -InputObject ([System.AppDomain]::CurrentDomain) `
    -EventName ProcessExit `
    -Action { Invoke-Cleanup -EventSubscriptions $using:subscriptions }

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