37 lines
1.3 KiB
C#
37 lines
1.3 KiB
C#
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Jellyfin.Server.Implementations;
|
|
using Microsoft.Extensions.Diagnostics.HealthChecks;
|
|
|
|
namespace Jellyfin.Server.HealthChecks
|
|
{
|
|
/// <summary>
|
|
/// Checks connectivity to the database.
|
|
/// </summary>
|
|
public class JellyfinDbHealthCheck : IHealthCheck
|
|
{
|
|
private readonly JellyfinDbProvider _dbProvider;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="JellyfinDbHealthCheck"/> class.
|
|
/// </summary>
|
|
/// <param name="dbProvider">The jellyfin db provider.</param>
|
|
public JellyfinDbHealthCheck(JellyfinDbProvider dbProvider)
|
|
{
|
|
_dbProvider = dbProvider;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
|
|
{
|
|
await using var jellyfinDb = _dbProvider.CreateContext();
|
|
if (await jellyfinDb.Database.CanConnectAsync(cancellationToken).ConfigureAwait(false))
|
|
{
|
|
return HealthCheckResult.Healthy("Database connection successful.");
|
|
}
|
|
|
|
return HealthCheckResult.Unhealthy("Unable to connect to the database.");
|
|
}
|
|
}
|
|
}
|