Use custom database healthcheck (#8973)

This commit is contained in:
Cody Robibero 2022-12-29 19:40:39 -07:00 committed by GitHub
parent 1f658f59b8
commit 8326349137
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 45 additions and 1 deletions

View File

@ -0,0 +1,43 @@
using System;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Diagnostics.HealthChecks;
namespace Jellyfin.Server.HealthChecks;
/// <summary>
/// Implementation of the <see cref="DbContextHealthCheck{TContext}"/> for a <see cref="IDbContextFactory{TContext}"/>.
/// </summary>
/// <typeparam name="TContext">The type of database context.</typeparam>
public class DbContextFactoryHealthCheck<TContext> : IHealthCheck
where TContext : DbContext
{
private readonly IDbContextFactory<TContext> _dbContextFactory;
/// <summary>
/// Initializes a new instance of the <see cref="DbContextFactoryHealthCheck{TContext}"/> class.
/// </summary>
/// <param name="contextFactory">Instance of the <see cref="IDbContextFactory{TContext}"/> interface.</param>
public DbContextFactoryHealthCheck(IDbContextFactory<TContext> contextFactory)
{
_dbContextFactory = contextFactory;
}
/// <inheritdoc />
public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
{
ArgumentNullException.ThrowIfNull(context);
var dbContext = await _dbContextFactory.CreateDbContextAsync(cancellationToken).ConfigureAwait(false);
await using (dbContext.ConfigureAwait(false))
{
if (await dbContext.Database.CanConnectAsync(cancellationToken).ConfigureAwait(false))
{
return HealthCheckResult.Healthy();
}
}
return HealthCheckResult.Unhealthy();
}
}

View File

@ -8,6 +8,7 @@ using System.Text;
using Jellyfin.MediaEncoding.Hls.Extensions;
using Jellyfin.Networking.Configuration;
using Jellyfin.Server.Extensions;
using Jellyfin.Server.HealthChecks;
using Jellyfin.Server.Implementations;
using Jellyfin.Server.Implementations.Extensions;
using Jellyfin.Server.Infrastructure;
@ -122,7 +123,7 @@ namespace Jellyfin.Server
.ConfigurePrimaryHttpMessageHandler(defaultHttpClientHandlerDelegate);
services.AddHealthChecks()
.AddDbContextCheck<JellyfinDb>();
.AddCheck<DbContextFactoryHealthCheck<JellyfinDb>>(nameof(JellyfinDb));
services.AddHlsPlaylistGenerator();
}