public void IsWarmUpRequest_ReturnsExpectedValue()
{
var request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup");
Assert.False(StandbyManager.IsWarmUpRequest(request));
var vars = new Dictionary <string, string>
{
{ EnvironmentSettingNames.AzureWebsitePlaceholderMode, "0" },
{ EnvironmentSettingNames.AzureWebsiteInstanceId, null }
};
using (var env = new TestScopedEnvironmentVariable(vars))
{
// in this test we're forcing a transition from non-placeholder mode to placeholder mode
// which can't happen in the wild, so we force a reset here
WebScriptHostManager.ResetStandbyMode();
_settingsManager.SetSetting(EnvironmentSettingNames.AzureWebsitePlaceholderMode, "1");
Assert.False(StandbyManager.IsWarmUpRequest(request));
_settingsManager.SetSetting(EnvironmentSettingNames.AzureWebsiteInstanceId, "12345");
Assert.True(StandbyManager.IsWarmUpRequest(request));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/csharphttpwarmup");
Assert.True(StandbyManager.IsWarmUpRequest(request));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup");
request.Headers.Add(ScriptConstants.AntaresLogIdHeaderName, "xyz123");
Assert.False(StandbyManager.IsWarmUpRequest(request));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/foo");
Assert.False(StandbyManager.IsWarmUpRequest(request));
}
vars = new Dictionary <string, string>
{
{ EnvironmentSettingNames.AzureWebsitePlaceholderMode, "0" },
{ EnvironmentSettingNames.AzureWebsiteInstanceId, null }
};
using (var env = new TestScopedEnvironmentVariable(vars))
{
WebScriptHostManager.ResetStandbyMode();
_settingsManager.SetSetting(EnvironmentSettingNames.AzureWebsitePlaceholderMode, "1");
Assert.False(StandbyManager.IsWarmUpRequest(request));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup");
_settingsManager.SetSetting(EnvironmentSettingNames.ContainerName, "TestContainer");
Assert.True(_settingsManager.IsLinuxContainerEnvironment);
Assert.True(StandbyManager.IsWarmUpRequest(request));
}
}
public async Task Invoke(HttpContext httpContext)
{
if (StandbyManager.IsWarmUpRequest(httpContext.Request))
{
if (!httpContext.Items.TryGetValue(ScriptConstants.AzureFunctionsHostManagerKey, out object scriptHostManager))
{
throw new InvalidOperationException($"Warmup request received, but no instance of {nameof(WebScriptHostManager)} in HTTP context");
}
await StandbyManager.WarmUp(httpContext.Request, (WebScriptHostManager)scriptHostManager);
}
await _next.Invoke(httpContext);
}
protected override async Task <HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
{
SetRequestId(request);
var resolver = _config.DependencyResolver;
var scriptHostManager = resolver.GetService <WebScriptHostManager>();
if (!scriptHostManager.Initialized)
{
scriptHostManager.Initialize();
}
var webHostSettings = resolver.GetService <WebHostSettings>();
if (webHostSettings.IsAuthDisabled)
{
request.SetProperty(ScriptConstants.AzureFunctionsHttpRequestAuthorizationDisabledKey, true);
}
// some routes do not require the host to be running (most do)
bool bypassHostCheck = request.MatchRoute("admin/host/status");
if (!bypassHostCheck)
{
// If the host is not running, we'll wait a bit for it to fully
// initialize. This might happen if http requests come in while the
// host is starting up for the first time, or if it is restarting.
await scriptHostManager.DelayUntilHostReady(_hostTimeoutSeconds, _hostRunningPollIntervalMilliseconds);
}
if (StandbyManager.IsWarmUpRequest(request))
{
await StandbyManager.WarmUp(request, scriptHostManager);
}
return(await base.SendAsync(request, cancellationToken));
}
