490 lines
21 KiB
C#
490 lines
21 KiB
C#
using Microsoft.EntityFrameworkCore;
|
||
using Microsoft.Extensions.Logging;
|
||
using PARR.BLL.Domain.Mq;
|
||
using PARR.BLL.Services.Interfaces;
|
||
using PARR.Common.Domain;
|
||
using PARR.Constants;
|
||
using PARR.DAL.DomainServices.Interfaces;
|
||
using PARR.DAL.Models;
|
||
using PARR.DAL.Models.Job;
|
||
using PARR.DAL.Services.Interfaces;
|
||
using PARR.DAL.Services.Interfaces.Job;
|
||
using PARR.DAL.TransformServices;
|
||
using PARR.TemplateMatcher.Settings;
|
||
using System.Text.Json;
|
||
|
||
namespace PARR.TemplateMatcher
|
||
{
|
||
internal class TemplateMatcher : ITemplateMatcher
|
||
{
|
||
private const int UnusedCandidateBatchSize = 10;
|
||
private const bool DefaultUnusedTemplateState = false;
|
||
private const bool DefaultUnusedScheduleState = false;
|
||
private const bool DefaultUsedTemplateState = false;
|
||
private const bool DefaultUsedScheduleState = false;
|
||
|
||
private readonly ILogger<TemplateMatcher> logger;
|
||
private readonly IUnitFilterService unitFilterService;
|
||
private readonly MqSettings mqSettings;
|
||
private readonly IMqService mqService;
|
||
private readonly ITemplateService templateService;
|
||
private readonly IJobService jobService;
|
||
private readonly IShortcodesService shortcodesService;
|
||
private readonly IEsppScheduleTransformService esppScheduleTransformService;
|
||
|
||
public TemplateMatcher(
|
||
ILogger<TemplateMatcher> logger,
|
||
IUnitFilterService unitFilterService,
|
||
MqSettings mqSettings,
|
||
IMqService mqService,
|
||
ITemplateService templateService,
|
||
IJobService jobService,
|
||
IShortcodesService shortcodesService,
|
||
IEsppScheduleTransformService esppScheduleTransformService)
|
||
{
|
||
this.logger = logger;
|
||
this.unitFilterService = unitFilterService;
|
||
this.mqSettings = mqSettings;
|
||
this.mqService = mqService;
|
||
this.templateService = templateService;
|
||
this.jobService = jobService;
|
||
this.shortcodesService = shortcodesService;
|
||
this.esppScheduleTransformService = esppScheduleTransformService;
|
||
}
|
||
|
||
public async Task SyncTemplatesForJob(Guid jobId, HistoryInitiator initiator)
|
||
{
|
||
logger.LogDebug("Начало синхронизации шаблонов для JobId {JobId}", jobId);
|
||
|
||
var expectedUnitIds = await GetExpectedUnitIdsAsync(jobId) ?? new HashSet<Guid>();
|
||
logger.LogDebug("JobId {JobId}: найдено {Count} UnitId по фильтрам.", jobId, expectedUnitIds.Count);
|
||
|
||
var job = await GetJobWithGroupAndAutoControlAsync(jobId);
|
||
if (job == null)
|
||
{
|
||
logger.LogError("Job с Id {JobId} не найден.", jobId);
|
||
return;
|
||
}
|
||
|
||
var existingTemplates = await templateService.Get()
|
||
.Where(t => t.JobId == jobId)
|
||
.ToListAsync();
|
||
|
||
logger.LogDebug("JobId {JobId}: {Expected} ожидаемых UnitId, {Existing} существующих шаблонов.",
|
||
jobId, expectedUnitIds.Count, existingTemplates.Count);
|
||
|
||
// Обработка случая: фильтр вернул 0 UnitId → деактивировать ВСЕ шаблоны
|
||
if (!expectedUnitIds.Any())
|
||
{
|
||
if (existingTemplates.Any())
|
||
{
|
||
logger.LogInformation("Для JobId {JobId} фильтры не дали Unit'ов — будет деактивировано {Count} шаблонов.",
|
||
jobId, existingTemplates.Count);
|
||
|
||
foreach (var template in existingTemplates)
|
||
{
|
||
await DeactivateTemplateAsync(template, jobId, initiator);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
logger.LogInformation("Для JobId {JobId} нет Unit'ов по фильтрам и нет существующих шаблонов — синхронизация завершена.", jobId);
|
||
}
|
||
|
||
logger.LogInformation("Синхронизация завершена для JobId {JobId} (фильтр пуст).", jobId);
|
||
return;
|
||
}
|
||
|
||
// Деактивация шаблонов, которые вышли из фильтра
|
||
var templatesToDeactivate = existingTemplates
|
||
.Where(t => !expectedUnitIds.Contains(t.UnitId))
|
||
.ToList();
|
||
|
||
foreach (var template in templatesToDeactivate)
|
||
{
|
||
await DeactivateTemplateAsync(template, jobId, initiator);
|
||
}
|
||
|
||
// Перечитываем шаблоны после деактивации
|
||
existingTemplates = await templateService.Get()
|
||
.Where(t => t.JobId == jobId)
|
||
.ToListAsync();
|
||
|
||
var unitToTemplate = existingTemplates.ToDictionary(t => t.UnitId, t => t);
|
||
|
||
// UnitId без шаблона → попытка переиспользования или создание
|
||
var unitIdsMissingTemplates = expectedUnitIds
|
||
.Where(unitId => !unitToTemplate.ContainsKey(unitId))
|
||
.ToList();
|
||
|
||
var unitIdsToCreateFresh = new List<Guid>();
|
||
|
||
foreach (var unitId in unitIdsMissingTemplates)
|
||
{
|
||
var reused = await TryReuseOneUnusedTemplateAsync(jobId, unitId, initiator);
|
||
if (reused != null)
|
||
{
|
||
logger.LogInformation("Переиспользован шаблон {TemplateId} для UnitId {UnitId}.", reused.Id, unitId);
|
||
|
||
var expectedName = await GetNormalizedTemplateNameAsync(job, unitId);
|
||
var nextRun = await GetNextRunAsync(job); // всегда пересчитываем для нового назначения
|
||
|
||
var updateRequest = new TemplateUpdaterMq
|
||
{
|
||
TemplateId = reused.Id,
|
||
JobId = jobId,
|
||
UnitId = unitId,
|
||
Name = expectedName,
|
||
IsActiveTemplate = job.AutoControl?.InitUsedTemplateState ?? DefaultUsedTemplateState,
|
||
IsActiveSchedule = job.AutoControl?.InitUsedScheduleState ?? DefaultUsedScheduleState,
|
||
StatusTypeId = TemplateStatusTypeEnum.Used,
|
||
Initiator = initiator,
|
||
NextRun = nextRun
|
||
};
|
||
|
||
await SendTemplateUpdateMessage(updateRequest);
|
||
}
|
||
else
|
||
{
|
||
logger.LogInformation("Нет доступных Unused-шаблонов для UnitId {UnitId} → создадим новый.", unitId);
|
||
unitIdsToCreateFresh.Add(unitId);
|
||
}
|
||
}
|
||
|
||
// Обновление/реактивация шаблонов, оставшихся в фильтре
|
||
var templatesInFilter = existingTemplates
|
||
.Where(t => expectedUnitIds.Contains(t.UnitId))
|
||
.ToList();
|
||
|
||
foreach (var template in templatesInFilter)
|
||
{
|
||
var expectedName = await GetNormalizedTemplateNameAsync(job, template.UnitId);
|
||
await ReactivateOrRenameTemplateAsync(template, job, expectedName, initiator);
|
||
}
|
||
|
||
// Создание новых шаблонов
|
||
foreach (var unitId in unitIdsToCreateFresh)
|
||
{
|
||
await SendTemplateGeneratorMessageAsync(jobId, unitId, initiator);
|
||
}
|
||
|
||
logger.LogInformation("Синхронизация завершена для JobId {JobId}.", jobId);
|
||
}
|
||
|
||
public async Task UpdateTemplatesForJob(Guid jobId, HistoryInitiator initiator)
|
||
{
|
||
logger.LogDebug("Начало обновления шаблонов для JobId {JobId}", jobId);
|
||
|
||
var existingTemplates = await templateService.Get()
|
||
.Where(t => t.JobId == jobId)
|
||
.ToListAsync();
|
||
|
||
if (!existingTemplates.Any()) return;
|
||
|
||
var job = await GetJobWithGroupAndAutoControlAsync(jobId);
|
||
if (job == null) return;
|
||
|
||
var currentUnitIds = await GetExpectedUnitIdsAsync(jobId) ?? new HashSet<Guid>();
|
||
logger.LogDebug("JobId {JobId}: {Count} UnitId по текущему фильтру.", jobId, currentUnitIds.Count);
|
||
|
||
foreach (var template in existingTemplates)
|
||
{
|
||
logger.LogDebug("Обработка шаблона {TemplateId} (UnitId {UnitId}).", template.Id, template.UnitId);
|
||
|
||
bool unitStillInFilter = currentUnitIds.Contains(template.UnitId);
|
||
var expectedName = await GetNormalizedTemplateNameAsync(job, template.UnitId);
|
||
|
||
TemplateStatusTypeEnum targetStatus;
|
||
bool targetIsActiveTemplate;
|
||
bool targetIsActiveSchedule;
|
||
|
||
if (unitStillInFilter)
|
||
{
|
||
targetStatus = TemplateStatusTypeEnum.Used;
|
||
targetIsActiveTemplate = template.IsActiveTemplate;
|
||
targetIsActiveSchedule = template.IsActiveSchedule;
|
||
}
|
||
else
|
||
{
|
||
targetStatus = TemplateStatusTypeEnum.Unused;
|
||
targetIsActiveTemplate = DefaultUnusedTemplateState;
|
||
targetIsActiveSchedule = DefaultUnusedScheduleState;
|
||
expectedName = GetTemplateNameForUnused(expectedName);
|
||
logger.LogInformation("Шаблон {TemplateId} (UnitId {UnitId}) → деактивация.", template.Id, template.UnitId);
|
||
}
|
||
|
||
bool needsUpdate =
|
||
template.StatusTypeId != TemplateStatusTypeEnum.Updating &&
|
||
(
|
||
template.StatusTypeId != targetStatus ||
|
||
template.Name != expectedName ||
|
||
template.IsActiveTemplate != targetIsActiveTemplate ||
|
||
template.IsActiveSchedule != targetIsActiveSchedule
|
||
);
|
||
|
||
if (!needsUpdate) continue;
|
||
|
||
template.StatusTypeId = TemplateStatusTypeEnum.Updating;
|
||
template.DateModified = DateTimeOffset.UtcNow;
|
||
|
||
if (!await templateService.CommitAsync(initiator))
|
||
{
|
||
logger.LogError("Не удалось перевести шаблон {TemplateId} в Updating.", template.Id);
|
||
continue;
|
||
}
|
||
|
||
var nextRun = await GetNextRunAsync(job, template.NextRun);
|
||
|
||
var updateRequest = new TemplateUpdaterMq
|
||
{
|
||
TemplateId = template.Id,
|
||
JobId = jobId,
|
||
UnitId = template.UnitId,
|
||
Name = expectedName,
|
||
IsActiveTemplate = targetIsActiveTemplate,
|
||
IsActiveSchedule = targetIsActiveSchedule,
|
||
LastRun = template.LastRun,
|
||
NextRun = nextRun,
|
||
Index = template.Index,
|
||
StatusTypeId = targetStatus,
|
||
Initiator = initiator
|
||
};
|
||
|
||
await SendTemplateUpdateMessage(updateRequest);
|
||
}
|
||
|
||
logger.LogInformation("Обновление шаблонов завершено для JobId {JobId}.", jobId);
|
||
}
|
||
|
||
|
||
private async Task<bool> DeactivateTemplateAsync(
|
||
Template template,
|
||
Guid jobId,
|
||
HistoryInitiator initiator)
|
||
{
|
||
if (template.StatusTypeId == TemplateStatusTypeEnum.Updating)
|
||
return true; // уже в обработке
|
||
|
||
logger.LogInformation("Шаблон {TemplateId} (UnitId {UnitId}) → деактивация.",
|
||
template.Id, template.UnitId);
|
||
|
||
template.StatusTypeId = TemplateStatusTypeEnum.Updating;
|
||
template.DateModified = DateTimeOffset.UtcNow;
|
||
|
||
if (!await templateService.CommitAsync(initiator))
|
||
{
|
||
logger.LogError("Не удалось перевести шаблон {TemplateId} в Updating.", template.Id);
|
||
return false;
|
||
}
|
||
|
||
var updateRequest = new TemplateUpdaterMq
|
||
{
|
||
TemplateId = template.Id,
|
||
JobId = jobId,
|
||
UnitId = template.UnitId,
|
||
Name = GetTemplateNameForUnused(template.Name),
|
||
IsActiveTemplate = DefaultUnusedTemplateState,
|
||
IsActiveSchedule = DefaultUnusedScheduleState,
|
||
LastRun = template.LastRun,
|
||
NextRun = template.NextRun,
|
||
Index = template.Index,
|
||
StatusTypeId = TemplateStatusTypeEnum.Unused,
|
||
Initiator = initiator
|
||
};
|
||
|
||
await SendTemplateUpdateMessage(updateRequest);
|
||
return true;
|
||
}
|
||
|
||
private async Task<bool> ReactivateOrRenameTemplateAsync(
|
||
Template template,
|
||
Job job,
|
||
string expectedName,
|
||
HistoryInitiator initiator)
|
||
{
|
||
if (template.StatusTypeId == TemplateStatusTypeEnum.Updating)
|
||
return true;
|
||
|
||
bool needsUpdate = template.Name != expectedName
|
||
|| template.StatusTypeId != TemplateStatusTypeEnum.Used;
|
||
|
||
if (!needsUpdate) return true;
|
||
|
||
logger.LogInformation("Шаблон {TemplateId}: требуется обновление имени или реактивация.", template.Id);
|
||
|
||
template.StatusTypeId = TemplateStatusTypeEnum.Updating;
|
||
template.DateModified = DateTimeOffset.UtcNow;
|
||
|
||
if (!await templateService.CommitAsync(initiator))
|
||
{
|
||
logger.LogError("Не удалось перевести шаблон {TemplateId} в Updating.", template.Id);
|
||
return false;
|
||
}
|
||
|
||
var nextRun = await GetNextRunAsync(job, template.NextRun);
|
||
|
||
var updateRequest = new TemplateUpdaterMq
|
||
{
|
||
TemplateId = template.Id,
|
||
JobId = job.Id,
|
||
UnitId = template.UnitId,
|
||
Name = expectedName,
|
||
IsActiveTemplate = template.IsActiveTemplate,
|
||
IsActiveSchedule = template.IsActiveSchedule,
|
||
LastRun = template.LastRun,
|
||
NextRun = nextRun,
|
||
Index = template.Index,
|
||
StatusTypeId = TemplateStatusTypeEnum.Used,
|
||
Initiator = initiator
|
||
};
|
||
|
||
await SendTemplateUpdateMessage(updateRequest);
|
||
return true;
|
||
}
|
||
|
||
private async Task<bool> SendTemplateGeneratorMessageAsync(
|
||
Guid jobId,
|
||
Guid unitId,
|
||
HistoryInitiator initiator)
|
||
{
|
||
logger.LogInformation("Создание нового шаблона для UnitId {UnitId}.", unitId);
|
||
var mqRequest = new TemplateGeneratorWorkerMq
|
||
{
|
||
JobId = jobId,
|
||
UnitId = unitId,
|
||
HistoryInitiator = initiator
|
||
};
|
||
|
||
var msg = JsonSerializer.Serialize(mqRequest);
|
||
var result = await mqService.SendAsync(mqSettings.TemplateGenerator, new[] { msg });
|
||
|
||
if (!result.IsSuccess)
|
||
logger.LogError("Ошибка отправки команды создания шаблона для UnitId {UnitId}.", unitId);
|
||
|
||
return result.IsSuccess;
|
||
}
|
||
|
||
private async Task SendTemplateUpdateMessage(TemplateUpdaterMq updateRequest)
|
||
{
|
||
logger.LogDebug("Отправка сообщения в очередь '{Queue}' для шаблона {TemplateId}",
|
||
mqSettings.TemplateUpdater.QueueName, updateRequest.TemplateId);
|
||
|
||
var msg = JsonSerializer.Serialize(updateRequest);
|
||
var result = await mqService.SendAsync(mqSettings.TemplateUpdater, new[] { msg });
|
||
|
||
if (result.IsSuccess)
|
||
{
|
||
logger.LogInformation("Отправлен запрос на обновление шаблона {TemplateId}", updateRequest.TemplateId);
|
||
}
|
||
else
|
||
{
|
||
logger.LogError("Ошибка при отправке запроса на обновление шаблона {TemplateId} в очередь '{Queue}'.",
|
||
updateRequest.TemplateId, mqSettings.TemplateUpdater.QueueName);
|
||
}
|
||
}
|
||
|
||
private async Task<Template?> TryReuseOneUnusedTemplateAsync(
|
||
Guid jobId,
|
||
Guid unitId,
|
||
HistoryInitiator initiator,
|
||
int maxAttempts = 3)
|
||
{
|
||
for (int attempt = 1; attempt <= maxAttempts; attempt++)
|
||
{
|
||
try
|
||
{
|
||
var unusedCandidates = await templateService.Get()
|
||
.Where(t => t.StatusTypeId == TemplateStatusTypeEnum.Unused)
|
||
.OrderBy(t => t.DateModified ?? t.DateCreated)
|
||
.Take(UnusedCandidateBatchSize)
|
||
.ToListAsync();
|
||
|
||
if (!unusedCandidates.Any())
|
||
{
|
||
logger.LogDebug("Нет Unused-шаблонов (попытка {Attempt}).", attempt);
|
||
return null;
|
||
}
|
||
|
||
foreach (var candidate in unusedCandidates)
|
||
{
|
||
var originalStatus = candidate.StatusTypeId;
|
||
var originalModified = candidate.DateModified;
|
||
|
||
try
|
||
{
|
||
candidate.StatusTypeId = TemplateStatusTypeEnum.Updating;
|
||
candidate.DateModified = DateTimeOffset.UtcNow;
|
||
|
||
if (await templateService.CommitAsync(initiator))
|
||
{
|
||
logger.LogInformation("Успешно захвачен шаблон {TemplateId} для UnitId {UnitId} (попытка {Attempt}).",
|
||
candidate.Id, unitId, attempt);
|
||
return candidate;
|
||
}
|
||
|
||
// Откат при неудаче
|
||
candidate.StatusTypeId = originalStatus;
|
||
candidate.DateModified = originalModified;
|
||
}
|
||
catch (Exception ex) when (
|
||
ex is DbUpdateException ||
|
||
ex.InnerException?.Message.Contains("deadlock", StringComparison.OrdinalIgnoreCase) == true ||
|
||
ex.InnerException?.Message.Contains("timeout", StringComparison.OrdinalIgnoreCase) == true)
|
||
{
|
||
logger.LogWarning(ex, "Конфликт при захвате шаблона {TemplateId} (попытка {Attempt}).", candidate.Id, attempt);
|
||
candidate.StatusTypeId = originalStatus;
|
||
candidate.DateModified = originalModified;
|
||
}
|
||
}
|
||
|
||
if (attempt < maxAttempts)
|
||
await Task.Delay(Random.Shared.Next(5, 15) * attempt);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
logger.LogError(ex, "Ошибка в попытке захвата (попытка {Attempt}).", attempt);
|
||
if (attempt == maxAttempts) throw;
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
private string GetTemplateNameForUnused(string templateName)
|
||
{
|
||
return templateName + "_" + DateTimeOffset.UtcNow.ToUnixTimeSeconds();
|
||
}
|
||
|
||
private async Task<Job?> GetJobWithGroupAndAutoControlAsync(Guid jobId)
|
||
{
|
||
return await jobService.Get()
|
||
.Include(j => j.Group)
|
||
.Include(j => j.AutoControl)
|
||
.FirstOrDefaultAsync(j => j.Id == jobId);
|
||
}
|
||
|
||
private async Task<HashSet<Guid>> GetExpectedUnitIdsAsync(Guid jobId)
|
||
{
|
||
var units = await unitFilterService.GetUnitsIdByJobFilterAsync(jobId);
|
||
return units?.ToHashSet() ?? new HashSet<Guid>();
|
||
}
|
||
|
||
private async Task<string> GetNormalizedTemplateNameAsync(Job job, Guid unitId)
|
||
{
|
||
var rawName = await shortcodesService.ApplyShortcodesAsync(job.TemplateNameMask, unitId, job.Id);
|
||
return rawName.ToUpper();
|
||
}
|
||
|
||
private async Task<DateTimeOffset> GetNextRunAsync(Job job, DateTimeOffset? currentNextRun = null)
|
||
{
|
||
var now = DateTimeOffset.UtcNow;
|
||
if (currentNextRun.HasValue && currentNextRun.Value > now)
|
||
{
|
||
return currentNextRun.Value;
|
||
}
|
||
|
||
var referenceDate = job.Group?.ReferenceDate ?? now;
|
||
return await esppScheduleTransformService.GetNextDateAsync(job.GroupId, referenceDate);
|
||
}
|
||
}
|
||
} |