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

Fix request id middleware

Browse Source
This commit is contained in:
Philip Wood
2017-04-18 11:05:37 +01:00
parent da922c0e04
commit 8b93f44077
2 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Ocelot/Middleware/OcelotMiddlewareExtensions.cs
View File

@@ -72,7 +72,7 @@ namespace Ocelot.Middleware
builder.UseRateLimiting();
// Now we can look for the requestId
builder.UseRequestIdMiddleware(); //obsolete?
builder.UseRequestIdMiddleware();
// Allow pre authentication logic. The idea being people might want to run something custom before what is built in.
builder.UseIfNotNull(middlewareConfiguration.PreAuthenticationMiddleware);

44
src/Ocelot/RequestId/Middleware/RequestIdMiddleware.cs
View File

@@ -41,34 +41,40 @@ namespace Ocelot.RequestId.Middleware
private void SetOcelotRequestId(HttpContext context)
{
var key = DefaultRequestIdKey.Value;
if (DownstreamRoute.ReRoute.RequestIdKey != null)
{
key = DownstreamRoute.ReRoute.RequestIdKey;
}
// if get request ID is set on upstream request then retrieve it
var key = DownstreamRoute.ReRoute.RequestIdKey ?? DefaultRequestIdKey.Value;
StringValues requestIds;
if (context.Request.Headers.TryGetValue(key, out requestIds))
StringValues upstreamRequestIds;
if (context.Request.Headers.TryGetValue(key, out upstreamRequestIds))
{
var requestId = requestIds.First();
var downstreamRequestHeaders = DownstreamRequest.Headers;
context.TraceIdentifier = upstreamRequestIds.First();
}
if (!string.IsNullOrEmpty(requestId) &&
!HeaderExists(key, downstreamRequestHeaders))
{
downstreamRequestHeaders.Add(key, requestId);
}
// set request ID on downstream request, if required
var requestId = new RequestId(DownstreamRoute?.ReRoute?.RequestIdKey, context.TraceIdentifier);
context.TraceIdentifier = requestId;
if (ShouldAddRequestId(requestId, DownstreamRequest.Headers))
{
AddRequestIdHeader(requestId, DownstreamRequest);
}
}
private bool HeaderExists(string headerKey, HttpRequestHeaders headers)
private bool ShouldAddRequestId(RequestId requestId, HttpRequestHeaders headers)
{
return !string.IsNullOrEmpty(requestId?.RequestIdKey)
&& !string.IsNullOrEmpty(requestId.RequestIdValue)
&& !RequestIdInHeaders(requestId, headers);
}
private bool RequestIdInHeaders(RequestId requestId, HttpRequestHeaders headers)
{
IEnumerable<string> value;
return headers.TryGetValues(headerKey, out value);
return headers.TryGetValues(requestId.RequestIdKey, out value);
}
private void AddRequestIdHeader(RequestId requestId, HttpRequestMessage httpRequestMessage)
{
httpRequestMessage.Headers.Add(requestId.RequestIdKey, requestId.RequestIdValue);
}
}
}