Mediator work

Framework/Foundation/Lifecycles/CommandClassHandlerWrapper.cs

@@ -0,0 +1,25 @@
+using Mediator;
+
+namespace Toolkit.Framework.Foundation;
+
+public class CommandClassHandlerWrapper<TRequest, TResponse> where TRequest : class, ICommand<TResponse>
+{
+    private readonly MessageHandlerDelegate<TRequest, TResponse> handler;
+
+    public CommandClassHandlerWrapper(ICommandHandler<TRequest, TResponse> concreteHandler, 
+        IEnumerable<IPipelineBehavior<TRequest, TResponse>> pipelineBehaviours)
+    {
+        MessageHandlerDelegate<TRequest, TResponse> handler = concreteHandler.Handle;
+
+        foreach (IPipelineBehavior<TRequest, TResponse>? pipeline in pipelineBehaviours.Reverse())
+        {
+            MessageHandlerDelegate<TRequest, TResponse> handlerCopy = handler;
+            IPipelineBehavior<TRequest, TResponse> pipelineCopy = pipeline;
+            handler = (TRequest message, CancellationToken cancellationToken) => pipelineCopy.Handle(message, cancellationToken, handlerCopy);
+        }
+
+        this.handler = handler;
+    }
+
+    public ValueTask<TResponse> Handle(TRequest request, CancellationToken cancellationToken) => handler(request, cancellationToken);
+}

Framework/Foundation/Lifecycles/Mediator.cs

@@ -1,5 +1,4 @@
 using Mediator;
-using System.Diagnostics;
 
 namespace Toolkit.Framework.Foundation;
 
@@ -44,8 +43,7 @@ public class Mediator : IMediator
 
     public ValueTask<TResponse> Send<TResponse>(IRequest<TResponse> request, CancellationToken cancellationToken = default)
     {
-        dynamic? handler = factory.GetService(typeof(RequestClassHandlerWrapper<,>).MakeGenericType(request.GetType(), typeof(TResponse)));
-        
+        dynamic? handler = factory.GetService(typeof(RequestClassHandlerWrapper<,>).MakeGenericType(request.GetType(), typeof(TResponse)));  
         if (handler is not null)
         {
             return handler.Handle((dynamic)request, cancellationToken);
@@ -56,16 +54,70 @@ public class Mediator : IMediator
 
     public ValueTask<TResponse> Send<TResponse>(ICommand<TResponse> command, CancellationToken cancellationToken = default)
     {
-        throw new NotImplementedException();
+        dynamic? handler = factory.GetService(typeof(CommandClassHandlerWrapper<,>).MakeGenericType(command.GetType(), typeof(TResponse)));
+        if (handler is not null)
+        {
+            return handler.Handle((dynamic)command, cancellationToken);
+        }
+
+        return default;
     }
 
     public ValueTask<TResponse> Send<TResponse>(IQuery<TResponse> query, CancellationToken cancellationToken = default)
     {
-        throw new NotImplementedException();
+        dynamic? handler = factory.GetService(typeof(QueryClassHandlerWrapper<,>).MakeGenericType(query.GetType(), typeof(TResponse)));
+        if (handler is not null)
+        {
+            return handler.Handle((dynamic)query, cancellationToken);
+        }
+
+        return default;
     }
 
     public ValueTask<object?> Send(object message, CancellationToken cancellationToken = default)
     {
-        throw new NotImplementedException();
+        if (message.GetType().GetInterface(typeof(IRequest<>).Name) is { } requestInterface)
+        {
+            if (requestInterface.GetGenericArguments() is { Length: 1 } arguments)
+            {
+                Type responseType = arguments[0];
+
+                dynamic? handler = factory.GetService(typeof(RequestClassHandlerWrapper<,>).MakeGenericType(message.GetType(), responseType));
+                if (handler is not null)
+                {
+                    return handler.Handle((dynamic)message, cancellationToken);
+                }
+            }
+        }
+
+        if (message.GetType().GetInterface(typeof(ICommand<>).Name) is { } commandInterface)
+        {
+            if (commandInterface.GetGenericArguments() is { Length: 1 } arguments)
+            {
+                Type responseType = arguments[0];
+
+                dynamic? handler = factory.GetService(typeof(CommandClassHandlerWrapper<,>).MakeGenericType(message.GetType(), responseType));
+                if (handler is not null)
+                {
+                    return handler.Handle((dynamic)message, cancellationToken);
+                }
+            }
+        }
+
+        if (message.GetType().GetInterface(typeof(IQuery<>).Name) is { } queryInterface)
+        {
+            if (queryInterface.GetGenericArguments() is { Length: 1 } arguments)
+            {
+                Type responseType = arguments[0];
+
+                dynamic? handler = factory.GetService(typeof(QueryClassHandlerWrapper<,>).MakeGenericType(message.GetType(), responseType));
+                if (handler is not null)
+                {
+                    return handler.Handle((dynamic)message, cancellationToken);
+                }
+            }
+        }
+
+        return default;
     }
 }

Framework/Foundation/Lifecycles/QueryClassHandlerWrapper.cs

@@ -0,0 +1,26 @@
+using Mediator;
+
+namespace Toolkit.Framework.Foundation;
+
+public class QueryClassHandlerWrapper<TRequest, TResponse> where TRequest : class, IQuery<TResponse>
+{
+    private readonly MessageHandlerDelegate<TRequest, TResponse> handler;
+
+    public QueryClassHandlerWrapper(IQueryHandler<TRequest, TResponse> concreteHandler,
+        IEnumerable<IPipelineBehavior<TRequest, TResponse>> pipelineBehaviours)
+    {
+        MessageHandlerDelegate<TRequest, TResponse> handler = concreteHandler.Handle;
+
+        foreach (IPipelineBehavior<TRequest, TResponse>? pipeline in pipelineBehaviours.Reverse())
+        {
+            MessageHandlerDelegate<TRequest, TResponse> handlerCopy = handler;
+            IPipelineBehavior<TRequest, TResponse> pipelineCopy = pipeline;
+            handler = (TRequest message, CancellationToken cancellationToken) => pipelineCopy.Handle(message, cancellationToken, handlerCopy);
+        }
+
+        this.handler = handler;
+    }
+
+    public ValueTask<TResponse> Handle(TRequest request, CancellationToken cancellationToken) =>
+        handler(request, cancellationToken);
+}