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