diff --git a/src/Ocelot/Infrastructure/Placeholders.cs b/src/Ocelot/Infrastructure/Placeholders.cs index 3c319e6d..58b4c5d4 100644 --- a/src/Ocelot/Infrastructure/Placeholders.cs +++ b/src/Ocelot/Infrastructure/Placeholders.cs @@ -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> @@ -130,5 +132,27 @@ namespace Ocelot.Infrastructure { return () => new OkResponse(_finder.Find()); } + + private Func> GetUpstreamHost() + { + return () => + { + try + { + if (_httpContextAccessor.HttpContext.Request.Headers.TryGetValue("Host", out var upstreamHost)) + { + return new OkResponse(upstreamHost.First()); + } + else + { + return new ErrorResponse(new CouldNotFindPlaceholderError("{UpstreamHost}")); + } + } + catch + { + return new ErrorResponse(new CouldNotFindPlaceholderError("{UpstreamHost}")); + } + }; + } } } diff --git a/test/Ocelot.UnitTests/Infrastructure/PlaceholdersTests.cs b/test/Ocelot.UnitTests/Infrastructure/PlaceholdersTests.cs index ccd48902..d12f4ea7 100644 --- a/test/Ocelot.UnitTests/Infrastructure/PlaceholdersTests.cs +++ b/test/Ocelot.UnitTests/Infrastructure/PlaceholdersTests.cs @@ -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); + } } }