diff --git a/src/Flo/Builder.cs b/src/Flo/Builder.cs index 54f3ac1..31c4a3e 100644 --- a/src/Flo/Builder.cs +++ b/src/Flo/Builder.cs @@ -83,7 +83,7 @@ public TBuilder Add(Func> handlerFactory) if (handler != null) { - return handler.HandleAsync(input, next, CancellationToken.None); + return handler.HandleAsync(input, next); } return next.Invoke(input); diff --git a/src/Flo/IHandler.cs b/src/Flo/IHandler.cs index 43a2c1b..1eefe8f 100644 --- a/src/Flo/IHandler.cs +++ b/src/Flo/IHandler.cs @@ -6,7 +6,7 @@ namespace Flo { public interface IHandler { - Task HandleAsync(TIn input, Func> next, CancellationToken token); + Task HandleAsync(TIn input, Func> next); } public interface IHandler : IHandler diff --git a/test/Flo.Tests/HandlerTests.cs b/test/Flo.Tests/HandlerTests.cs index fdc9573..5453e41 100644 --- a/test/Flo.Tests/HandlerTests.cs +++ b/test/Flo.Tests/HandlerTests.cs @@ -56,7 +56,7 @@ async Task it_can_use_a_custom_service_provider() class TestHandler : IHandler { - public Task HandleAsync(TestContext input, Func> next, CancellationToken cancellationToken) + public Task HandleAsync(TestContext input, Func> next) { input.Add(Guid.NewGuid().ToString(), Guid.NewGuid()); return next.Invoke(input); @@ -72,7 +72,7 @@ public LazyHandler(Action callback) _callback = callback; } - public Task HandleAsync(object input, Func> next, CancellationToken cancellationToken) + public Task HandleAsync(object input, Func> next) { _callback.Invoke(); return next.Invoke(input); @@ -81,7 +81,7 @@ public Task HandleAsync(object input, Func> next, C class StringLengthCountHandler : IHandler { - public Task HandleAsync(string input, Func> next, CancellationToken cancellationToken) + public Task HandleAsync(string input, Func> next) { return Task.FromResult(input.Length); } @@ -98,7 +98,7 @@ public OverridingHandler(string output) _output = output; } - public Task HandleAsync(string input, Func> next, CancellationToken cancellationToken) + public Task HandleAsync(string input, Func> next) { return Task.FromResult("Override"); }