97 lines
2.9 KiB
C#
97 lines
2.9 KiB
C#
using DotBased.Logging;
|
|
using Manager.App.Services.System;
|
|
using ILogger = Microsoft.Extensions.Logging.ILogger;
|
|
|
|
namespace Manager.App.Services;
|
|
|
|
public abstract class ExtendedBackgroundService : BackgroundService
|
|
{
|
|
private readonly ManualResetEventSlim _resetEvent = new(true);
|
|
private readonly ILogger _logger;
|
|
public ServiceState State { get; private set; } = ServiceState.Stopped;
|
|
public CircularBuffer<ServiceProgress> ProgressLog { get; } = new(500);
|
|
public string Name { get; }
|
|
public TimeSpan ExecuteInterval { get; set; }
|
|
|
|
public ExtendedBackgroundService(string name, ILogger logger, BackgroundServiceManager manager, TimeSpan? executeInterval = null)
|
|
{
|
|
Name = name;
|
|
_logger = logger;
|
|
manager.RegisterService(this);
|
|
ExecuteInterval = executeInterval ?? TimeSpan.Zero;
|
|
}
|
|
|
|
protected sealed override async Task ExecuteAsync(CancellationToken stoppingToken)
|
|
{
|
|
State = ServiceState.Running;
|
|
_logger.LogInformation("Starting background service: {ServiceName}", Name);
|
|
|
|
try
|
|
{
|
|
while (!stoppingToken.IsCancellationRequested)
|
|
{
|
|
_resetEvent.Wait(stoppingToken);
|
|
await Task.Delay(ExecuteInterval, stoppingToken);
|
|
await ExecuteServiceAsync(stoppingToken);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
if (e is not OperationCanceledException)
|
|
{
|
|
State = ServiceState.Faulted;
|
|
_logger.LogError(e,"Background service {ServiceName} faulted!", Name);
|
|
throw;
|
|
}
|
|
_logger.LogInformation(e,"Service {ServiceName} received cancellation", Name);
|
|
}
|
|
finally
|
|
{
|
|
State = ServiceState.Stopped;
|
|
}
|
|
}
|
|
|
|
protected void LogProgress(string message, LogSeverity severity = LogSeverity.Info) => ProgressLog.Add(new ServiceProgress(message, DateTime.UtcNow, severity));
|
|
|
|
public void Pause()
|
|
{
|
|
if (State == ServiceState.Running)
|
|
{
|
|
_resetEvent.Reset();
|
|
State = ServiceState.Paused;
|
|
_logger.LogInformation("Pauses service: {ServiceName}", Name);
|
|
}
|
|
}
|
|
|
|
public void Resume()
|
|
{
|
|
if (State == ServiceState.Paused)
|
|
{
|
|
_resetEvent.Set();
|
|
State = ServiceState.Running;
|
|
_logger.LogInformation("Resumes service: {ServiceName}", Name);
|
|
}
|
|
}
|
|
|
|
protected abstract Task ExecuteServiceAsync(CancellationToken stoppingToken);
|
|
|
|
public override bool Equals(object? obj)
|
|
{
|
|
return obj is ExtendedBackgroundService bgService && bgService.Name.Equals(Name, StringComparison.OrdinalIgnoreCase);
|
|
}
|
|
|
|
public override int GetHashCode()
|
|
{
|
|
return Name.GetHashCode();
|
|
}
|
|
}
|
|
|
|
public enum ServiceState
|
|
{
|
|
Stopped,
|
|
Faulted,
|
|
Running,
|
|
Paused
|
|
}
|
|
|
|
public record ServiceProgress(string Message, DateTime StartTime, LogSeverity Severity); |