feat: Added RegisterMessageType method for explicit registration of m…

src/Testing/CoreTests/Runtime/Handlers/HandlerGraphTests.cs

@@ -34,8 +34,41 @@ public async Task can_find_the_message_type_by_the_message_type_name_of_one_of_i
         graph.TryFindMessageType(typeof(IMessageMarker).ToMessageTypeName(), out var markedType).ShouldBeTrue();
         markedType.ShouldBe(typeof(MarkedMessage));
     }
+
+    [Fact]
+    public async Task register_message_type()
+    {
+        using var host = await Host.CreateDefaultBuilder()
+            .UseWolverine(opts =>
+            {
+                opts.RegisterMessageType(typeof(DummyMessage), "custom-alias");
+            }).StartAsync();
+
+        var graph = host.Services.GetRequiredService<HandlerGraph>();
+
+        graph.TryFindMessageType(typeof(DummyMessage).ToMessageTypeName(), out _).ShouldBeFalse();
+
+        graph.TryFindMessageType("custom-alias", out var type).ShouldBeTrue();
+        type.ShouldBe(typeof(DummyMessage));
+    }
+
+    [Fact]
+    public async Task register_message_types_with_same_alias_throws_invalid_operation_exception()
+    {
+        await Should.ThrowAsync<InvalidOperationException>(async () =>
+        {
+            using var host = await Host.CreateDefaultBuilder()
+                .UseWolverine(opts =>
+                {
+                    opts.RegisterMessageType(typeof(DummyMessage), "custom-alias");
+                    opts.RegisterMessageType(typeof(string), "custom-alias");
+                }).StartAsync();
+        });
+    }
 }
 
+public class DummyMessage { }
+
 public interface IMessageMarker;
 
 [InteropMessage(typeof(IMessageMarker))]

src/Wolverine/Runtime/Handlers/HandlerGraph.cs

@@ -460,6 +460,22 @@ public void RegisterMessageType(Type messageType)
         _messageTypes = _messageTypes.AddOrUpdate(messageType.ToMessageTypeName(), messageType);
         _replyTypes = _replyTypes.Add(messageType);
     }
+
+    public void RegisterMessageType(Type messageType, string messageAlias)
+    {
+        if (_messageTypes.TryFind(messageAlias, out var type))
+        {
+            throw new InvalidOperationException($"Cannot register type {type} with alias {messageAlias} because alias is already used");
+        }
+
+        if (_replyTypes.Contains(messageType))
+        {
+            return;
+        }
+
+        _messageTypes = _messageTypes.AddOrUpdate(messageAlias, messageType);
+        _replyTypes = _replyTypes.Add(messageType);
+    }
 
     public IEnumerable<HandlerChain> AllChains()
     {

src/Wolverine/WolverineOptions.cs

@@ -299,4 +299,9 @@ internal void ReadJasperFxOptions(JasperFxOptions jasperfx)
             _autoBuildMessageStorageOnStartup = jasperfx.ActiveProfile.ResourceAutoCreate;
         }
     }
+
+    public void RegisterMessageType(Type messageType, string messageAlias)
+    {
+        HandlerGraph.RegisterMessageType(messageType, messageAlias);
+    }
 }