Refactor LoadSimulator to use IMessageSession and update Program.cs for host-based endpoint initialization

Author: hsvotwa

samples/logging/new-relic/Metrics_6/Endpoint/LoadSimulator.cs

@@ -4,9 +4,9 @@
 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 CancellationTokenSource();
+    CancellationTokenSource tokenSource = new();
     TimeSpan idleDuration = TimeSpan.FromTicks(idleDuration.Ticks / 2);
     Task fork;
 
@@ -42,7 +42,7 @@ TimeSpan NextDelay()
         return delay;
     }
 
-    Task Work() => endpointInstance.SendLocal(new SomeCommand());
+    Task Work() => messageSession.SendLocal(new SomeCommand());
 
     public Task Stop()
     {

samples/logging/new-relic/Metrics_6/Endpoint/Program.cs

@@ -1,15 +1,29 @@
 using System;
+using Microsoft.Extensions.Hosting;
 using Endpoint;
+using Microsoft.Extensions.DependencyInjection;
 using NServiceBus;
 
-var endpointConfiguration = new EndpointConfiguration(Console.Title = "TracingEndpoint");
+Console.Title = "TracingEndpoint";
 
-endpointConfiguration.UseSerialization<SystemJsonSerializer>();
-endpointConfiguration.UseTransport<LearningTransport>();
+var host = Host.CreateDefaultBuilder(args)
+    .UseConsoleLifetime()
+    .UseNServiceBus(_ =>
+    {
+        var endpointConfiguration = new EndpointConfiguration("TracingEndpoint");
+
+        endpointConfiguration.UseSerialization<SystemJsonSerializer>();
+        endpointConfiguration.UseTransport<LearningTransport>();
+
+        NewRelicMetrics.Setup(endpointConfiguration);
+
+        return endpointConfiguration;
+    })
+    .Build();
 
-NewRelicMetrics.Setup(endpointConfiguration);
+await host.StartAsync();
 
-var endpointInstance = await NServiceBus.Endpoint.Start(endpointConfiguration);
+var endpointInstance = host.Services.GetRequiredService<IMessageSession>();
 
 #region newrelic-load-simulator
 
@@ -22,13 +36,13 @@
 {
     Console.WriteLine("Endpoint started.");
     Console.WriteLine("Press [ENTER] to send additional messages.");
-    Console.WriteLine("Press [ESC] to quit.");
+    Console.WriteLine("Press [Q] to quit.");
 
     while (true)
     {
         switch (Console.ReadKey(true).Key)
         {
-            case ConsoleKey.Escape:
+            case ConsoleKey.Q:
                 return;
             case ConsoleKey.Enter:
                 await endpointInstance.SendLocal(new SomeCommand());
@@ -39,5 +53,5 @@
 finally
 {
     await simulator.Stop();
-    await endpointInstance.Stop();
+    await host.StopAsync();
 }