mirror of
https://github.com/dotnet/MQTTnet.git
synced 2026-10-18 10:26:11 +08:00
Use latest language features (#2176)
* Make use of language features line AsyncDispose, init etc. * Fix build errors * Improve Unit Tests * Improve Unit Tests * Improve Unit Tests * Update samples * Remove obsolete benchmarks * Code cleanup * Improve unit tests * Improve Unit Tests * Fix Unit Tests * Improve Unit Tests * Improve Unit Tests * Restore unit tests * Exclude load tests from unit tests
This commit is contained in:
Christian
@@ -1,342 +1,345 @@
|
||||
using MQTTnet.Server;
|
||||
using System.Collections.Generic;
|
||||
using System;
|
||||
using System.Diagnostics;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using System.Linq;
|
||||
using MQTTnet.Internal;
|
||||
|
||||
namespace MQTTnet.TestApp
|
||||
namespace MQTTnet.TestApp;
|
||||
|
||||
/// <summary>
|
||||
/// Connect a number of publisher clients and one subscriber client, then publish messages and measure
|
||||
/// the number of messages per second that can be exchanged between publishers and subscriber.
|
||||
/// Measurements are performed for subscriptions containing no wildcard, a single wildcard or multiple wildcards.
|
||||
/// </summary>
|
||||
public class MessageThroughputTest
|
||||
{
|
||||
/// <summary>
|
||||
/// Connect a number of publisher clients and one subscriber client, then publish messages and measure
|
||||
/// the number of messages per second that can be exchanged between publishers and subscriber.
|
||||
/// Measurements are performed for subscriptions containing no wildcard, a single wildcard or multiple wildcards.
|
||||
/// </summary>
|
||||
public class MessageThroughputTest
|
||||
// Change these constants to suit
|
||||
const int NumPublishers = 5000;
|
||||
const int NumTopicsPerPublisher = 10;
|
||||
|
||||
// Note: Other code changes may be required when changing this constant:
|
||||
const int NumSubscribers = 1; // Fixed
|
||||
|
||||
// Number of publish calls before a response for all published messages is awaited.
|
||||
// This must be limited to a reasonable value that the server or TCP pipeline can handle.
|
||||
const int NumPublishCallsPerBatch = 250;
|
||||
|
||||
// Message counters are set/reset in the PublishAllAsync loop
|
||||
int _messagesReceivedCount;
|
||||
int _messagesExpectedCount;
|
||||
|
||||
CancellationTokenSource _cancellationTokenSource;
|
||||
|
||||
MqttServer _mqttServer;
|
||||
Dictionary<string, IMqttClient> _mqttPublisherClientsByPublisherName;
|
||||
List<IMqttClient> _mqttSubscriberClients;
|
||||
|
||||
Dictionary<string, List<string>> _topicsByPublisher;
|
||||
Dictionary<string, List<string>> _singleWildcardTopicsByPublisher;
|
||||
Dictionary<string, List<string>> _multiWildcardTopicsByPublisher;
|
||||
|
||||
public async Task Run()
|
||||
{
|
||||
// Change these constants to suit
|
||||
const int NumPublishers = 5000;
|
||||
const int NumTopicsPerPublisher = 10;
|
||||
|
||||
// Note: Other code changes may be required when changing this constant:
|
||||
const int NumSubscribers = 1; // Fixed
|
||||
|
||||
// Number of publish calls before a response for all published messages is awaited.
|
||||
// This must be limited to a reasonable value that the server or TCP pipeline can handle.
|
||||
const int NumPublishCallsPerBatch = 250;
|
||||
|
||||
// Message counters are set/reset in the PublishAllAsync loop
|
||||
int _messagesReceivedCount;
|
||||
int _messagesExpectedCount;
|
||||
|
||||
CancellationTokenSource _cancellationTokenSource;
|
||||
|
||||
MqttServer _mqttServer;
|
||||
Dictionary<string, IMqttClient> _mqttPublisherClientsByPublisherName;
|
||||
List<IMqttClient> _mqttSubscriberClients;
|
||||
|
||||
Dictionary<string, List<string>> _topicsByPublisher;
|
||||
Dictionary<string, List<string>> _singleWildcardTopicsByPublisher;
|
||||
Dictionary<string, List<string>> _multiWildcardTopicsByPublisher;
|
||||
|
||||
public async Task Run()
|
||||
try
|
||||
{
|
||||
try
|
||||
{
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Begin message throughput test");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Begin message throughput test");
|
||||
Console.WriteLine();
|
||||
|
||||
Console.WriteLine("Number of publishers: " + NumPublishers);
|
||||
Console.WriteLine("Number of published topics (total): " + NumPublishers * NumTopicsPerPublisher);
|
||||
Console.WriteLine("Number of subscribers: " + NumSubscribers);
|
||||
Console.WriteLine("Number of publishers: " + NumPublishers);
|
||||
Console.WriteLine("Number of published topics (total): " + NumPublishers * NumTopicsPerPublisher);
|
||||
Console.WriteLine("Number of subscribers: " + NumSubscribers);
|
||||
|
||||
await Setup();
|
||||
await Setup();
|
||||
|
||||
await Subscribe_to_No_Wildcard_Topics();
|
||||
await Subscribe_to_Single_Wildcard_Topics();
|
||||
await Subscribe_to_Multi_Wildcard_Topics();
|
||||
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("End message throughput test");
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
ConsoleWriteLineError(ex.Message);
|
||||
}
|
||||
finally
|
||||
{
|
||||
await Cleanup();
|
||||
}
|
||||
}
|
||||
|
||||
public async Task Setup()
|
||||
{
|
||||
new TopicGenerator().Generate(NumPublishers, NumTopicsPerPublisher, out _topicsByPublisher, out _singleWildcardTopicsByPublisher, out _multiWildcardTopicsByPublisher);
|
||||
|
||||
var serverOptions = new MqttServerOptionsBuilder().WithDefaultEndpoint().Build();
|
||||
var mqttClientFactory = new MqttClientFactory();
|
||||
var mqttServerFactory = new MqttServerFactory();
|
||||
_mqttServer = mqttServerFactory.CreateMqttServer(serverOptions);
|
||||
await _mqttServer.StartAsync();
|
||||
await Subscribe_to_No_Wildcard_Topics();
|
||||
await Subscribe_to_Single_Wildcard_Topics();
|
||||
await Subscribe_to_Multi_Wildcard_Topics();
|
||||
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Begin connect " + NumPublishers + " publisher(s)...");
|
||||
Console.WriteLine("End message throughput test");
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
ConsoleWriteLineError(ex.Message);
|
||||
}
|
||||
finally
|
||||
{
|
||||
await Cleanup();
|
||||
}
|
||||
}
|
||||
|
||||
var stopWatch = new System.Diagnostics.Stopwatch();
|
||||
stopWatch.Start();
|
||||
_mqttPublisherClientsByPublisherName = new Dictionary<string, IMqttClient>();
|
||||
foreach (var pt in _topicsByPublisher)
|
||||
{
|
||||
var publisherName = pt.Key;
|
||||
var mqttClient = mqttClientFactory.CreateMqttClient();
|
||||
var publisherOptions = new MqttClientOptionsBuilder()
|
||||
.WithTcpServer("localhost")
|
||||
.WithClientId(publisherName)
|
||||
.Build();
|
||||
await mqttClient.ConnectAsync(publisherOptions);
|
||||
_mqttPublisherClientsByPublisherName.Add(publisherName, mqttClient);
|
||||
}
|
||||
stopWatch.Stop();
|
||||
public async Task Setup()
|
||||
{
|
||||
new TopicGenerator().Generate(NumPublishers, NumTopicsPerPublisher, out _topicsByPublisher, out _singleWildcardTopicsByPublisher, out _multiWildcardTopicsByPublisher);
|
||||
|
||||
Console.Write($"{NumPublishers} publisher(s) connected in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
Console.WriteLine($"connections per second: {NumPublishers / (stopWatch.ElapsedMilliseconds / 1000.0):0.000}");
|
||||
var serverOptions = new MqttServerOptionsBuilder().WithDefaultEndpoint().Build();
|
||||
var mqttClientFactory = new MqttClientFactory();
|
||||
var mqttServerFactory = new MqttServerFactory();
|
||||
_mqttServer = mqttServerFactory.CreateMqttServer(serverOptions);
|
||||
await _mqttServer.StartAsync();
|
||||
|
||||
_mqttSubscriberClients = new List<IMqttClient>();
|
||||
for (var i = 0; i < NumSubscribers; ++i)
|
||||
{
|
||||
var mqttClient = mqttClientFactory.CreateMqttClient();
|
||||
var subsriberOptions = new MqttClientOptionsBuilder()
|
||||
.WithTcpServer("localhost")
|
||||
.WithClientId("sub" + i)
|
||||
.Build();
|
||||
await mqttClient.ConnectAsync(subsriberOptions);
|
||||
mqttClient.ApplicationMessageReceivedAsync += HandleApplicationMessageReceivedAsync;
|
||||
_mqttSubscriberClients.Add(mqttClient);
|
||||
}
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Begin connect " + NumPublishers + " publisher(s)...");
|
||||
|
||||
var stopWatch = new Stopwatch();
|
||||
stopWatch.Start();
|
||||
_mqttPublisherClientsByPublisherName = [];
|
||||
foreach (var pt in _topicsByPublisher)
|
||||
{
|
||||
var publisherName = pt.Key;
|
||||
var mqttClient = mqttClientFactory.CreateMqttClient();
|
||||
var publisherOptions = new MqttClientOptionsBuilder()
|
||||
.WithTcpServer("localhost")
|
||||
.WithClientId(publisherName)
|
||||
.Build();
|
||||
await mqttClient.ConnectAsync(publisherOptions);
|
||||
_mqttPublisherClientsByPublisherName.Add(publisherName, mqttClient);
|
||||
}
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{NumPublishers} publisher(s) connected in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
Console.WriteLine($"connections per second: {NumPublishers / (stopWatch.ElapsedMilliseconds / 1000.0):0.000}");
|
||||
|
||||
_mqttSubscriberClients = new List<IMqttClient>();
|
||||
for (var i = 0; i < NumSubscribers; ++i)
|
||||
{
|
||||
var mqttClient = mqttClientFactory.CreateMqttClient();
|
||||
var subsriberOptions = new MqttClientOptionsBuilder()
|
||||
.WithTcpServer("localhost")
|
||||
.WithClientId("sub" + i)
|
||||
.Build();
|
||||
await mqttClient.ConnectAsync(subsriberOptions);
|
||||
mqttClient.ApplicationMessageReceivedAsync += HandleApplicationMessageReceivedAsync;
|
||||
_mqttSubscriberClients.Add(mqttClient);
|
||||
}
|
||||
}
|
||||
|
||||
Task HandleApplicationMessageReceivedAsync(MqttApplicationMessageReceivedEventArgs eventArgs)
|
||||
{
|
||||
// count messages and signal cancellation when expected message count is reached
|
||||
++_messagesReceivedCount;
|
||||
if (_messagesReceivedCount == _messagesExpectedCount)
|
||||
{
|
||||
_cancellationTokenSource.Cancel();
|
||||
}
|
||||
return CompletedTask.Instance;
|
||||
}
|
||||
|
||||
public async Task Cleanup()
|
||||
{
|
||||
foreach (var mqttClient in _mqttSubscriberClients)
|
||||
{
|
||||
await mqttClient.DisconnectAsync();
|
||||
mqttClient.ApplicationMessageReceivedAsync -= HandleApplicationMessageReceivedAsync;
|
||||
mqttClient.Dispose();
|
||||
}
|
||||
|
||||
Task HandleApplicationMessageReceivedAsync(MqttApplicationMessageReceivedEventArgs eventArgs)
|
||||
foreach (var pub in _mqttPublisherClientsByPublisherName)
|
||||
{
|
||||
// count messages and signal cancellation when expected message count is reached
|
||||
++_messagesReceivedCount;
|
||||
if (_messagesReceivedCount == _messagesExpectedCount)
|
||||
{
|
||||
_cancellationTokenSource.Cancel();
|
||||
}
|
||||
return CompletedTask.Instance;
|
||||
var mqttClient = pub.Value;
|
||||
await mqttClient.DisconnectAsync();
|
||||
mqttClient.Dispose();
|
||||
}
|
||||
|
||||
public async Task Cleanup()
|
||||
await _mqttServer.StopAsync();
|
||||
_mqttServer.Dispose();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Measure no-wildcard topic subscription message exchange performance
|
||||
/// </summary>
|
||||
public Task Subscribe_to_No_Wildcard_Topics()
|
||||
{
|
||||
return ProcessMessages(_topicsByPublisher, "no wildcards");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Measure single-wildcard topic subscription message exchange performance
|
||||
/// </summary>
|
||||
public Task Subscribe_to_Single_Wildcard_Topics()
|
||||
{
|
||||
return ProcessMessages(_singleWildcardTopicsByPublisher, "single wildcard");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Measure multi-wildcard topic subscription message exchange performance
|
||||
/// </summary>
|
||||
public Task Subscribe_to_Multi_Wildcard_Topics()
|
||||
{
|
||||
return ProcessMessages(_multiWildcardTopicsByPublisher, "multi wildcard");
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Subscribe to all topics, then run message exchange
|
||||
/// </summary>
|
||||
public async Task ProcessMessages(Dictionary<string, List<string>> topicsByPublisher, string topicTypeDescription)
|
||||
{
|
||||
var numTopics = CountTopics(topicsByPublisher);
|
||||
|
||||
Console.WriteLine();
|
||||
Console.Write($"Subscribing to {numTopics} topics ");
|
||||
ConsoleWriteInfo($"({topicTypeDescription})");
|
||||
Console.WriteLine($" for {NumSubscribers} subscriber(s)...");
|
||||
|
||||
var stopWatch = new Stopwatch();
|
||||
stopWatch.Start();
|
||||
|
||||
// each subscriber subscribes to all topics, one call per topic
|
||||
foreach (var subscriber in _mqttSubscriberClients)
|
||||
{
|
||||
foreach (var mqttClient in _mqttSubscriberClients)
|
||||
{
|
||||
await mqttClient.DisconnectAsync();
|
||||
mqttClient.ApplicationMessageReceivedAsync -= HandleApplicationMessageReceivedAsync;
|
||||
mqttClient.Dispose();
|
||||
}
|
||||
|
||||
foreach (var pub in _mqttPublisherClientsByPublisherName)
|
||||
{
|
||||
var mqttClient = pub.Value;
|
||||
await mqttClient.DisconnectAsync();
|
||||
mqttClient.Dispose();
|
||||
}
|
||||
|
||||
await _mqttServer.StopAsync();
|
||||
_mqttServer.Dispose();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Measure no-wildcard topic subscription message exchange performance
|
||||
/// </summary>
|
||||
public Task Subscribe_to_No_Wildcard_Topics()
|
||||
{
|
||||
return ProcessMessages(_topicsByPublisher, "no wildcards");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Measure single-wildcard topic subscription message exchange performance
|
||||
/// </summary>
|
||||
public Task Subscribe_to_Single_Wildcard_Topics()
|
||||
{
|
||||
return ProcessMessages(_singleWildcardTopicsByPublisher, "single wildcard");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Measure multi-wildcard topic subscription message exchange performance
|
||||
/// </summary>
|
||||
public Task Subscribe_to_Multi_Wildcard_Topics()
|
||||
{
|
||||
return ProcessMessages(_multiWildcardTopicsByPublisher, "multi wildcard");
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Subscribe to all topics, then run message exchange
|
||||
/// </summary>
|
||||
public async Task ProcessMessages(Dictionary<string, List<string>> topicsByPublisher, string topicTypeDescription)
|
||||
{
|
||||
var numTopics = CountTopics(topicsByPublisher);
|
||||
|
||||
Console.WriteLine();
|
||||
Console.Write($"Subscribing to {numTopics} topics ");
|
||||
ConsoleWriteInfo($"({topicTypeDescription})");
|
||||
Console.WriteLine($" for {NumSubscribers} subscriber(s)...");
|
||||
|
||||
var stopWatch = new System.Diagnostics.Stopwatch();
|
||||
stopWatch.Start();
|
||||
|
||||
// each subscriber subscribes to all topics, one call per topic
|
||||
foreach (var subscriber in _mqttSubscriberClients)
|
||||
{
|
||||
foreach (var tp in topicsByPublisher)
|
||||
{
|
||||
var topics = tp.Value;
|
||||
foreach (var topic in topics)
|
||||
{
|
||||
var topicFilter = new Packets.MqttTopicFilter() { Topic = topic };
|
||||
await subscriber.SubscribeAsync(topicFilter).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{NumSubscribers} subscriber(s) subscribed in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
Console.WriteLine($"subscribe calls per second: {numTopics / (stopWatch.ElapsedMilliseconds / 1000.0):0.000}");
|
||||
|
||||
await PublishAllAsync();
|
||||
|
||||
Console.WriteLine($"Unsubscribing {numTopics} topics ({topicTypeDescription}) for {NumSubscribers} subscriber(s)...");
|
||||
|
||||
stopWatch.Restart();
|
||||
|
||||
// unsubscribe to all topics, one call per topic
|
||||
foreach (var subscriber in _mqttSubscriberClients)
|
||||
{
|
||||
foreach (var tp in topicsByPublisher)
|
||||
{
|
||||
var topics = tp.Value;
|
||||
foreach (var topic in topics)
|
||||
{
|
||||
var topicFilter = new Packets.MqttTopicFilter() { Topic = topic };
|
||||
|
||||
MqttClientUnsubscribeOptions options =
|
||||
new MqttClientUnsubscribeOptionsBuilder()
|
||||
.WithTopicFilter(topicFilter)
|
||||
.Build();
|
||||
await subscriber.UnsubscribeAsync(options).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{NumSubscribers} subscriber(s) unsubscribed in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
Console.WriteLine($"unsubscribe calls per second: {numTopics / (stopWatch.ElapsedMilliseconds / 1000.0):0.000}");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Publish messages in batches of NumPublishCallsPerBatch, wait for messages sent to subscriber
|
||||
/// </summary>
|
||||
async Task PublishAllAsync()
|
||||
{
|
||||
Console.WriteLine("Begin message exchange...");
|
||||
|
||||
int publisherIndexCounter = 0; // index to loop around all publishers to publish
|
||||
int topicIndexCounter = 0; // index to loop around all topics to publish
|
||||
int totalNumMessagesReceived = 0;
|
||||
|
||||
var publisherNames = _topicsByPublisher.Keys.ToList();
|
||||
|
||||
// There should be one message received per publish
|
||||
_messagesExpectedCount = NumPublishCallsPerBatch;
|
||||
|
||||
var stopWatch = new System.Diagnostics.Stopwatch();
|
||||
stopWatch.Start();
|
||||
|
||||
// Loop for a while and exchange messages
|
||||
|
||||
while (stopWatch.ElapsedMilliseconds < 10000)
|
||||
{
|
||||
_messagesReceivedCount = 0;
|
||||
|
||||
_cancellationTokenSource = new CancellationTokenSource();
|
||||
|
||||
for (var publishCallCount = 0; publishCallCount < NumPublishCallsPerBatch; ++publishCallCount)
|
||||
{
|
||||
// pick a publisher
|
||||
var publisherName = publisherNames[publisherIndexCounter % publisherNames.Count];
|
||||
var publisherTopics = _topicsByPublisher[publisherName];
|
||||
// pick a publisher topic
|
||||
var topic = publisherTopics[topicIndexCounter % publisherTopics.Count];
|
||||
var message = new MqttApplicationMessageBuilder()
|
||||
.WithTopic(topic)
|
||||
.Build();
|
||||
await _mqttPublisherClientsByPublisherName[publisherName].PublishAsync(message).ConfigureAwait(false);
|
||||
++topicIndexCounter;
|
||||
++publisherIndexCounter;
|
||||
}
|
||||
|
||||
// Wait for at least one message per publish to be received by subscriber (in the subscriber's application message handler),
|
||||
// then loop around to send another batch
|
||||
try
|
||||
{
|
||||
await Task.Delay(30000, _cancellationTokenSource.Token).ConfigureAwait(false);
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
|
||||
_cancellationTokenSource.Dispose();
|
||||
|
||||
if (_messagesReceivedCount < _messagesExpectedCount)
|
||||
{
|
||||
ConsoleWriteLineError(string.Format("Messages Received Count mismatch, expected {0}, received {1}", _messagesExpectedCount, _messagesReceivedCount));
|
||||
return;
|
||||
}
|
||||
|
||||
totalNumMessagesReceived += _messagesReceivedCount;
|
||||
}
|
||||
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{totalNumMessagesReceived} messages published and received in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
ConsoleWriteLineSuccess($"messages per second: {(int)(totalNumMessagesReceived / (stopWatch.ElapsedMilliseconds / 1000.0))}");
|
||||
}
|
||||
|
||||
int CountTopics(Dictionary<string, List<string>> topicsByPublisher)
|
||||
{
|
||||
var count = 0;
|
||||
foreach (var tp in topicsByPublisher)
|
||||
{
|
||||
count += tp.Value.Count;
|
||||
var topics = tp.Value;
|
||||
foreach (var topic in topics)
|
||||
{
|
||||
var topicFilter = new Packets.MqttTopicFilter() { Topic = topic };
|
||||
await subscriber.SubscribeAsync(topicFilter).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
return count;
|
||||
}
|
||||
|
||||
void ConsoleWriteLineError(string message)
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{NumSubscribers} subscriber(s) subscribed in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
Console.WriteLine($"subscribe calls per second: {numTopics / (stopWatch.ElapsedMilliseconds / 1000.0):0.000}");
|
||||
|
||||
await PublishAllAsync();
|
||||
|
||||
Console.WriteLine($"Unsubscribing {numTopics} topics ({topicTypeDescription}) for {NumSubscribers} subscriber(s)...");
|
||||
|
||||
stopWatch.Restart();
|
||||
|
||||
// unsubscribe to all topics, one call per topic
|
||||
foreach (var subscriber in _mqttSubscriberClients)
|
||||
{
|
||||
Console.ForegroundColor = ConsoleColor.Red;
|
||||
Console.WriteLine(message);
|
||||
Console.ResetColor();
|
||||
foreach (var tp in topicsByPublisher)
|
||||
{
|
||||
var topics = tp.Value;
|
||||
foreach (var topic in topics)
|
||||
{
|
||||
var topicFilter = new Packets.MqttTopicFilter { Topic = topic };
|
||||
|
||||
var options =
|
||||
new MqttClientUnsubscribeOptionsBuilder()
|
||||
.WithTopicFilter(topicFilter)
|
||||
.Build();
|
||||
await subscriber.UnsubscribeAsync(options).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void ConsoleWriteLineSuccess(string message)
|
||||
{
|
||||
Console.ForegroundColor = ConsoleColor.Green;
|
||||
Console.WriteLine(message);
|
||||
Console.ResetColor();
|
||||
}
|
||||
|
||||
void ConsoleWriteInfo(string message)
|
||||
{
|
||||
Console.ForegroundColor = ConsoleColor.White;
|
||||
Console.Write(message);
|
||||
Console.ResetColor();
|
||||
}
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{NumSubscribers} subscriber(s) unsubscribed in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
Console.WriteLine($"unsubscribe calls per second: {numTopics / (stopWatch.ElapsedMilliseconds / 1000.0):0.000}");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Publish messages in batches of NumPublishCallsPerBatch, wait for messages sent to subscriber
|
||||
/// </summary>
|
||||
async Task PublishAllAsync()
|
||||
{
|
||||
Console.WriteLine("Begin message exchange...");
|
||||
|
||||
var publisherIndexCounter = 0; // index to loop around all publishers to publish
|
||||
var topicIndexCounter = 0; // index to loop around all topics to publish
|
||||
var totalNumMessagesReceived = 0;
|
||||
|
||||
var publisherNames = _topicsByPublisher.Keys.ToList();
|
||||
|
||||
// There should be one message received per publish
|
||||
_messagesExpectedCount = NumPublishCallsPerBatch;
|
||||
|
||||
var stopWatch = new Stopwatch();
|
||||
stopWatch.Start();
|
||||
|
||||
// Loop for a while and exchange messages
|
||||
|
||||
while (stopWatch.ElapsedMilliseconds < 10000)
|
||||
{
|
||||
_messagesReceivedCount = 0;
|
||||
|
||||
_cancellationTokenSource = new CancellationTokenSource();
|
||||
|
||||
for (var publishCallCount = 0; publishCallCount < NumPublishCallsPerBatch; ++publishCallCount)
|
||||
{
|
||||
// pick a publisher
|
||||
var publisherName = publisherNames[publisherIndexCounter % publisherNames.Count];
|
||||
var publisherTopics = _topicsByPublisher[publisherName];
|
||||
// pick a publisher topic
|
||||
var topic = publisherTopics[topicIndexCounter % publisherTopics.Count];
|
||||
var message = new MqttApplicationMessageBuilder()
|
||||
.WithTopic(topic)
|
||||
.Build();
|
||||
|
||||
await _mqttPublisherClientsByPublisherName[publisherName].PublishAsync(message).ConfigureAwait(false);
|
||||
++topicIndexCounter;
|
||||
++publisherIndexCounter;
|
||||
}
|
||||
|
||||
// Wait for at least one message per publish to be received by subscriber (in the subscriber's application message handler),
|
||||
// then loop around to send another batch
|
||||
try
|
||||
{
|
||||
await Task.Delay(30000, _cancellationTokenSource.Token).ConfigureAwait(false);
|
||||
}
|
||||
catch (Exception exception)
|
||||
{
|
||||
Debug.Write(exception.ToString());
|
||||
}
|
||||
|
||||
_cancellationTokenSource.Dispose();
|
||||
|
||||
if (_messagesReceivedCount < _messagesExpectedCount)
|
||||
{
|
||||
ConsoleWriteLineError($"Messages Received Count mismatch, expected {_messagesExpectedCount}, received {_messagesReceivedCount}");
|
||||
return;
|
||||
}
|
||||
|
||||
totalNumMessagesReceived += _messagesReceivedCount;
|
||||
}
|
||||
|
||||
stopWatch.Stop();
|
||||
|
||||
Console.Write($"{totalNumMessagesReceived} messages published and received in {stopWatch.ElapsedMilliseconds / 1000.0:0.000} seconds, ");
|
||||
ConsoleWriteLineSuccess($"messages per second: {(int)(totalNumMessagesReceived / (stopWatch.ElapsedMilliseconds / 1000.0))}");
|
||||
}
|
||||
|
||||
static int CountTopics(Dictionary<string, List<string>> topicsByPublisher)
|
||||
{
|
||||
var count = 0;
|
||||
foreach (var tp in topicsByPublisher)
|
||||
{
|
||||
count += tp.Value.Count;
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
static void ConsoleWriteLineError(string message)
|
||||
{
|
||||
Console.ForegroundColor = ConsoleColor.Red;
|
||||
Console.WriteLine(message);
|
||||
Console.ResetColor();
|
||||
}
|
||||
|
||||
static void ConsoleWriteLineSuccess(string message)
|
||||
{
|
||||
Console.ForegroundColor = ConsoleColor.Green;
|
||||
Console.WriteLine(message);
|
||||
Console.ResetColor();
|
||||
}
|
||||
|
||||
static void ConsoleWriteInfo(string message)
|
||||
{
|
||||
Console.ForegroundColor = ConsoleColor.White;
|
||||
Console.Write(message);
|
||||
Console.ResetColor();
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user