forked from ddrilling/AsbCloudServer
79 lines
2.5 KiB
C#
79 lines
2.5 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.Hosting;
|
||
using System;
|
||
using System.Diagnostics;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace AsbCloudInfrastructure.Background
|
||
{
|
||
/// <summary>
|
||
/// Сервис для фонового выполнения работы
|
||
/// </summary>
|
||
public class BackgroundWorker : BackgroundService
|
||
{
|
||
private static readonly TimeSpan executePeriod = TimeSpan.FromSeconds(10);
|
||
private static readonly TimeSpan minDelay = TimeSpan.FromSeconds(2);
|
||
private readonly IServiceProvider serviceProvider;
|
||
private readonly WorkStore workQueue = new WorkStore();
|
||
public string? CurrentWorkId;
|
||
|
||
public BackgroundWorker(IServiceProvider serviceProvider)
|
||
{
|
||
this.serviceProvider = serviceProvider;
|
||
}
|
||
|
||
/// <summary>
|
||
/// Добавление задачи в очередь.
|
||
/// Не периодические задачи будут выполняться вперед.
|
||
/// </summary>
|
||
/// <param name="work"></param>
|
||
/// <exception cref="ArgumentException">Id mast be unique</exception>
|
||
public void Push(WorkBase work)
|
||
{
|
||
workQueue.Push(work);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Проверяет наличие работы с указанным Id
|
||
/// </summary>
|
||
/// <param name="id"></param>
|
||
/// <returns></returns>
|
||
public bool Contains(string id)
|
||
{
|
||
return workQueue.Contains(id);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Удаление работы по ID
|
||
/// </summary>
|
||
/// <param name="id"></param>
|
||
/// <returns></returns>
|
||
public bool Delete(string id)
|
||
{
|
||
return workQueue.Delete(id);
|
||
}
|
||
|
||
protected override async Task ExecuteAsync(CancellationToken token)
|
||
{
|
||
while (!token.IsCancellationRequested)
|
||
{
|
||
var work = workQueue.Pop();
|
||
if (work is null)
|
||
{
|
||
await Task.Delay(executePeriod, token);
|
||
continue;
|
||
}
|
||
|
||
CurrentWorkId = work.Id;
|
||
using var scope = serviceProvider.CreateScope();
|
||
|
||
await work.Start(scope.ServiceProvider, token);
|
||
|
||
CurrentWorkId = null;
|
||
await Task.Delay(minDelay, token);
|
||
}
|
||
}
|
||
}
|
||
}
|