forked from ddrilling/AsbCloudServer
106 lines
3.3 KiB
C#
106 lines
3.3 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
|
||
namespace AsbCloudInfrastructure.Background;
|
||
|
||
/// <summary>
|
||
/// <para>
|
||
/// Очередь работ
|
||
/// </para>
|
||
/// Не периодические задачи будут возвращаться первыми, как самые приоритетные.
|
||
/// </summary>
|
||
public class WorkStore
|
||
{
|
||
private readonly List<WorkPeriodic> periodics = new(8);
|
||
|
||
/// <summary>
|
||
/// Список периодических задач
|
||
/// </summary>
|
||
public IEnumerable<WorkPeriodic> Periodics => periodics;
|
||
|
||
/// <summary>
|
||
/// Работы выполняемые один раз
|
||
/// </summary>
|
||
public Queue<Work> RunOnceQueue { get; private set; } = new(8);
|
||
|
||
/// <summary>
|
||
/// Завершившиеся с ошибкой
|
||
/// </summary>
|
||
public CyclycArray<Work> Felled { get; } = new(16);
|
||
|
||
/// <summary>
|
||
/// Добавить фоновую работу выполняющуюся с заданным периодом
|
||
/// </summary>
|
||
/// <typeparam name="T"></typeparam>
|
||
/// <param name="period"></param>
|
||
public void AddPeriodic<T>(TimeSpan period)
|
||
where T : Work, new()
|
||
{
|
||
var work = new T();
|
||
var periodic = new WorkPeriodic(work, period);
|
||
periodics.Add(periodic);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Добавить фоновую работу выполняющуюся с заданным периодом
|
||
/// </summary>
|
||
/// <param name="work"></param>
|
||
/// <param name="period"></param>
|
||
public void AddPeriodic(Work work, TimeSpan period)
|
||
{
|
||
var periodic = new WorkPeriodic(work, period);
|
||
periodics.Add(periodic);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Удаление работы по ID из одноразовой очереди
|
||
/// </summary>
|
||
/// <param name="id"></param>
|
||
/// <returns></returns>
|
||
public bool TryRemoveFromRunOnceQueue(string id)
|
||
{
|
||
var work = RunOnceQueue.FirstOrDefault(w => w.Id == id);
|
||
if (work is not null)
|
||
{
|
||
RunOnceQueue = new Queue<Work>(RunOnceQueue.Where(w => w.Id != id));
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
/// <summary>
|
||
/// <para>
|
||
/// Возвращает приоритетную задачу.
|
||
/// </para>
|
||
/// <para>
|
||
/// Если приоритетные закончились, то ищет ближайшую периодическую.
|
||
/// Если до старта ближайшей периодической работы меньше 20 сек,
|
||
/// то этой задаче устанавливается время последнего запуска в now и она возвращается.
|
||
/// Если больше 20 сек, то возвращается null.
|
||
/// </para>
|
||
/// </summary>
|
||
/// <param name="maxTimeToNextWork"></param>
|
||
/// <returns></returns>
|
||
public Work? GetNext()
|
||
{
|
||
if (RunOnceQueue.Any())
|
||
return RunOnceQueue.Dequeue();
|
||
|
||
var work = GetNextPeriodic();
|
||
if (work is null || work.NextStart > DateTime.Now)
|
||
return null;
|
||
|
||
return work.Work;
|
||
}
|
||
|
||
private WorkPeriodic? GetNextPeriodic()
|
||
{
|
||
var work = Periodics
|
||
.OrderBy(w => w.NextStart)
|
||
.FirstOrDefault();
|
||
return work;
|
||
}
|
||
}
|