Use IMiddleware interface

Dotnet-Boxed · Oct 12, 2018 · eacb408 · eacb408
1 parent 5546cbe
commit eacb408
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 14 deletions.
diff --git a/Source/Boxed.AspNetCore/Middleware/HttpExceptionMiddleware.cs b/Source/Boxed.AspNetCore/Middleware/HttpExceptionMiddleware.cs
@@ -6,17 +6,13 @@ namespace Boxed.AspNetCore.Middleware
     using Microsoft.Extensions.DependencyInjection;
     using Microsoft.Extensions.Logging;
 
-    internal class HttpExceptionMiddleware
+    internal class HttpExceptionMiddleware : IMiddleware
     {
-        private readonly RequestDelegate next;
-
-        public HttpExceptionMiddleware(RequestDelegate next) => this.next = next;
-
-        public async Task Invoke(HttpContext context)
+        public async Task InvokeAsync(HttpContext context, RequestDelegate next)
         {
             try
             {
-                await this.next.Invoke(context).ConfigureAwait(false);
+                await next.Invoke(context).ConfigureAwait(false);
             }
             catch (HttpException httpException)
             {

diff --git a/Source/Boxed.AspNetCore/Middleware/InternalServerErrorOnExceptionMiddleware.cs b/Source/Boxed.AspNetCore/Middleware/InternalServerErrorOnExceptionMiddleware.cs
@@ -5,17 +5,13 @@ namespace Boxed.AspNetCore.Middleware
     using Microsoft.Extensions.DependencyInjection;
     using Microsoft.Extensions.Logging;
 
-    internal class InternalServerErrorOnExceptionMiddleware
+    internal class InternalServerErrorOnExceptionMiddleware : IMiddleware
     {
-        private readonly RequestDelegate next;
-
-        public InternalServerErrorOnExceptionMiddleware(RequestDelegate next) => this.next = next;
-
-        public async Task Invoke(HttpContext context)
+        public async Task InvokeAsync(HttpContext context, RequestDelegate next)
         {
             try
             {
-                await this.next.Invoke(context).ConfigureAwait(false);
+                await next.Invoke(context).ConfigureAwait(false);
             }
             catch
             {