diff --git a/Net.Sdk.Web.Extensions/Middleware/CorrelationVectorMiddleware.cs b/Net.Sdk.Web.Extensions/Middleware/CorrelationVectorMiddleware.cs
index 4e5019d..ec6d0cf 100644
--- a/Net.Sdk.Web.Extensions/Middleware/CorrelationVectorMiddleware.cs
+++ b/Net.Sdk.Web.Extensions/Middleware/CorrelationVectorMiddleware.cs
@@ -18,7 +18,7 @@ public sealed class CorrelationVectorMiddleware : IMiddleware
this.options = options.ThrowIfNull().Value.ThrowIfNull();
}
- public Task InvokeAsync(HttpContext context, RequestDelegate next)
+ public async Task InvokeAsync(HttpContext context, RequestDelegate next)
{
var cv = new CorrelationVector();
if (context.Items.TryGetValue(this.options.Header, out var correlationVectorVal) &&
@@ -29,6 +29,8 @@ public sealed class CorrelationVectorMiddleware : IMiddleware
}
context.SetCorrelationVector(cv);
- return next(context);
+ await next(context);
+
+ context.Response.Headers.Append(this.options.Header, context.GetCorrelationVector().ToString());
}
}
diff --git a/Net.Sdk.Web.Extensions/Net.Sdk.Web.Extensions.csproj b/Net.Sdk.Web.Extensions/Net.Sdk.Web.Extensions.csproj
index fa14bd6..0193d53 100644
--- a/Net.Sdk.Web.Extensions/Net.Sdk.Web.Extensions.csproj
+++ b/Net.Sdk.Web.Extensions/Net.Sdk.Web.Extensions.csproj
@@ -6,7 +6,7 @@
enable
Library
true
- 0.6
+ 0.6.1
Alexandru Macocian
LICENSE
true