Merge pull request #50 from mayuanyang/responseHandling

Unify response middleware can now handle request/response

src/Mediator.Net.TestUtil/Messages/GetGuidResponse.cs

@@ -6,6 +6,8 @@ namespace Mediator.Net.TestUtil.Messages
     public class GetGuidResponse : IResponse
     {
         public Guid Id { get; }
+
+        public string ToBeSetByMiddleware { get; set; }
 
         public GetGuidResponse(Guid id)
         {

src/Mediator.Net.TestUtil/Middlewares/ChangeRequestResultMiddleware.cs

@@ -0,0 +1,53 @@
+using System;
+using System.Runtime.ExceptionServices;
+using System.Threading;
+using System.Threading.Tasks;
+using Mediator.Net.Context;
+using Mediator.Net.Contracts;
+using Mediator.Net.Pipeline;
+using Mediator.Net.TestUtil.TestUtils;
+
+namespace Mediator.Net.TestUtil.Middlewares
+{
+    public static class ChangeRequestResultMiddleware
+    {
+        public static void UseChangeRequestResultMiddleware<TContext>(this IPipeConfigurator<TContext> configurator)
+            where TContext : IContext<IMessage>
+        {
+            configurator.AddPipeSpecification(new ChangeRequestResultMiddlewareSpecification<TContext>());
+        }
+    }
+
+    public class ChangeRequestResultMiddlewareSpecification<TContext> : IPipeSpecification<TContext>
+        where TContext : IContext<IMessage>
+    {
+        public bool ShouldExecute(TContext context, CancellationToken cancellationToken)
+        {
+            return true;
+        }
+
+        public Task BeforeExecute(TContext context, CancellationToken cancellationToken)
+        {
+            return Task.CompletedTask;
+        }
+
+        public Task Execute(TContext context, CancellationToken cancellationToken)
+        {
+            return Task.CompletedTask;
+        }
+
+        public Task AfterExecute(TContext context, CancellationToken cancellationToken)
+        {
+            (context.Result as dynamic).ToBeSetByMiddleware = "i am from middleware";
+            return Task.CompletedTask;
+        }
+
+
+        public Task OnException(Exception ex, TContext context)
+        {
+            RubishBox.Rublish.Add(ex);
+            ExceptionDispatchInfo.Capture(ex).Throw();
+            throw ex;
+        }
+    }
+}

src/Mediator.Net/Pipeline/RequestPipe.cs

@@ -34,6 +34,7 @@ public async Task<object> Connect(TContext context, CancellationToken cancellati
                 await _specification.Execute(context, cancellationToken).ConfigureAwait(false);
                 result = await (Next?.Connect(context, cancellationToken) ??
                                 ConnectToHandler(context, cancellationToken)).ConfigureAwait(false);
+                context.Result = result;
                 await _specification.AfterExecute(context, cancellationToken).ConfigureAwait(false);
             }
             catch (TargetInvocationException e)