src/Library/Consul.AspNetCore/AgentServiceRegistrationHostedService.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using System;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Hosting.Server;
using Microsoft.AspNetCore.Hosting.Server.Features;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;

namespace Consul.AspNetCore
{
    public class AgentServiceRegistrationHostedService : IHostedService
    {
        private readonly IConsulClient _consulClient;
        private readonly ILogger<AgentServiceRegistrationHostedService> _logger;
        private readonly AgentServiceRegistration _serviceRegistration;
        private Timer _timer;
        internal static int PORT = 0;
        private static bool _registered = false;

        public AgentServiceRegistrationHostedService(
            IConsulClient consulClient,
            ILogger<AgentServiceRegistrationHostedService> logger,
            AgentServiceRegistration serviceRegistration)
        {
            _consulClient = consulClient;
            _logger = logger;
            _serviceRegistration = serviceRegistration;

        }

        public Task StartAsync(CancellationToken cancellationToken)
        {
            _timer = new Timer(async state => await Execute(cancellationToken), null, dueTime: TimeSpan.FromSeconds(3), period: TimeSpan.FromSeconds(5));
            return Task.CompletedTask;
        }

        private async Task Execute(CancellationToken cancellationToken)
        {
            try
            {
                if (PORT != 0 && !_registered)
                {
                    _serviceRegistration.Port = PORT;
                    await _consulClient.Agent.ServiceRegister(_serviceRegistration, cancellationToken);
                    _registered = true;
                }
                else if (_registered)
                {
                    await _consulClient.Agent.PassTTL((_serviceRegistration.Check as AgentCheckRegistration_1_7).CheckID, $"{DateTimeOffset.UtcNow} was checked");
                    _logger.LogInformation($"Consul is executing passTTL for a health report");
                }
            }
            catch (Exception ex)
            {
                _serviceRegistration.Port = PORT;
                await _consulClient.Agent.ServiceRegister(_serviceRegistration, cancellationToken);
                _logger.LogError(ex, $"Consul failed to execute health report");
            }