Update AbpCorrelationIdMiddleware.cs

2 years ago · 570209818d
parent 21e899b8bd
commit 570209818d
1 changed files with 12 additions and 0 deletions
--- 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,