Merge pull request #368 from dariogriffo/PostProcessorCancellationToken

Add CancellationToken to IRequestPostProcessor
jbogard · Feb 26, 2019 · 8e0c349 · 8e0c349
2 parents 94b83c1 + 8e35024
commit 8e0c349
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 7 deletions.
diff --git a/Directory.Build.props b/Directory.Build.props
@@ -2,6 +2,6 @@
   <PropertyGroup>
     <Authors>Jimmy Bogard</Authors>
     <LangVersion>latest</LangVersion>
-    <VersionPrefix>6.1.0</VersionPrefix>
+    <VersionPrefix>7.0.0</VersionPrefix>
   </PropertyGroup>
 </Project>
diff --git a/samples/MediatR.Examples/ConstrainedRequestPostProcessor.cs b/samples/MediatR.Examples/ConstrainedRequestPostProcessor.cs
@@ -1,4 +1,5 @@
 using System.IO;
+using System.Threading;
 using System.Threading.Tasks;
 using MediatR.Pipeline;
 
@@ -15,7 +16,7 @@ public ConstrainedRequestPostProcessor(TextWriter writer)
             _writer = writer;
         }
 
-        public Task Process(TRequest request, TResponse response)
+        public Task Process(TRequest request, TResponse response, CancellationToken cancellationToken)
         {
             return _writer.WriteLineAsync("- All Done with Ping");
         }

diff --git a/samples/MediatR.Examples/GenericRequestPostProcessor.cs b/samples/MediatR.Examples/GenericRequestPostProcessor.cs
@@ -1,4 +1,5 @@
 using System.IO;
+using System.Threading;
 using System.Threading.Tasks;
 using MediatR.Pipeline;
 
@@ -13,7 +14,7 @@ public GenericRequestPostProcessor(TextWriter writer)
             _writer = writer;
         }
 
-        public Task Process(TRequest request, TResponse response)
+        public Task Process(TRequest request, TResponse response, CancellationToken cancellationToken)
         {
             return _writer.WriteLineAsync("- All Done");
         }

diff --git a/src/MediatR/Pipeline/IRequestPostProcessor.cs b/src/MediatR/Pipeline/IRequestPostProcessor.cs
@@ -1,5 +1,6 @@
-namespace MediatR.Pipeline
+namespace MediatR.Pipeline
 {
+    using System.Threading;
     using System.Threading.Tasks;
 
     /// <summary>
@@ -14,7 +15,8 @@ public interface IRequestPostProcessor<in TRequest, in TResponse>
         /// </summary>
         /// <param name="request">Request instance</param>
         /// <param name="response">Response instance</param>
+        /// <param name="cancellationToken">Cancellation token</param>
         /// <returns>An awaitable task</returns>
-        Task Process(TRequest request, TResponse response);
+        Task Process(TRequest request, TResponse response, CancellationToken cancellationToken);
     }
 }
diff --git a/src/MediatR/Pipeline/RequestPostProcessorBehavior.cs b/src/MediatR/Pipeline/RequestPostProcessorBehavior.cs
@@ -24,7 +24,7 @@ public async Task<TResponse> Handle(TRequest request, CancellationToken cancella
 
             foreach (var processor in _postProcessors)
             {
-                await processor.Process(request, response).ConfigureAwait(false);
+                await processor.Process(request, response, cancellationToken).ConfigureAwait(false);
             }
 
             return response;

diff --git a/test/MediatR.Tests/Pipeline/RequestPostProcessorTests.cs b/test/MediatR.Tests/Pipeline/RequestPostProcessorTests.cs
@@ -30,7 +30,7 @@ public Task<Pong> Handle(Ping request, CancellationToken cancellationToken)
 
         public class PingPongPostProcessor : IRequestPostProcessor<Ping, Pong>
         {
-            public Task Process(Ping request, Pong response)
+            public Task Process(Ping request, Pong response, CancellationToken cancellationToken)
             {
                 response.Message = response.Message + " " + request.Message;