wabbajack/Wabbajack.Server/Services/AbstractService.cs

122 lines
3.3 KiB
C#
Raw Normal View History

2020-05-13 21:52:34 +00:00
using System;
using System.Linq;
2020-05-13 21:52:34 +00:00
using System.Threading.Tasks;
using Microsoft.AspNetCore.Builder;
using Microsoft.Extensions.Logging;
using Wabbajack.BuildServer;
namespace Wabbajack.Server.Services
{
public interface IStartable
{
public void Start();
}
public interface IReportingService
{
public TimeSpan Delay { get; }
public DateTime LastStart { get; }
public DateTime LastEnd { get; }
public (String, DateTime)[] ActiveWorkStatus { get; }
}
2020-05-13 21:52:34 +00:00
public abstract class AbstractService<TP, TR> : IStartable, IReportingService
2020-05-13 21:52:34 +00:00
{
protected AppSettings _settings;
private TimeSpan _delay;
protected ILogger<TP> _logger;
protected QuickSync _quickSync;
2020-05-13 21:52:34 +00:00
public TimeSpan Delay => _delay;
public DateTime LastStart { get; private set; }
public DateTime LastEnd { get; private set; }
public (String, DateTime)[] ActiveWorkStatus { get; private set; }= { };
public AbstractService(ILogger<TP> logger, AppSettings settings, QuickSync quickSync, TimeSpan delay)
2020-05-13 21:52:34 +00:00
{
_settings = settings;
_delay = delay;
_logger = logger;
_quickSync = quickSync;
}
public virtual async Task Setup()
{
2020-05-13 21:52:34 +00:00
}
public void Start()
{
2020-05-13 21:52:34 +00:00
if (_settings.RunBackEndJobs)
{
Task.Run(async () =>
{
await Setup();
await _quickSync.Register(this);
2020-05-13 21:52:34 +00:00
while (true)
{
await _quickSync.ResetToken<TP>();
2020-05-13 21:52:34 +00:00
try
{
_logger.LogInformation($"Running: {GetType().Name}");
LastStart = DateTime.UtcNow;
2020-05-13 21:52:34 +00:00
await Execute();
LastEnd = DateTime.UtcNow;
2020-05-13 21:52:34 +00:00
}
catch (Exception ex)
{
_logger.LogError(ex, "Running Service Loop");
}
var token = await _quickSync.GetToken<TP>();
2020-06-08 20:42:59 +00:00
try
{
await Task.Delay(_delay, token);
}
catch (TaskCanceledException)
{
}
2020-05-13 21:52:34 +00:00
}
});
}
}
public abstract Task<TR> Execute();
protected void ReportStarting(string value)
{
lock (this)
{
ActiveWorkStatus = ActiveWorkStatus.Cons((value, DateTime.UtcNow)).ToArray();
}
}
protected void ReportEnding(string value)
{
lock (this)
{
ActiveWorkStatus = ActiveWorkStatus.Where(x => x.Item1 != value).ToArray();
}
}
2020-05-13 21:52:34 +00:00
}
public static class AbstractServiceExtensions
{
public static void UseService<T>(this IApplicationBuilder b)
{
var poll = (IStartable)b.ApplicationServices.GetService(typeof(T));
poll.Start();
}
}
2020-05-13 21:52:34 +00:00
}