mirror of https://github.com/Squidex/squidex.git
Sebastian Stehle
7 years ago
18 changed files with 85 additions and 246 deletions
@ -1,27 +0,0 @@ |
|||||
// ==========================================================================
|
|
||||
// Squidex Headless CMS
|
|
||||
// ==========================================================================
|
|
||||
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
|
||||
// All rights reserved. Licensed under the MIT license.
|
|
||||
// ==========================================================================
|
|
||||
|
|
||||
using System.Collections.Generic; |
|
||||
|
|
||||
namespace Squidex.Infrastructure.Diagnostics |
|
||||
{ |
|
||||
public sealed class HealthCheckResult |
|
||||
{ |
|
||||
public bool IsHealthy { get; } |
|
||||
|
|
||||
public string Description { get; } |
|
||||
|
|
||||
public Dictionary<string, object> Data { get; } |
|
||||
|
|
||||
public HealthCheckResult(bool isHealthy, string description = null, Dictionary<string, object> data = null) |
|
||||
{ |
|
||||
IsHealthy = isHealthy; |
|
||||
Data = data; |
|
||||
Description = description; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
@ -1,16 +0,0 @@ |
|||||
// ==========================================================================
|
|
||||
// Squidex Headless CMS
|
|
||||
// ==========================================================================
|
|
||||
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
|
||||
// All rights reserved. Licensed under the MIT license.
|
|
||||
// ==========================================================================
|
|
||||
|
|
||||
namespace Squidex.Infrastructure.Diagnostics |
|
||||
{ |
|
||||
public static class HealthCheckScopes |
|
||||
{ |
|
||||
public const string Any = "*"; |
|
||||
public const string Node = "node"; |
|
||||
public const string Cluster = "cluster"; |
|
||||
} |
|
||||
} |
|
||||
@ -1,20 +0,0 @@ |
|||||
// ==========================================================================
|
|
||||
// Squidex Headless CMS
|
|
||||
// ==========================================================================
|
|
||||
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
|
||||
// All rights reserved. Licensed under the MIT license.
|
|
||||
// ==========================================================================
|
|
||||
|
|
||||
using System.Collections.Generic; |
|
||||
using System.Threading; |
|
||||
using System.Threading.Tasks; |
|
||||
|
|
||||
namespace Squidex.Infrastructure.Diagnostics |
|
||||
{ |
|
||||
public interface IHealthCheck |
|
||||
{ |
|
||||
IEnumerable<string> Scopes { get; } |
|
||||
|
|
||||
Task<HealthCheckResult> CheckHealthAsync(CancellationToken cancellationToken = default(CancellationToken)); |
|
||||
} |
|
||||
} |
|
||||
@ -1,113 +0,0 @@ |
|||||
// ==========================================================================
|
|
||||
// Squidex Headless CMS
|
|
||||
// ==========================================================================
|
|
||||
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
|
||||
// All rights reserved. Licensed under the MIT license.
|
|
||||
// ==========================================================================
|
|
||||
|
|
||||
using System; |
|
||||
using System.Collections.Generic; |
|
||||
using System.Linq; |
|
||||
using System.Threading; |
|
||||
using System.Threading.Tasks; |
|
||||
using Microsoft.AspNetCore.Http; |
|
||||
using Squidex.Infrastructure; |
|
||||
using Squidex.Infrastructure.Diagnostics; |
|
||||
using Squidex.Infrastructure.Json; |
|
||||
|
|
||||
namespace Squidex.Pipeline.Diagnostics |
|
||||
{ |
|
||||
public sealed class HealthCheckMiddleware |
|
||||
{ |
|
||||
private const string Suffix = "HealthCheck"; |
|
||||
private static readonly TimeSpan Timeout = TimeSpan.FromSeconds(2); |
|
||||
|
|
||||
private readonly Dictionary<string, IHealthCheck> healthChecks; |
|
||||
private readonly IJsonSerializer serializer; |
|
||||
private readonly RequestDelegate next; |
|
||||
private readonly List<string> scopes; |
|
||||
|
|
||||
public HealthCheckMiddleware(IEnumerable<IHealthCheck> healthChecks, IJsonSerializer serializer, RequestDelegate next, string scopes) |
|
||||
{ |
|
||||
Guard.NotNull(healthChecks, nameof(healthChecks)); |
|
||||
Guard.NotNull(serializer, nameof(serializer)); |
|
||||
|
|
||||
this.healthChecks = healthChecks.ToDictionary(GetName); |
|
||||
this.next = next; |
|
||||
this.serializer = serializer; |
|
||||
this.scopes = SplitScopes(scopes); |
|
||||
} |
|
||||
|
|
||||
public async Task Invoke(HttpContext context) |
|
||||
{ |
|
||||
if (CanServeRequest(context.Request)) |
|
||||
{ |
|
||||
using (var cts = new CancellationTokenSource(Timeout)) |
|
||||
{ |
|
||||
var matchingChecks = healthChecks.Where(x => CanUseCheck(x.Value)); |
|
||||
|
|
||||
var results = await Task.WhenAll(matchingChecks.Select(x => MakeHealthCheckAsync(x.Key, x.Value, cts.Token))); |
|
||||
|
|
||||
context.Response.StatusCode = 200; |
|
||||
context.Response.Headers.Add("Content-Type", "application/json"); |
|
||||
|
|
||||
if (results.Any(x => !x.Result.IsHealthy)) |
|
||||
{ |
|
||||
context.Response.StatusCode = 503; |
|
||||
} |
|
||||
|
|
||||
var response = results.ToDictionary(x => x.Name, x => x.Result); |
|
||||
|
|
||||
var json = serializer.Serialize(new { status = response }); |
|
||||
|
|
||||
await context.Response.WriteAsync(json); |
|
||||
} |
|
||||
} |
|
||||
else |
|
||||
{ |
|
||||
await next(context); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private bool CanUseCheck(IHealthCheck check) |
|
||||
{ |
|
||||
return scopes.Count == 0 || check.Scopes.Intersect(scopes).Any(); |
|
||||
} |
|
||||
|
|
||||
private bool CanServeRequest(HttpRequest request) |
|
||||
{ |
|
||||
return HttpMethods.IsGet(request.Method) && (request.Path == "/" || string.IsNullOrEmpty(request.Path)); |
|
||||
} |
|
||||
|
|
||||
private static List<string> SplitScopes(string scopes) |
|
||||
{ |
|
||||
return scopes.Split(",").Where(x => x != "*").ToList(); |
|
||||
} |
|
||||
|
|
||||
private static string GetName(IHealthCheck check) |
|
||||
{ |
|
||||
var name = check.GetType().Name.ToCamelCase(); |
|
||||
|
|
||||
if (name.EndsWith(Suffix, StringComparison.OrdinalIgnoreCase)) |
|
||||
{ |
|
||||
name = name.Substring(0, name.Length - Suffix.Length); |
|
||||
} |
|
||||
|
|
||||
return name; |
|
||||
} |
|
||||
|
|
||||
private async Task<(string Name, HealthCheckResult Result)> MakeHealthCheckAsync(string name, IHealthCheck check, CancellationToken ct) |
|
||||
{ |
|
||||
try |
|
||||
{ |
|
||||
var result = await check.CheckHealthAsync(ct); |
|
||||
|
|
||||
return (name, result); |
|
||||
} |
|
||||
catch |
|
||||
{ |
|
||||
return (name, new HealthCheckResult(false)); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
Loading…
Reference in new issue