From 8dbf2b2b94fc996ce09753fc0f187120918831b8 Mon Sep 17 00:00:00 2001
From: Alex Macocian <amacocian@yahoo.com>
Date: Fri, 16 Aug 2024 17:27:33 +0200
Subject: [PATCH] Add CV to response

---
 .../Middleware/CorrelationVectorMiddleware.cs               | 6 ++++--
 Net.Sdk.Web.Extensions/Net.Sdk.Web.Extensions.csproj        | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

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 @@
     <Nullable>enable</Nullable>
     <OutputType>Library</OutputType>
     <IsPackable>true</IsPackable>
-    <Version>0.6</Version>
+    <Version>0.6.1</Version>
     <Authors>Alexandru Macocian</Authors>
     <PackageLicenseFile>LICENSE</PackageLicenseFile>
     <PackageRequireLicenseAcceptance>true</PackageRequireLicenseAcceptance>