From 570209818de6d535c4dfcd0619c86bb653aca7f9 Mon Sep 17 00:00:00 2001
From: maliming <malimings@gmail.com>
Date: Sat, 29 Jul 2023 09:49:32 +0800
Subject: [PATCH] Update AbpCorrelationIdMiddleware.cs

---
 .../AspNetCore/Tracing/AbpCorrelationIdMiddleware.cs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/framework/src/Volo.Abp.AspNetCore/Volo/Abp/AspNetCore/Tracing/AbpCorrelationIdMiddleware.cs b/framework/src/Volo.Abp.AspNetCore/Volo/Abp/AspNetCore/Tracing/AbpCorrelationIdMiddleware.cs
index c73f8641ad..de93914d71 100644
--- a/framework/src/Volo.Abp.AspNetCore/Volo/Abp/AspNetCore/Tracing/AbpCorrelationIdMiddleware.cs
+++ b/framework/src/Volo.Abp.AspNetCore/Volo/Abp/AspNetCore/Tracing/AbpCorrelationIdMiddleware.cs
@@ -29,6 +29,18 @@ public class AbpCorrelationIdMiddleware : IMiddleware, ITransientDependency
         }
     }
 
+    protected virtual string GetCorrelationIdFromRequest(HttpContext context)
+    {
+        string correlationId = context.Request.Headers[_options.HttpHeaderName];
+        if (correlationId.IsNullOrEmpty())
+        {
+            correlationId = Guid.NewGuid().ToString("N");
+            context.Request.Headers[_options.HttpHeaderName] = correlationId;
+        }
+
+        return correlationId;
+    }
+
     protected virtual void CheckAndSetCorrelationIdOnResponse(
         HttpContext httpContext,
         AbpCorrelationIdOptions options,