tk/Ocelot
1
0
Fork 0
mirror of https://github.com/nsnail/Ocelot.git synced 2025-04-22 06:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'release-5.4.0'

Browse Source
This commit is contained in:
Tom Gardham-Pallister 2018-04-07 21:36:29 +01:00
commit 035f8f1123
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/Ocelot/RequestId/Middleware/ReRouteRequestIdMiddleware.cs
View File

@ -32,29 +32,24 @@ namespace Ocelot.RequestId.Middleware
private void SetOcelotRequestId(DownstreamContext context) private void SetOcelotRequestId(DownstreamContext context)
{ {
// if get request ID is set on upstream request then retrieve it
var key = context.DownstreamReRoute.RequestIdKey ?? DefaultRequestIdKey.Value; var key = context.DownstreamReRoute.RequestIdKey ?? DefaultRequestIdKey.Value;
if (context.HttpContext.Request.Headers.TryGetValue(key, out var upstreamRequestIds)) if (context.HttpContext.Request.Headers.TryGetValue(key, out var upstreamRequestIds))
{ {
context.HttpContext.TraceIdentifier = upstreamRequestIds.First(); context.HttpContext.TraceIdentifier = upstreamRequestIds.First();
//check if we have previous id in scoped repo
var previousRequestId = _requestScopedDataRepository.Get<string>("RequestId"); var previousRequestId = _requestScopedDataRepository.Get<string>("RequestId");
if (!previousRequestId.IsError && !string.IsNullOrEmpty(previousRequestId.Data) && previousRequestId.Data != context.HttpContext.TraceIdentifier) if (!previousRequestId.IsError && !string.IsNullOrEmpty(previousRequestId.Data) && previousRequestId.Data != context.HttpContext.TraceIdentifier)
{ {
//we have a previous request id lets store it and update request id
_requestScopedDataRepository.Add("PreviousRequestId", previousRequestId.Data); _requestScopedDataRepository.Add("PreviousRequestId", previousRequestId.Data);
_requestScopedDataRepository.Update("RequestId", context.HttpContext.TraceIdentifier); _requestScopedDataRepository.Update("RequestId", context.HttpContext.TraceIdentifier);
} }
else else
{ {
//else just add request id
_requestScopedDataRepository.Add("RequestId", context.HttpContext.TraceIdentifier); _requestScopedDataRepository.Add("RequestId", context.HttpContext.TraceIdentifier);
} }
} }
// set request ID on downstream request, if required
var requestId = new RequestId(context.DownstreamReRoute.RequestIdKey, context.HttpContext.TraceIdentifier); var requestId = new RequestId(context.DownstreamReRoute.RequestIdKey, context.HttpContext.TraceIdentifier);
if (ShouldAddRequestId(requestId, context.DownstreamRequest.Headers)) if (ShouldAddRequestId(requestId, context.DownstreamRequest.Headers))