public async Task Invoke(HttpContext context, IBeatPulseService pulseService) { var request = context.Request; if (!IsBeatPulseRequest(request, out string beatPulsePath)) { await _next.Invoke(context); return; } else { if (TryFromCache(beatPulsePath, out OutputMessage output)) { await WriteResponseAsync(request.HttpContext, output, _options.DetailedOutput); } else { output = new OutputMessage(); using (var cancellationTokenSource = new CancellationTokenSource()) { var task = pulseService.IsHealthy(beatPulsePath, context, cancellationTokenSource.Token); if (await Task.WhenAny(task, Task.Delay(_options.Timeout, cancellationTokenSource.Token)) == task) { var responses = await task; if (!responses.Any()) { await WriteNotFoundAsync(request.HttpContext, _options.DetailedOutput); return; } output.AddHealthCheckMessages(responses); output.EndAtUtc = DateTime.UtcNow; if (_options.CacheMode.UseServerMemory() && _options.CacheOutput) { _cache.TryAdd(beatPulsePath, output); } await WriteResponseAsync(request.HttpContext, output, _options.DetailedOutput); } else { //timeout _cache.TryRemove(beatPulsePath, out OutputMessage removed); cancellationTokenSource.Cancel(); await WriteTimeoutAsync(request.HttpContext, _options.DetailedOutput); } } } } }