mirror of
https://github.com/nsnail/Ocelot.git
synced 2025-04-23 00:32:50 +08:00
Merge branch 'master' of https://github.com/samirsyed/Ocelot into samirsyed-master
This commit is contained in:
commit
6c20b1173f
@ -7,6 +7,7 @@ namespace Ocelot.Infrastructure
|
||||
using Ocelot.Responses;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
|
||||
public class Placeholders : IPlaceholders
|
||||
{
|
||||
@ -25,7 +26,8 @@ namespace Ocelot.Infrastructure
|
||||
{
|
||||
{ "{BaseUrl}", GetBaseUrl() },
|
||||
{ "{TraceId}", GetTraceId() },
|
||||
{ "{RemoteIpAddress}", GetRemoteIpAddress() }
|
||||
{ "{RemoteIpAddress}", GetRemoteIpAddress() },
|
||||
{ "{UpstreamHost}", GetUpstreamHost() },
|
||||
};
|
||||
|
||||
_requestPlaceholders = new Dictionary<string, Func<DownstreamRequest, string>>
|
||||
@ -130,5 +132,27 @@ namespace Ocelot.Infrastructure
|
||||
{
|
||||
return () => new OkResponse<string>(_finder.Find());
|
||||
}
|
||||
|
||||
private Func<Response<string>> GetUpstreamHost()
|
||||
{
|
||||
return () =>
|
||||
{
|
||||
try
|
||||
{
|
||||
if (_httpContextAccessor.HttpContext.Request.Headers.TryGetValue("Host", out var upstreamHost))
|
||||
{
|
||||
return new OkResponse<string>(upstreamHost.First());
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse<string>(new CouldNotFindPlaceholderError("{UpstreamHost}"));
|
||||
}
|
||||
}
|
||||
catch
|
||||
{
|
||||
return new ErrorResponse<string>(new CouldNotFindPlaceholderError("{UpstreamHost}"));
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -123,5 +123,16 @@ namespace Ocelot.UnitTests.Infrastructure
|
||||
result.IsError.ShouldBeTrue();
|
||||
result.Errors[0].Message.ShouldBe("Unable to remove placeholder: {Test}, placeholder does not exists");
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void should_return_upstreamHost()
|
||||
{
|
||||
var upstreamHost = "UpstreamHostA";
|
||||
var httpContext = new DefaultHttpContext();
|
||||
httpContext.Request.Headers.Add("Host", upstreamHost);
|
||||
_accessor.Setup(x => x.HttpContext).Returns(httpContext);
|
||||
var result = _placeholders.Get("{UpstreamHost}");
|
||||
result.Data.ShouldBe(upstreamHost);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user