Refactor LoadSimulator to use IMessageSession and update DataDogMetrics setup

Author: hsvotwa

samples/logging/datadog/Core_10/Endpoint/DataDogMetrics.cs

@@ -86,8 +86,8 @@ static string[] ComposeTags(string messageType)
         return tags.ToArray();
     }
 
-    static readonly Dictionary<string, string> nameMapping = new Dictionary<string, string>
-        {
+    static readonly Dictionary<string, string> nameMapping = new()
+    {
             {"# of msgs successfully processed / sec", "nservicebus.processed"},
             {"# of msgs pulled from the input queue /sec", "nservicebus.fetched"},
             {"# of msgs failures / sec", "nservicebus.failed"},

samples/logging/datadog/Core_10/Endpoint/LoadSimulator.cs

@@ -4,7 +4,7 @@
 using NServiceBus;
 
 // Simulates busy (almost no delay) / quiet time in a sine wave
-class LoadSimulator(IEndpointInstance endpointInstance, TimeSpan minimumDelay, TimeSpan idleDuration)
+class LoadSimulator(IMessageSession messageSession, TimeSpan minimumDelay, TimeSpan idleDuration)
 {
     CancellationTokenSource tokenSource = new();
     TimeSpan idleDuration = TimeSpan.FromTicks(idleDuration.Ticks / 2);
@@ -43,7 +43,7 @@ TimeSpan NextDelay()
         return delay;
     }
 
-    Task Work() => endpointInstance.SendLocal(new SomeCommand());
+    Task Work() => messageSession.SendLocal(new SomeCommand());
 
     public Task Stop()
     {

samples/logging/datadog/Core_10/Endpoint/Program.cs

@@ -1,30 +1,44 @@
 using System;
+using Microsoft.Extensions.DependencyInjection;
 using NServiceBus;
+using Microsoft.Extensions.Hosting;
 
 const string EndpointName = "Samples.Metrics.Tracing.Endpoint";
 
 Console.Title = EndpointName;
-var endpointConfiguration = new EndpointConfiguration(EndpointName);
 
-endpointConfiguration.UseSerialization<SystemJsonSerializer>();
-endpointConfiguration.UseTransport<LearningTransport>();
+var host = Host.CreateDefaultBuilder(args)
+    .UseConsoleLifetime()
+    .UseNServiceBus(_ =>
+    {
+        var endpointConfiguration = new EndpointConfiguration(EndpointName);
+
+        endpointConfiguration.UseSerialization<SystemJsonSerializer>();
+        endpointConfiguration.UseTransport<LearningTransport>();
+
+        DataDogMetrics.Setup(endpointConfiguration);
+
+        return endpointConfiguration;
+    })
+    .Build();
 
-DataDogMetrics.Setup(endpointConfiguration);
+await host.StartAsync();
 
-var endpointInstance = await Endpoint.Start(endpointConfiguration);
+var endpointInstance = host.Services.GetRequiredService<IMessageSession>();
 
 var simulator = new LoadSimulator(endpointInstance, TimeSpan.Zero, TimeSpan.FromSeconds(10));
 await simulator.Start();
 
 try
 {
-    Console.WriteLine("Endpoint started. Press 'enter' to send a message");
-    Console.WriteLine("Press ESC key to quit");
+    Console.WriteLine("Endpoint started.");
+    Console.WriteLine("Press [ENTER] to send additional messages.");
+    Console.WriteLine("Press [Q] to quit.");
 
     while (true)
     {
         var key = Console.ReadKey(true);
-        if (key.Key == ConsoleKey.Escape)
+        if (key.Key == ConsoleKey.Q)
         {
             break;
         }
@@ -35,5 +49,5 @@
 finally
 {
     await simulator.Stop();
-    await endpointInstance.Stop();
+    await host.StopAsync();
 }