我有两个应用程序,一个 C++ 服务器和一个 C# WPF UI。C++ 代码通过 ZeroMQ 消息传递 [PUB/SUB] 服务接受请求(来自任何地方/任何人)。我使用我的 C# 代码进行回溯测试并创建“回溯测试”并执行它们。这些回溯测试可以由许多“单元测试”组成,每个测试都从 C++ 服务器发送/接收数千条消息。
目前,单个回溯测试运行良好,可以发送 N 个单元测试,每个测试包含数千个请求和捕获。我的问题是建筑;当我发送另一个回测(在第一个测试之后)时,由于轮询线程没有被取消和处置,我遇到了第二次完成事件订阅的问题。这会导致错误的输出。这似乎是一个微不足道的问题(也许对你们中的一些人来说),但是在我当前的配置下取消这个轮询任务是很麻烦的。一些代码...
我的消息代理类很简单,看起来像
public class MessageBroker : IMessageBroker<Taurus.FeedMux>, IDisposable
{
private Task pollingTask;
private NetMQContext context;
private PublisherSocket pubSocket;
private CancellationTokenSource source;
private CancellationToken token;
private ManualResetEvent pollerCancelled;
public MessageBroker()
{
this.source = new CancellationTokenSource();
this.token = source.Token;
StartPolling();
context = NetMQContext.Create();
pubSocket = context.CreatePublisherSocket();
pubSocket.Connect(PublisherAddress);
}
public void Dispatch(Taurus.FeedMux message)
{
pubSocket.Send(message.ToByteArray<Taurus.FeedMux>());
}
private void StartPolling()
{
pollerCancelled = new ManualResetEvent(false);
pollingTask = Task.Run(() =>
{
try
{
using (var context = NetMQContext.Create())
using (var subSocket = context.CreateSubscriberSocket())
{
byte[] buffer = null;
subSocket.Options.ReceiveHighWatermark = 1000;
subSocket.Connect(SubscriberAddress);
subSocket.Subscribe(String.Empty);
while (true)
{
buffer = subSocket.Receive();
MessageRecieved.Report(buffer.ToObject<Taurus.FeedMux>());
if (this.token.IsCancellationRequested)
this.token.ThrowIfCancellationRequested();
}
}
}
catch (OperationCanceledException)
{
pollerCancelled.Set();
}
}, this.token);
}
private void CancelPolling()
{
source.Cancel();
pollerCancelled.WaitOne();
pollerCancelled.Close();
}
public IProgress<Taurus.FeedMux> MessageRecieved { get; set; }
public string PublisherAddress { get { return "tcp://127.X.X.X:6500"; } }
public string SubscriberAddress { get { return "tcp://127.X.X.X:6501"; } }
private bool disposed = false;
protected virtual void Dispose(bool disposing)
{
if (!disposed)
{
if (disposing)
{
if (this.pollingTask != null)
{
CancelPolling();
if (this.pollingTask.Status == TaskStatus.RanToCompletion ||
this.pollingTask.Status == TaskStatus.Faulted ||
this.pollingTask.Status == TaskStatus.Canceled)
{
this.pollingTask.Dispose();
this.pollingTask = null;
}
}
if (this.context != null)
{
this.context.Dispose();
this.context = null;
}
if (this.pubSocket != null)
{
this.pubSocket.Dispose();
this.pubSocket = null;
}
if (this.source != null)
{
this.source.Dispose();
this.source = null;
}
}
disposed = true;
}
}
public void Dispose()
{
Dispose(true);
GC.SuppressFinalize(this);
}
~MessageBroker()
{
Dispose(false);
}
}
回测“引擎”用于执行每个回测,首先构造一个Dictionary
包含每个Test
(单元测试)和消息以分派给每个测试的 C++ 应用程序。
DispatchTests
方法,来了
private void DispatchTests(ConcurrentDictionary<Test, List<Taurus.FeedMux>> feedMuxCollection)
{
broker = new MessageBroker();
broker.MessageRecieved = new Progress<Taurus.FeedMux>(OnMessageRecieved);
testCompleted = new ManualResetEvent(false);
try
{
// Loop through the tests.
foreach (var kvp in feedMuxCollection)
{
testCompleted.Reset();
Test t = kvp.Key;
t.Bets = new List<Taurus.Bet>();
foreach (Taurus.FeedMux mux in kvp.Value)
{
token.ThrowIfCancellationRequested();
broker.Dispatch(mux);
}
broker.Dispatch(new Taurus.FeedMux()
{
type = Taurus.FeedMux.Type.PING,
ping = new Taurus.Ping() { event_id = t.EventID }
});
testCompleted.WaitOne(); // Wait until all messages are received for this test.
}
testCompleted.Close();
}
finally
{
broker.Dispose(); // Dispose the broker.
}
}
最后的PING
消息,它告诉 C++ 我们已经完成了。然后我们强制等待,以便在从 C++ 代码接收到所有返回之前不会调度下一个 [unit] 测试 - 我们使用ManualResetEvent
.
当 C++ 收到 PING 消息时,它会直接返回消息。我们通过处理接收到的消息OnMessageRecieved
,PING 告诉我们设置,ManualResetEvent.Set()
以便我们可以继续单元测试;“下一位”...
private async void OnMessageRecieved(Taurus.FeedMux mux)
{
string errorMsg = String.Empty;
if (mux.type == Taurus.FeedMux.Type.MSG)
{
// Do stuff.
}
else if (mux.type == Taurus.FeedMux.Type.PING)
{
// Do stuff.
// We are finished reciving messages for this "unit test"
testCompleted.Set();
}
}
我的问题是,。broker.Dispose()
在 finally 上面永远不会被击中。我很欣赏在后台线程上执行的 finally 块不能保证被执行
上面划掉的文字是因为我弄乱了代码;在孩子完成之前,我正在停止父线程。但是,仍然存在问题...
现在broker.Dispose()
被正确调用,并且broker.Dispose()
被调用,在这个方法中我尝试取消轮询线程并Task
正确处理以避免任何多个订阅。
要取消线程,我使用该CancelPolling()
方法
private void CancelPolling()
{
source.Cancel();
pollerCancelled.WaitOne(); <- Blocks here waiting for cancellation.
pollerCancelled.Close();
}
但在StartPolling()
方法中
while (true)
{
buffer = subSocket.Receive();
MessageRecieved.Report(buffer.ToObject<Taurus.FeedMux>());
if (this.token.IsCancellationRequested)
this.token.ThrowIfCancellationRequested();
}
ThrowIfCancellationRequested()
永远不会被调用并且线程永远不会被取消,因此永远不会被正确处理。轮询线程被该subSocket.Receive()
方法阻塞。
现在,我不清楚如何实现我想要的,我需要在用于轮询消息之外的线程上调用broker.Dispose()
/以及如何强制取消。PollerCancel()
线程中止不是我想不惜一切代价进入的。
本质上,我想broker
在执行下一个回溯测试之前正确处理,如何正确处理这个问题,拆分轮询并在单独的应用程序域中运行它?
我已经尝试过,在处理程序中进行OnMessageRecived
处理,但这显然是在与轮询器相同的线程上执行的,并且不是这样做的方法,没有调用其他线程,它会阻塞。
实现我想要的最好方法是什么,我可以遵循这种情况的模式吗?
谢谢你的时间。