51 lines
2.1 KiB
C#
51 lines
2.1 KiB
C#
|
using System.Net.Sockets;
|
||
|
|
using JiShe.CollectBus.Cassandra;
|
||
|
|
using JiShe.CollectBus.IoTDB.Interface;
|
||
|
|
using JiShe.CollectBus.IoTDB.Options;
|
||
|
|
using JiShe.CollectBus.IoTDB.Provider;
|
||
|
|
using Microsoft.Extensions.Diagnostics.HealthChecks;
|
||
|
|
|
||
|
|
namespace JiShe.CollectBus.Host.HealthChecks
|
||
|
|
{
|
||
|
|
/// <summary>
|
||
|
|
/// IoTDBHealthCheck
|
||
|
|
/// </summary>
|
||
|
|
/// <seealso cref="Microsoft.Extensions.Diagnostics.HealthChecks.IHealthCheck" />
|
||
|
|
public class IoTdbHealthCheck : IHealthCheck
|
||
|
|
{
|
||
|
|
private readonly IConfiguration _configuration;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Initializes a new instance of the <see cref="IoTdbHealthCheck"/> class.
|
||
|
|
/// </summary>
|
||
|
|
/// <param name="configuration">The configuration.</param>
|
||
|
|
public IoTdbHealthCheck(IConfiguration configuration)
|
||
|
|
{
|
||
|
|
_configuration = configuration;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Runs the health check, returning the status of the component being checked.
|
||
|
|
/// </summary>
|
||
|
|
/// <param name="context">A context object associated with the current execution.</param>
|
||
|
|
/// <param name="cancellationToken">A <see cref="T:System.Threading.CancellationToken" /> that can be used to cancel the health check.</param>
|
||
|
|
/// <returns>
|
||
|
|
/// A <see cref="T:System.Threading.Tasks.Task`1" /> that completes when the health check has finished, yielding the status of the component being checked.
|
||
|
|
/// </returns>
|
||
|
|
public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
var ioTDbOptions = new IoTDbOptions();
|
||
|
|
_configuration.GetSection("IoTDBOptions").Bind(ioTDbOptions);
|
||
|
|
var pool = new SessionPoolAdapter(ioTDbOptions);
|
||
|
|
await pool.OpenAsync();
|
||
|
|
return HealthCheckResult.Healthy($"IoTDB is healthy.");
|
||
|
|
}
|
||
|
|
catch (Exception ex)
|
||
|
|
{
|
||
|
|
return new HealthCheckResult(context.Registration.FailureStatus, $"IoTDB不健康: {ex.Message}", ex);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|