如何将后台工作线程设置为单线程单元？

Question

我正在创建一个自动化测试运行应用程序。在应用程序的这一部分中，我正在开发一个轮询服务器。它的工作原理是不断轮询 Web 服务器以确定何时应运行新的自动化测试（对于 GUI 应用程序的夜间自动运行）。

当轮询服务器看到请求时，它会下载所有必要的信息，然后在后台工作程序中执行测试运行。问题是测试运行的一部分包含在后台工作线程中发生的 OLE、COM 和其他调用（例如，Clipboard.Clear()）。当发生这些调用之一时，会发生以下异常：

在进行 OLE 调用之前，当前线程必须设置为单线程单元 (STA) 模式。确保您的 Main 函数上标记有 STAThreadAttribute。

如何将后台工作线程标记为单线程单元？我的 Program.cs 中的 Main 调用显然已经具有该属性。

Answer 1

这是不可能的，BGW使用的是threadpool线程。 TP 线程始终是 MTA，无法更改。您必须使用常规线程，在启动之前调用 SetApartmentState()。该线程还应该泵送消息循环，调用Application.Run()。

也许您应该考虑从 UI 线程调用此代码。因为无论如何，COM 服务器很可能都在 UI 线程上运行其方法。从工作线程到创建 COM 服务器的 STA 线程的封送调用是自动的，COM 负责处理它。

或者不畏艰险，亲自指挥。你可以创建自己的STA线程来给服务器一个幸福的家。您将在这篇文章中找到代码，请务必在 Initialize() 重写中创建 COM 对象。

Answer 2

默认情况下，BackgroundWorker 使用 ThreadPool 线程，但您可以覆盖此行为。首先，您需要定义一个自定义 SynchronizationContext：

public class MySynchronizationContext : SynchronizationContext
{
    public override void Post(SendOrPostCallback d, object state)
    {
        Thread t = new Thread(d.Invoke);
        t.SetApartmentState(ApartmentState.STA);
        t.Start(state);
    }
}

并覆盖默认的 SynchronizationContext，如下所示，然后再进行使用您的BackgroundWorker：

   AsyncOperationManager.SynchronizationContext = new MySynchronizationContext();

注意：这可能会对应用程序的其余部分产生性能影响，因此您可能希望限制新的Post实现（例如使用state或d参数）。

Answer 3

我还没有测试过它，但如果您调用 WinForms 表单，您应该返回到 UI 线程，并且大多数内容应该再次工作。

BackgroundWorker bgw = new BackgroundWorker();
bgw.DoWork += new DoWorkEventHandler(this.bgw_DoWork);
bgw.RunWorkerAsync();

private void bgw_DoWork(object sender, DoWorkEventArgs e)
{
    // Invoke the UI thread
    // "this" is referring to the Form1, or what ever your form is
    this.Invoke((MethodInvoker)delegate
    {
        Clipboard.GetText();
        // etc etc
    });
}

Answer 4

通常，您可以通过在入口点（例如 Static Main）上定义属性 [STAThread()] 来设置它。

Answer 5

我使用了 +Conrad de Wet 的想法，效果非常好！

不过，该代码有一个小问题，您必须像使用 }) 一样关闭“this.Invoke.....”；

以下是 Conrad de Wet 的修复代码：

    BackgroundWorker bgw = new BackgroundWorker();
    bgw.DoWork += new DoWorkEventHandler(this.bgw_DoWork);
    bgw.RunWorkerAsync();>

    private void bgw_DoWork(object sender, DoWorkEventArgs e)
    {
        // Invoke the UI thread
        // "this" is referring to the Form1, or what ever your form is
        this.Invoke((MethodInvoker)delegate
        {
            Clipboard.GetText();
            // etc etc
        });
    }