57 lines
2.0 KiB
C#
57 lines
2.0 KiB
C#
using DysonNetwork.Shared.Registry;
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Microsoft.Extensions.Logging;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace DysonNetwork.Pusher.Startup;
|
|
|
|
public class ServiceRegistrationHostedService : IHostedService
|
|
{
|
|
private readonly ServiceRegistry _serviceRegistry;
|
|
private readonly IConfiguration _configuration;
|
|
private readonly ILogger<ServiceRegistrationHostedService> _logger;
|
|
|
|
public ServiceRegistrationHostedService(
|
|
ServiceRegistry serviceRegistry,
|
|
IConfiguration configuration,
|
|
ILogger<ServiceRegistrationHostedService> logger)
|
|
{
|
|
_serviceRegistry = serviceRegistry;
|
|
_configuration = configuration;
|
|
_logger = logger;
|
|
}
|
|
|
|
public async Task StartAsync(CancellationToken cancellationToken)
|
|
{
|
|
var serviceName = "DysonNetwork.Pusher"; // Preset service name
|
|
var serviceUrl = _configuration["Service:Url"];
|
|
|
|
if (string.IsNullOrEmpty(serviceUrl))
|
|
{
|
|
_logger.LogWarning("Service URL not configured. Skipping Etcd registration.");
|
|
return;
|
|
}
|
|
|
|
_logger.LogInformation("Registering service {ServiceName} at {ServiceUrl} with Etcd.", serviceName, serviceUrl);
|
|
try
|
|
{
|
|
await _serviceRegistry.RegisterService(serviceName, serviceUrl);
|
|
_logger.LogInformation("Service {ServiceName} registered successfully.", serviceName);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Failed to register service {ServiceName} with Etcd.", serviceName);
|
|
}
|
|
}
|
|
|
|
public Task StopAsync(CancellationToken cancellationToken)
|
|
{
|
|
// The lease will expire automatically if the service stops.
|
|
// For explicit unregistration, you would implement it here.
|
|
_logger.LogInformation("Service registration hosted service is stopping.");
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|