109 lines
5.1 KiB
C#
109 lines
5.1 KiB
C#
using Microsoft.EntityFrameworkCore;
|
||
using Microsoft.Extensions.Logging;
|
||
using PARR.BLL.Domain.Mq;
|
||
using PARR.Constants;
|
||
using PARR.DAL.NextRunServices;
|
||
using PARR.DAL.Services.Interfaces;
|
||
using PARR.Domain.Enums;
|
||
|
||
namespace PARR.TemplateDistributor
|
||
{
|
||
internal class TemplateDistributor : ITemplateDistributor
|
||
{
|
||
private readonly ILogger<TemplateDistributor> logger;
|
||
//private readonly INextRunService nextRunService;
|
||
private readonly INextRunService nextRunService;
|
||
private readonly ITemplateService templateService;
|
||
private readonly IRobotConfigurationService robotConfigurationService;
|
||
|
||
public TemplateDistributor(
|
||
ILogger<TemplateDistributor> logger,
|
||
//INextRunService nextRunService,
|
||
INextRunService nextRunService,
|
||
ITemplateService templateService,
|
||
IRobotConfigurationService robotConfigurationService
|
||
)
|
||
{
|
||
this.logger = logger;
|
||
this.nextRunService = nextRunService;
|
||
this.templateService = templateService;
|
||
this.robotConfigurationService = robotConfigurationService;
|
||
}
|
||
|
||
|
||
public async Task DistributeAsync(TemplateDistributorMq mqResponse)
|
||
{
|
||
var jobGroupId = mqResponse.JobGroupId;
|
||
// распределяем шаблоны только в статусе Used
|
||
var templateStatusType = TemplateStatusTypeEnum.Used;
|
||
|
||
// вызвать метод распределения, и получить новые даты
|
||
var distributedTemplates = await nextRunService.GetNextRunForJobGroupWithAutoDistributionAsync(jobGroupId, templateStatusType);
|
||
|
||
if (distributedTemplates == null)
|
||
{
|
||
logger.LogError("При распределении шаблонов по jobGroupId {jobGroupId} вернулся null. Это ошибка. Прекращаю распределение.", jobGroupId);
|
||
return;
|
||
}
|
||
|
||
// получить список шаблонов, сравнить их с распределенными, обновить даты, сохранить
|
||
var groupTemplates = await templateService.Get()
|
||
.Include(t => t.RobotConfigurations)
|
||
.Where(t => t.Job!.GroupId == jobGroupId && t.StatusTypeId == templateStatusType)
|
||
.ToListAsync();
|
||
|
||
if (!groupTemplates.Any())
|
||
{
|
||
logger.LogWarning("Для jobGroupId {jobGroupId} не найдено ни одного шаблона в БД. Прекращаю обновление.", jobGroupId);
|
||
return;
|
||
}
|
||
|
||
// создать словарь для быстрого поиска
|
||
var distributedDict = distributedTemplates.ToDictionary(t => t.Id);
|
||
|
||
int updatedCount = 0;
|
||
|
||
foreach (var templateDb in groupTemplates)
|
||
{
|
||
if (distributedDict.TryGetValue(templateDb.Id, out var distributedTemplate))
|
||
{
|
||
if (templateDb.NextRun != distributedTemplate.NextRun)
|
||
{
|
||
templateDb.NextRun = distributedTemplate.NextRun;
|
||
templateDb.LastRun = distributedTemplate.NextRunOld;
|
||
templateDb.DateModified = DateTimeOffset.UtcNow;
|
||
|
||
// ставим задание роботу - обновить расписания
|
||
var config = robotConfigurationService.GetFromTemplateByRobotCode(RobotsEnum.ScheduleOrder, templateDb);
|
||
//robotConfigurationService.ChangeTaskStatus(TaskStatusEnum.Updating, config);
|
||
robotConfigurationService.SetUpdateTaskStatusIfAllow(config);
|
||
|
||
updatedCount++;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
logger.LogWarning("Шаблон с Id {templateId} не найден в распределённых данных.", templateDb.Id);
|
||
}
|
||
}
|
||
|
||
if (updatedCount > 0)
|
||
{
|
||
// сохранить изменения
|
||
if (await templateService.CommitAsync(mqResponse.Initiator))
|
||
{
|
||
logger.LogInformation("Обновлено {updatedCount} шаблонов в БД для jobGroupId {jobGroupId}. Для расписаний установлен статус: {taskStatus}", updatedCount, jobGroupId, TaskStatusEnum.Updating.ToString());
|
||
}
|
||
else
|
||
{
|
||
logger.LogError("Ошибка при обновлении записей в БД. jobGroupId {jobGroupId}, требовалось обновить шаблонов: {updatedCount}", jobGroupId, updatedCount);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
logger.LogInformation("Для jobGroupId {jobGroupId} не найдено изменений. Ничего не обновлено.", jobGroupId);
|
||
}
|
||
}
|
||
}
|
||
}
|