mirror of
https://github.com/anjoy8/Blog.Core.git
synced 2024-09-20 23:48:27 +08:00
71 lines
2.7 KiB
C#
71 lines
2.7 KiB
C#
|
using System;
|
|||
|
using System.Threading;
|
|||
|
using System.Threading.Tasks;
|
|||
|
using Consul;
|
|||
|
using Microsoft.Extensions.Configuration;
|
|||
|
using Microsoft.Extensions.Hosting;
|
|||
|
using Microsoft.Extensions.Logging;
|
|||
|
|
|||
|
namespace Blog.Core.Extensions.HostedService;
|
|||
|
|
|||
|
public class ConsulHostedService : IHostedService
|
|||
|
{
|
|||
|
private readonly IConfiguration _configuration;
|
|||
|
private readonly IHostApplicationLifetime _hostApplicationLifetime;
|
|||
|
private readonly ILogger<ConsulHostedService> _logger;
|
|||
|
|
|||
|
public ConsulHostedService(IConfiguration configuration, IHostApplicationLifetime hostApplicationLifetime, ILogger<ConsulHostedService> logger)
|
|||
|
{
|
|||
|
_configuration = configuration;
|
|||
|
_hostApplicationLifetime = hostApplicationLifetime;
|
|||
|
_logger = logger;
|
|||
|
}
|
|||
|
|
|||
|
public async Task StartAsync(CancellationToken cancellationToken)
|
|||
|
{
|
|||
|
_logger.LogInformation("Start Consul Service!");
|
|||
|
await DoWork();
|
|||
|
}
|
|||
|
|
|||
|
public async Task DoWork()
|
|||
|
{
|
|||
|
if (_configuration["Middleware:Consul:Enabled"].ObjToBool())
|
|||
|
{
|
|||
|
var consulClient = new ConsulClient(c =>
|
|||
|
{
|
|||
|
//consul地址
|
|||
|
c.Address = new Uri(_configuration["ConsulSetting:ConsulAddress"]);
|
|||
|
});
|
|||
|
|
|||
|
var registration = new AgentServiceRegistration()
|
|||
|
{
|
|||
|
ID = Guid.NewGuid().ToString(),//服务实例唯一标识
|
|||
|
Name = _configuration["ConsulSetting:ServiceName"],//服务名
|
|||
|
Address = _configuration["ConsulSetting:ServiceIP"], //服务IP
|
|||
|
Port = int.Parse(_configuration["ConsulSetting:ServicePort"]),//服务端口
|
|||
|
Check = new AgentServiceCheck()
|
|||
|
{
|
|||
|
DeregisterCriticalServiceAfter = TimeSpan.FromSeconds(5),//服务启动多久后注册
|
|||
|
Interval = TimeSpan.FromSeconds(10),//健康检查时间间隔
|
|||
|
HTTP = $"http://{_configuration["ConsulSetting:ServiceIP"]}:{_configuration["ConsulSetting:ServicePort"]}{_configuration["ConsulSetting:ServiceHealthCheck"]}",//健康检查地址
|
|||
|
Timeout = TimeSpan.FromSeconds(5)//超时时间
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
//服务注册
|
|||
|
await consulClient.Agent.ServiceRegister(registration);
|
|||
|
|
|||
|
//应用程序终止时,取消注册
|
|||
|
_hostApplicationLifetime.ApplicationStopping.Register(async () =>
|
|||
|
{
|
|||
|
await consulClient.Agent.ServiceDeregister(registration.ID);
|
|||
|
});
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public Task StopAsync(CancellationToken cancellationToken)
|
|||
|
{
|
|||
|
_logger.LogInformation("Stop Consul Service!");
|
|||
|
return Task.CompletedTask;
|
|||
|
}
|
|||
|
}
|