如何在C#中实现线程关联?

时间:2017-07-11 02:42:39

标签: c# multithreading wcf synchronizationcontext

我有第三方API需要线程关联。我在我的服务应用程序中使用WCF处理来自客户端的请求,然后委托给此API。由于WCF使用线程池来处理请求,我尝试使用以下代码解决此问题(使用SynchronizationContext类):

using System;
using System.Collections.Generic;
using System.ServiceModel;
using System.Threading;

namespace MyAPIService
{
  [ServiceBehavior(InstanceContextMode = InstanceContextMode.PerSession)]
  public class MyService: IService
  {
      ThirdPartyAPI m_api;
      SynchronizationContext m_context;

      MyService()
      {
        m_api = new ThirdPartyAPI();
      }

      public bool Connect(string ipaddress, int port)
      {
        m_context = (SynchronizationContext.Current == null) ? new SynchronizationContext() : SynchronizationContext.Current;
        return m_api.Connect(ipaddress, port);
      }

      public bool Disconnect()
      {
        throw new NotImplementedException();
      }

      public bool IsConnected()
      {
        return Send(() => m_api.IsConnected());
      }

      public TResult Send<TResult>(Func<TResult> func)
      {
        TResult retval = default(TResult);
        m_context.Send(new SendOrPostCallback((x) =>
        {
          retval = func();
        })
        , null);
        return retval;
      }

  }
 }
}

我认为这将允许我在调用Connect的同一个完全线程上执行IsConnected方法,但是从测试开始并非如此。 IsConnected仍在池中的任何线程上执行。我做错了什么?

非常感谢任何帮助。 非常感谢你。

1 个答案:

答案 0 :(得分:2)

默认同步上下文在线程池上执行您的工作(因此,您没有线程关联)。

如果要确保工作始终发布到同一个线程,则需要编写自己的同步上下文。例如:

public class SingleThreadSynchronizationContext : SynchronizationContext
{
    private readonly BlockingCollection<(SendOrPostCallback callback, object state)> _queue;
    private readonly Thread _processingThread;

    public SingleThreadSynchronizationContext()
    {
        _queue = new BlockingCollection<(SendOrPostCallback, object)>();
        _processingThread = new Thread(Process) { IsBackground = true };
        _processingThread.Start();
    }

    public override void Send(SendOrPostCallback d, object state)
    {
        using (var mutex = new ManualResetEventSlim())
        {
            var callback = new SendOrPostCallback(s =>
            {
                d(s);
                mutex.Set();
            });

            _queue.Add((callback, state));
            mutex.Wait();
        }
    }

    public override void Post(SendOrPostCallback d, object state)
    {
        _queue.Add((d, state));
    }

    public override SynchronizationContext CreateCopy()
    {
        return this;
    }

    private void Process()
    {
        SetSynchronizationContext(this);

        foreach (var item in _queue.GetConsumingEnumerable())
        {
            item.callback(item.state);
        }
    }
}

请注意,此同步上下文假定回调中未捕获的异常会使进程崩溃。如果情况并非如此(例如,因为您有一个全局异常处理程序),那么您应该添加一些错误处理(在ProcessSend中)。