fix dependency injection

khavishbhundoo · May 19, 2024 · 8bf82dc · 8bf82dc
1 parent acc911f
commit 8bf82dc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/CallMyTrade.Api/Middleware/ProtectEndpointMiddleware.cs b/src/CallMyTrade.Api/Middleware/ProtectEndpointMiddleware.cs
@@ -2,6 +2,7 @@
 using System.Text.Json;
 using System.Threading.Tasks;
 using Core.CallMyTrade;
+using Light.GuardClauses;
 using Microsoft.AspNetCore.Http;
 using Serilog;
 
@@ -20,9 +21,11 @@ public sealed class ProtectEndpointMiddleware
         "52.32.178.7"
     ];
 
-    public ProtectEndpointMiddleware(RequestDelegate next)
+    public ProtectEndpointMiddleware(RequestDelegate next, IDiagnosticContext diagnosticContext)
     {
-        _next = next;
+        _diagnosticContext = diagnosticContext.MustNotBeNull();
+        _next = next.MustNotBeNull();
+
     }
 
     public async Task Invoke(HttpContext context)