如何使用 StaTaskScheduler 和 BlockingCollection 解决 STA 线程中的死锁？

StaTaskScheduler 和 STA 线程消息泵送

问题：

由于缺乏消息泵送，在 STA 线程上使用 StaTaskScheduler 和 BlockingCollection 时会发生死锁。

解决方案：

使用 MsgWaitForMultipleObjectsEx 代替 CoWaitForMultipleHandles 实现具有消息泵送功能的自定义同步上下文。

分步说明：

1. 了解问题：STA TaskScheduler 不会自动在 STA 线程上泵送消息。
2. 确定根本原因：BlockingCollection 执行的阻塞操作不会泵送消息。
3. 创建自定义同步上下文：重写 SynchronizationContext.Wait 以使用 MsgWaitForMultipleObjectsEx 实现消息泵送并查看/分派消息。
4. 安装自定义同步上下文：在 StaTaskScheduler 创建的每个 STA 线程上安装自定义同步上下文。
5. 使用自定义同步上下文：结合使用 StaTaskScheduler 和自定义同步上下文调度任务，以确保消息泵送。

实现说明：

• 使用带有 MWMO_INPUTAVAILABLE 标志的 MsgWaitForMultipleObjectsEx 检测传入的消息。
• 通过在自定义同步上下文的 Wait 方法中调用 PeekMessage(PM_REMOVE) 和 DispatchMessage 来手动泵送消息。
• 优雅地处理超时和潜在的竞争条件。

优势：

• 通过确保在 STA 线程上处理消息来解决死锁问题。
• 为旧版 COM 对象维护正确的 STA 执行环境。
• 使用后续的 await 延续时保留线程关联性。

示例代码：

// 带有消息泵送的自定义同步上下文
class CustomSynchronizationContext : SynchronizationContext
{
    protected override void OnWait(bool unused)
    {
        const uint QS_MASK = 0x0001;
        const uint MWMO_INPUTAVAILABLE = 0x0004;

        var nativeResult = MsgWaitForMultipleObjectsEx(
            0, null, WAIT_INFINITE, QS_MASK, MWMO_INPUTAVAILABLE);

        var msg = new MSG();
        while (true)
        {
            if (nativeResult == WAIT_OBJECT_0)
                break;

            if (PeekMessage(out msg, IntPtr.Zero, 0, 0, PM_REMOVE))
            {
                TranslateMessage(ref msg);
                DispatchMessage(ref msg);
            }

            nativeResult = MsgWaitForMultipleObjectsEx(0, null, 0, QS_MASK, MWMO_INPUTAVAILABLE);
        }
    }
}

// 使用方法
using (var staThread = new ThreadWithAffinityContext(true, true))
{
    staThread.Run(async () =>
    {
        // 在 STA 线程上安装自定义同步上下文
        SynchronizationContext.SetSynchronizationContext(new CustomSynchronizationContext());

        // 使用消息泵送的 STA 线程代码...
    });
}

以上是如何使用 StaTaskScheduler 和 BlockingCollection 解决 STA 线程中的死锁？的详细内容。更多信息请关注PHP中文网其他相关文章！