У меня есть два приложения: С++-сервер и С# WPF UI. Код С++ принимает запросы (из любого места/любого) через службу обмена сообщениями ZeroMQ [PUB/SUB]. Я использую свой код С# для повторного тестирования и создания "обратных тестов" и их выполнения. Эти обратные тесты могут состоять из множества "единичных тестов" и каждой из них отправлять/получать тысячи сообщений с сервера С++.
В настоящее время индивидуальные задние тесты хорошо работают, могут отправлять 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);
}
}
Проигрыватель "backtesting" использует для выполнения каждого теста обратной связи, сначала создает Dictionary, содержащий каждый Test (unit test), и сообщения для отправки в приложение С++ для каждого теста.
Метод 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 в конце, чтобы сообщить С++, что мы закончили. Затем мы заставляем ждать, так что следующий [unit] тест не отправляется до того, как все возвраты будут получены из кода С++ - мы делаем это с помощью ManualResetEvent.
Когда С++ получает сообщение 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() в последнем выше никогда не попадает. Я понимаю, что окончательные блоки, выполняемые по фоновым потокам, не гарантируются для выполнения
Перечеркнутый текст выше был из-за того, что я возился с кодом; Я прекратил родительский поток до того, как ребенок закончил. Однако есть еще проблемы...
Теперь broker.Dispose() вызывается правильно и вызывается broker.Dispose(), в этом методе я пытаюсь отменить поток poller и правильно распоряжаться 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() в потоке, отличном от того, который использовался для опроса сообщений, а некоторые - как принудительного отмены. Thread abort не то, что я хочу получить любой ценой.
По сути, я хочу правильно распорядиться broker перед выполнением следующего теста обратной связи, как я могу правильно это обработать, отделить опрос и запустить его в отдельном домене приложений?
Я попробовал, располагая внутри обработчика OnMessageRecived, но это явно выполняется в том же потоке, что и poller, и это не способ сделать это, не вызывая дополнительных потоков, он блокирует.
Каков наилучший способ добиться того, что я хочу и , есть ли шаблон для такого рода случаев, за которым я могу следовать?
Спасибо за ваше время.