Username-header sample to .net 10 (#7493)

Author: irinascurtu

samples/username-header/Core_10/Endpoint1/Endpoint1.csproj

@@ -0,0 +1,14 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <PropertyGroup>
+    <TargetFramework>net10.0</TargetFramework>
+    <LangVersion>preview</LangVersion>
+    <OutputType>Exe</OutputType>
+    <ImplicitUsings>enable</ImplicitUsings>
+  </PropertyGroup>
+  <ItemGroup>
+    <PackageReference Include="NServiceBus.Extensions.Hosting" Version="4.0.0-alpha.1" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\Shared\Shared.csproj" />
+  </ItemGroup>
+</Project>

samples/username-header/Core_10/Endpoint1/Program.cs

@@ -0,0 +1,59 @@
+using System;
+using System.Security.Principal;
+using System.Threading.Tasks;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Logging;
+using NServiceBus;
+using NServiceBus.MessageMutator;
+
+
+var principalAccessor = new PrincipalAccessor();
+
+var builder = Host.CreateApplicationBuilder(args);
+var host = Host.CreateDefaultBuilder(args)
+    .UseNServiceBus(x =>
+    {
+        var endpointConfiguration = new EndpointConfiguration("Samples.UsernameHeader.Endpoint1");
+        endpointConfiguration.UseSerialization<SystemJsonSerializer>();
+        endpointConfiguration.UseTransport(new LearningTransport());
+
+
+        #region component-registration-sender
+
+        builder.Services.AddSingleton<IPrincipalAccessor>(principalAccessor);
+        var serviceProvider = builder.Services.BuildServiceProvider();
+
+        var logger = serviceProvider.GetRequiredService<ILogger<AddUserNameToOutgoingHeadersMutator>>();
+        var mutator = new AddUserNameToOutgoingHeadersMutator(principalAccessor, logger);
+        endpointConfiguration.RegisterMessageMutator(mutator);
+
+        #endregion
+        return endpointConfiguration;
+    }).Build();
+
+
+await host.StartAsync();
+
+
+#region send-message
+
+async Task SendMessage(int userNumber)
+{
+    var identity = new GenericIdentity($"FakeUser{userNumber}");
+    principalAccessor.CurrentPrincipal = new GenericPrincipal(identity, new string[0]);
+
+    var messageSession = host.Services.GetRequiredService<IMessageSession>();
+
+    var message = new MyMessage();
+    await messageSession.Send("Samples.UsernameHeader.Endpoint2", message);
+}
+
+await Task.WhenAll(SendMessage(1), SendMessage(2));
+
+#endregion
+
+Console.WriteLine("Message sent. Press any key to exit");
+Console.ReadKey();
+
+await host.StopAsync();

samples/username-header/Core_10/Endpoint2/Endpoint2.csproj

@@ -0,0 +1,14 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <PropertyGroup>
+    <TargetFramework>net10.0</TargetFramework>
+    <LangVersion>preview</LangVersion>
+    <OutputType>Exe</OutputType>
+    <ImplicitUsings>enable</ImplicitUsings>
+  </PropertyGroup>
+  <ItemGroup>
+   <PackageReference Include="NServiceBus.Extensions.Hosting" Version="4.0.0-alpha.1" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\Shared\Shared.csproj" />
+  </ItemGroup>
+</Project>

samples/username-header/Core_10/Endpoint2/HandlerUsingAccessor.cs

@@ -0,0 +1,30 @@
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging;
+using NServiceBus;
+
+#region handler-using-custom-header
+
+public class HandlerUsingAccessor :
+    IHandleMessages<MyMessage>
+{
+    readonly IPrincipalAccessor principalAccessor;
+    private readonly ILogger<HandlerUsingAccessor> logger;
+
+    public HandlerUsingAccessor(IPrincipalAccessor principalAccessor, ILogger<HandlerUsingAccessor> logger)
+    {
+        this.principalAccessor = principalAccessor;
+        this.logger = logger;
+    }
+
+    public Task Handle(MyMessage message, IMessageHandlerContext context)
+    {
+        var headers = context.MessageHeaders;
+        var usernameFromHeader = headers["UserName"];
+        var usernameFromAccessor = principalAccessor?.CurrentPrincipal?.Identity?.Name ?? "null";
+        logger.LogInformation("Username extracted from header: {UsernameFromHeader}", usernameFromHeader);
+        logger.LogInformation("Username extracted from accessor: {UsernameFromAccessor}", usernameFromAccessor);
+        return Task.CompletedTask;
+    }
+}
+
+#endregion

samples/username-header/Core_10/Endpoint2/Program.cs

@@ -0,0 +1,36 @@
+using System;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+using NServiceBus;
+using NServiceBus.MessageMutator;
+
+Console.Title = "Endpoint2";
+
+var builder = Host.CreateApplicationBuilder(args);
+
+var endpointConfiguration = new EndpointConfiguration("Samples.UsernameHeader.Endpoint2");
+endpointConfiguration.UseSerialization<SystemJsonSerializer>();
+endpointConfiguration.UseTransport(new LearningTransport());
+
+#region component-registration-receiver
+// Register both services
+builder.Services.AddSingleton<IPrincipalAccessor, PrincipalAccessor>();
+builder.Services.AddSingleton<SetCurrentPrincipalBasedOnHeaderMutator>();
+
+var serviceProvider = builder.Services.BuildServiceProvider();
+var principalAccessor = serviceProvider.GetRequiredService<IPrincipalAccessor>(); // Use the interface type here
+var mutator = serviceProvider.GetRequiredService<SetCurrentPrincipalBasedOnHeaderMutator>();
+
+endpointConfiguration.RegisterMessageMutator(mutator);
+
+endpointConfiguration.RegisterComponents(c =>
+{
+    //Register the accessor in the container so that the handler can access it
+    c.AddSingleton<IPrincipalAccessor>(principalAccessor);
+});
+
+#endregion
+
+builder.UseNServiceBus(endpointConfiguration);
+await builder.Build().RunAsync();
+

samples/username-header/Core_10/Shared/AddUserNameToOutgoingHeadersMutator.cs

@@ -0,0 +1,32 @@
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging;
+using NServiceBus.MessageMutator;
+
+#region username-header-mutator
+
+public class AddUserNameToOutgoingHeadersMutator:
+    IMutateOutgoingTransportMessages
+{
+    readonly IPrincipalAccessor principalAccessor;
+    private readonly ILogger<AddUserNameToOutgoingHeadersMutator> logger;
+
+    public AddUserNameToOutgoingHeadersMutator(IPrincipalAccessor principalAccessor, ILogger<AddUserNameToOutgoingHeadersMutator> logger)
+    {
+        this.principalAccessor = principalAccessor;
+        this.logger = logger;
+    }
+
+    public Task MutateOutgoing(MutateOutgoingTransportMessageContext context)
+    {
+        if (principalAccessor.CurrentPrincipal?.Identity.Name != null)
+        {
+
+            logger.LogInformation("Adding CurrentPrincipal user to headers");
+            context.OutgoingHeaders["UserName"] = principalAccessor.CurrentPrincipal.Identity.Name;
+        }
+
+        return Task.CompletedTask;
+    }
+}
+
+#endregion

samples/username-header/Core_10/Shared/IPrincipalAccessor.cs

@@ -0,0 +1,6 @@
+using System.Security.Principal;
+
+public interface IPrincipalAccessor
+{
+    IPrincipal CurrentPrincipal { get; set; }
+}

samples/username-header/Core_10/Shared/MyMessage.cs

@@ -0,0 +1,6 @@
+using NServiceBus;
+
+public class MyMessage :
+    IMessage
+{
+}

samples/username-header/Core_10/Shared/PrincipalAccessor.cs

@@ -0,0 +1,36 @@
+using System.Security.Principal;
+using System.Threading;
+
+public class PrincipalAccessor : IPrincipalAccessor
+{
+    AsyncLocal<CurrentPrincipalHolder> principalCurrent = new AsyncLocal<CurrentPrincipalHolder>();
+
+    public IPrincipal CurrentPrincipal
+    {
+        get
+        {
+            return principalCurrent.Value?.CurrentPrincipal;
+        }
+        set
+        {
+            var holder = principalCurrent.Value;
+            if (holder != null)
+            {
+                // Clear current IPrincipal trapped in the AsyncLocals, as its done.
+                holder.CurrentPrincipal = null;
+            }
+
+            if (value != null)
+            {
+                // Use an object indirection to hold the IPrincipal in the AsyncLocal,
+                // so it can be cleared in all ExecutionContexts when its cleared.
+                principalCurrent.Value = new CurrentPrincipalHolder { CurrentPrincipal = value };
+            }
+        }
+    }
+
+    class CurrentPrincipalHolder
+    {
+        public IPrincipal CurrentPrincipal;
+    }
+}

samples/username-header/Core_10/Shared/SetCurrentPrincipalBasedOnHeaderMutator.cs

@@ -0,0 +1,32 @@
+using System.Security.Principal;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging;
+using NServiceBus.MessageMutator;
+
+#region set-principal-from-header-mutator
+public class SetCurrentPrincipalBasedOnHeaderMutator :
+    IMutateIncomingTransportMessages
+{
+  
+    readonly IPrincipalAccessor principalAccessor;
+    private readonly ILogger<SetCurrentPrincipalBasedOnHeaderMutator> logger;
+
+    public SetCurrentPrincipalBasedOnHeaderMutator(IPrincipalAccessor principalAccessor, ILogger<SetCurrentPrincipalBasedOnHeaderMutator> logger)
+    {
+        this.principalAccessor = principalAccessor;
+        this.logger = logger;
+    }
+
+    public Task MutateIncoming(MutateIncomingTransportMessageContext context)
+    {
+        if (context.Headers.TryGetValue("UserName", out var userNameHeader))
+        {
+            logger.LogInformation("Adding CurrentPrincipal user from headers");
+            var identity = new GenericIdentity(userNameHeader);
+            principalAccessor.CurrentPrincipal = new GenericPrincipal(identity, new string[0]);
+        }
+
+        return Task.CompletedTask;
+    }
+}
+#endregion