add perf check example

Author: mookid8000

Rebus.ServiceProvider.Tests/Examples/InMemoryPerformanceCheck.cs

@@ -0,0 +1,115 @@
+using System;
+using System.Diagnostics;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.DependencyInjection;
+using NUnit.Framework;
+using Rebus.Bus;
+using Rebus.Config;
+using Rebus.Handlers;
+using Rebus.Logging;
+using Rebus.Routing.TypeBased;
+using Rebus.Tests.Contracts;
+using Rebus.Transport.InMem;
+
+namespace Rebus.ServiceProvider.Tests.Examples;
+
+[TestFixture]
+public class InMemoryPerformanceCheck : FixtureBase
+{
+    Microsoft.Extensions.DependencyInjection.ServiceProvider provider;
+
+    protected override void SetUp()
+    {
+        base.SetUp();
+
+        var services = new ServiceCollection();
+
+        services.AddRebus(configure =>
+            configure.Transport(t => t.UseInMemoryTransport(new(), "test-queue"))
+                .Routing(r => r.TypeBased().Map<PerformanceData>("test-queue")) // Map all messages in the assembly of Program to the queue
+                .Logging(l => l.ColoredConsole(LogLevel.Warn))
+        );
+
+        services.AddRebusHandler<PerformanceConsumer>();
+
+        services.AddSingleton<PerformanceProducer>();
+
+        provider = services.BuildServiceProvider();
+        provider.StartHostedServices();
+    }
+
+    protected override void TearDown()
+    {
+        provider.Dispose();
+        base.TearDown();
+    }
+
+    [Test]
+    [Repeat(10)]
+    public async Task RunTheTest()
+    {
+        var producer = provider.GetRequiredService<PerformanceProducer>();
+
+        var elapsedMilliseconds = await producer.TestPerformance();
+
+        Console.WriteLine($"Perf test execution time: {elapsedMilliseconds} ms");
+    }
+
+    public static class PerformanceCounter
+    {
+        public static int Counter = 10000;
+    }
+
+    public class PerformanceProducer
+    {
+        public static AutoResetEvent AutoResetEvent { get; } = new AutoResetEvent(false);
+
+        private readonly IBus _bus;
+
+        public PerformanceProducer(IBus bus)
+        {
+            _bus = bus;
+        }
+
+        public async Task<long> TestPerformance()
+        {
+            Stopwatch stopwatch = new Stopwatch();
+            stopwatch.Start();
+
+            for (int i = 0; i < PerformanceCounter.Counter; i++)
+            {
+                await _bus.Send(new PerformanceData(i));
+            }
+
+            AutoResetEvent.WaitOne(); // Wait for all messages to be processed
+            stopwatch.Stop();
+            return stopwatch.ElapsedMilliseconds;
+        }
+    }
+
+    public class PerformanceConsumer : IHandleMessages<PerformanceData>
+    {
+        static int counter = 0;
+
+        public async Task Handle(PerformanceData message)
+        {
+            if (counter != message.Index)
+            {
+                Console.WriteLine($"Expected message index {counter}, but got {message.Index}");
+            }
+
+            Interlocked.Increment(ref counter);
+
+            if (counter == PerformanceCounter.Counter)
+            {
+                counter = 0; // Reset counter for next performance test
+                PerformanceProducer.AutoResetEvent.Set(); // Signal that all messages have been processed
+            }
+
+            await Task.CompletedTask;
+        }
+    }
+
+    public record PerformanceData(int Index);
+}