feat: add benchmark test project for Go vs .NET server comparison

Side-by-side performance benchmarks using NATS.Client.Core against both servers on ephemeral ports. Includes core pub/sub, request/reply latency, and JetStream throughput tests with comparison output and benchmarks_comparison.md results. Also fixes timestamp flakiness in StoreInterfaceTests by using explicit timestamps.
2026-03-13 01:23:31 -04:00
parent e9c86c51c3
commit 37575dc41c
28 changed files with 2264 additions and 12 deletions
--- a/tests/NATS.Server.Benchmark.Tests/JetStream/DurableConsumerFetchTests.cs
+++ b/tests/NATS.Server.Benchmark.Tests/JetStream/DurableConsumerFetchTests.cs
@@ -0,0 +1,109 @@
+using System.Diagnostics;
+using NATS.Client.Core;
+using NATS.Client.JetStream;
+using NATS.Client.JetStream.Models;
+using NATS.Server.Benchmark.Tests.Harness;
+using NATS.Server.Benchmark.Tests.Infrastructure;
+using Xunit.Abstractions;
+
+namespace NATS.Server.Benchmark.Tests.JetStream;
+
+[Collection("Benchmark-JetStream")]
+public class DurableConsumerFetchTests(JetStreamServerPairFixture fixture, ITestOutputHelper output)
+{
+    [Fact]
+    [Trait("Category", "Benchmark")]
+    public async Task JSDurableFetch_Throughput()
+    {
+        const int payloadSize = 128;
+        const int messageCount = 5_000;
+        const int fetchBatchSize = 500;
+
+        var dotnetResult = await RunDurableFetch("JS Durable Fetch (128B)", "DotNet", payloadSize, messageCount, fetchBatchSize, fixture.CreateDotNetClient);
+
+        if (fixture.GoAvailable)
+        {
+            var goResult = await RunDurableFetch("JS Durable Fetch (128B)", "Go", payloadSize, messageCount, fetchBatchSize, fixture.CreateGoClient);
+            BenchmarkResultWriter.WriteComparison(output, goResult, dotnetResult);
+        }
+        else
+        {
+            BenchmarkResultWriter.WriteSingle(output, dotnetResult);
+        }
+    }
+
+    private static async Task<BenchmarkResult> RunDurableFetch(
+        string name, string serverType, int payloadSize, int messageCount, int fetchBatchSize,
+        Func<NatsConnection> createClient)
+    {
+        var payload = new byte[payloadSize];
+        var streamName = $"BENCH_DUR_{serverType.ToUpperInvariant()}_{Guid.NewGuid():N}"[..30];
+        var subject = $"bench.js.durable.{serverType.ToLowerInvariant()}";
+        var consumerName = $"bench-dur-{serverType.ToLowerInvariant()}";
+
+        await using var nats = createClient();
+        await nats.ConnectAsync();
+        var js = new NatsJSContext(nats);
+
+        await js.CreateStreamAsync(new StreamConfig(streamName, [subject])
+        {
+            Storage = StreamConfigStorage.Memory,
+            Retention = StreamConfigRetention.Limits,
+            MaxMsgs = 10_000_000,
+        });
+
+        try
+        {
+            // Pre-populate stream
+            var pubTasks = new List<ValueTask<PubAckResponse>>(1000);
+            for (var i = 0; i < messageCount; i++)
+            {
+                pubTasks.Add(js.PublishAsync(subject, payload));
+                if (pubTasks.Count >= 1000)
+                {
+                    foreach (var t in pubTasks)
+                        await t;
+                    pubTasks.Clear();
+                }
+            }
+
+            foreach (var t in pubTasks)
+                await t;
+
+            // Create durable consumer
+            var consumer = await js.CreateOrUpdateConsumerAsync(streamName, new ConsumerConfig(consumerName)
+            {
+                AckPolicy = ConsumerConfigAckPolicy.None,
+            });
+
+            // Fetch in batches
+            var received = 0;
+            var sw = Stopwatch.StartNew();
+
+            while (received < messageCount)
+            {
+                await foreach (var msg in consumer.FetchAsync<byte[]>(new NatsJSFetchOpts { MaxMsgs = fetchBatchSize }))
+                {
+                    received++;
+                    if (received >= messageCount)
+                        break;
+                }
+            }
+
+            sw.Stop();
+
+            return new BenchmarkResult
+            {
+                Name = name,
+                ServerType = serverType,
+                TotalMessages = received,
+                TotalBytes = (long)received * payloadSize,
+                Duration = sw.Elapsed,
+            };
+        }
+        finally
+        {
+            await js.DeleteStreamAsync(streamName);
+        }
+    }
+}