322 lines
18 KiB
C#
322 lines
18 KiB
C#
using AutoMapper;
|
||
using Microsoft.AspNetCore.Authorization;
|
||
using Microsoft.AspNetCore.Mvc;
|
||
using Microsoft.EntityFrameworkCore;
|
||
using PARR.API.Contracts.V1;
|
||
using PARR.API.Contracts.V1.Requests.Queries;
|
||
using PARR.API.Contracts.V1.Responses;
|
||
using PARR.API.Contracts.V1.Responses.Base;
|
||
using PARR.API.Controllers.V1.Base;
|
||
using PARR.API.Services.Interfaces;
|
||
using PARR.BLL.Helpers;
|
||
using PARR.Common.Domain;
|
||
using PARR.Constants;
|
||
using PARR.DAL.Contracts;
|
||
using PARR.DAL.DomainServices.Shortcodes;
|
||
using PARR.DAL.Models;
|
||
using PARR.DAL.NextRunServices;
|
||
using PARR.DAL.Services.Interfaces;
|
||
using PARR.DAL.Services.Interfaces.Schedule;
|
||
|
||
namespace PARR.API.Controllers.V1
|
||
{
|
||
[Authorize(Roles = ParrRoles.EsppRobot.RoleOrAdmin)]
|
||
public class RobotTaskController : BaseApiController
|
||
{
|
||
private readonly IMapper mapper;
|
||
private readonly SettingsFromDb settingsFromDb;
|
||
private readonly IRobotConfigurationService robotConfigurationService;
|
||
//private readonly IEsppScheduleTransformService esppScheduleTransformService;
|
||
private readonly ILogger<RobotTaskController> logger;
|
||
private readonly IClientService clientService;
|
||
private readonly IRobotHistoryService robotHistoryService;
|
||
private readonly IShortcodesService shortcodesService;
|
||
private readonly INextRunService nextRunService;
|
||
private readonly IScheduleResponseAreaTimeOffsetService scheduleResponseAreaTimeOffsetService;
|
||
|
||
public RobotTaskController(
|
||
IMapper mapper,
|
||
SettingsFromDb settingsFromDb,
|
||
IRobotConfigurationService robotConfigurationService,
|
||
//IEsppScheduleTransformService esppScheduleTransformService,
|
||
ILogger<RobotTaskController> logger,
|
||
IClientService clientService,
|
||
IRobotHistoryService robotHistoryService,
|
||
IShortcodesService shortcodesService,
|
||
INextRunService nextRunService,
|
||
IScheduleResponseAreaTimeOffsetService scheduleResponseAreaTimeOffsetService
|
||
)
|
||
{
|
||
this.mapper = mapper;
|
||
this.settingsFromDb = settingsFromDb;
|
||
this.robotConfigurationService = robotConfigurationService;
|
||
//this.esppScheduleTransformService = esppScheduleTransformService;
|
||
this.logger = logger;
|
||
this.clientService = clientService;
|
||
this.robotHistoryService = robotHistoryService;
|
||
this.shortcodesService = shortcodesService;
|
||
this.nextRunService = nextRunService;
|
||
this.scheduleResponseAreaTimeOffsetService = scheduleResponseAreaTimeOffsetService;
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Получить задание для робота по коду робота и по статусу задания
|
||
/// </summary>
|
||
/// <param name="robotCode"></param>
|
||
/// <param name="taskStatusCode"></param>
|
||
/// <returns></returns>
|
||
[HttpGet(ApiRoutes.RobotTask.GetByRobotAndStatusTask)]
|
||
public async Task<IActionResult> GetByRobotAndStatusTask([FromRoute] RobotsEnum robotCode, [FromRoute] TaskStatusEnum taskStatusCode, [FromQuery] RobotTaskQuery requestQuery)
|
||
{
|
||
//Ищем все задания с превышенным кол-вом попыток и с просроченным временем и ставим им статус ошибки
|
||
await robotConfigurationService.FindUnfulfilledTaskAndSetRobotErrorStatusAsync(settingsFromDb.RobotAttemptsNumber, settingsFromDb.RobotWaitTime);
|
||
|
||
|
||
var query = robotConfigurationService.Get()
|
||
.Where(t => t.RobotCode == (int)robotCode && t.TaskStatusCode == (int)taskStatusCode);
|
||
|
||
switch (robotCode)
|
||
{
|
||
case RobotsEnum.TemplateOrder:
|
||
// шаблоны
|
||
query = query
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Unit)
|
||
.ThenInclude(t => t!.UnitValues)
|
||
.ThenInclude(t => t.Field)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Unit)
|
||
.ThenInclude(t => t!.UnitValues)
|
||
.ThenInclude(t => t.Value)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(a => a!.Job)
|
||
.ThenInclude(t => t!.Group)
|
||
.ThenInclude(g => g.GroupType)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(w => w!.Job)
|
||
.ThenInclude(t => t!.Tnk)
|
||
.ThenInclude(s => s!.Subprocess)
|
||
.ThenInclude(p => p!.Process);
|
||
|
||
query = query
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.UnitsInTemplate);
|
||
|
||
break;
|
||
|
||
case RobotsEnum.ScheduleOrder:
|
||
//расписание
|
||
query = query
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Unit)
|
||
.ThenInclude(t => t!.UnitValues)
|
||
.ThenInclude(t => t.Field)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Unit)
|
||
.ThenInclude(t => t!.UnitValues)
|
||
.ThenInclude(t => t.Value)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(a => a!.Job)
|
||
.ThenInclude(t => t!.Group)
|
||
.ThenInclude(g => g.GroupType)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(a => a!.Job)
|
||
.ThenInclude(t => t!.Group)
|
||
.ThenInclude(t => t!.EsppSchValues)
|
||
.ThenInclude(t => t!.EsppSchTypeConfig)
|
||
.ThenInclude(t => t!.EsppSchTypeSchedule)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Job)
|
||
.ThenInclude(t => t!.Group)
|
||
.ThenInclude(t => t!.ScheduleExcludeType)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Job)
|
||
.ThenInclude(t => t!.Group)
|
||
.ThenInclude(t => t.ScheduleExcludeTypeCalendar)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.Job)
|
||
.ThenInclude(t => t!.Tnk)
|
||
.Include(t => t.Template)
|
||
.ThenInclude(t => t!.UnitsInTemplate);
|
||
|
||
//выбираем только записи с созданными шаблонами (у которых статус 20 или 30), а только потом у них ищем расписания
|
||
var createdTemplates = robotConfigurationService.Get()
|
||
.Where(t => t.RobotCode == (int)RobotsEnum.TemplateOrder && (t.TaskStatusCode == (int)TaskStatusEnum.Ok))
|
||
.Select(t => t.TemplateId);
|
||
query = query.Where(t => t.RobotCode == (int)RobotsEnum.ScheduleOrder && createdTemplates.Contains(t.TemplateId));
|
||
// query = query.Where(t => t.RobotCode == (int)RobotsEnum.ScheduleOrder && t.TemplateId==Guid.Parse("7cb7c3be-506d-40e0-a63f-4554edb52459"));
|
||
break;
|
||
|
||
default:
|
||
break;
|
||
}
|
||
|
||
|
||
// сортируем по NextRun, чтобы те у которых дата след срабатывания ближе к текущей, выполнились скорее
|
||
query = query.OrderBy(t => t.Template!.NextRun).ThenBy(t => t.Template!.IsActiveSchedule).ThenBy(t => t.Template.IsActiveTemplate);
|
||
|
||
RobotConfiguration? task = null;
|
||
|
||
//ищем задание в ожидании, если нашли, выбираем его
|
||
task = await query.FirstOrDefaultAsync(t => t.RobotStatusCode == (int)RobotStatusEnum.Wait);
|
||
|
||
if (task == null)
|
||
{
|
||
//ищем задания в работе, которые можно перезапустить
|
||
//Поиск по `RobotStatusCode` = 22.
|
||
//Далее проверяется `LastStatusUpdated`, что время последнего смены статуса не превышает допустимого(берется из настроек, поле `RobotWaitTime`)
|
||
//и что текущая попытка не больше разрешенной(берется из настроек, поле `RobotAttemptsNumber`) - если это так, берется эта запись.
|
||
|
||
var endDate = DateTimeOffset.UtcNow.Add(-settingsFromDb.RobotWaitTime);
|
||
task = await query
|
||
.FirstOrDefaultAsync(t =>
|
||
t.RobotStatusCode == (int)RobotStatusEnum.InProgress
|
||
&& t.AttemptsNumber < settingsFromDb.RobotAttemptsNumber
|
||
&& t.LastRobotStatusUpdated < endDate
|
||
);
|
||
}
|
||
|
||
if (task == null)
|
||
return NotFound();
|
||
|
||
if (requestQuery?.SetInProgressStatus == true)
|
||
{
|
||
var resultSetStatus = await SetInProgressStatusAsync(task.Id);
|
||
if (resultSetStatus == false)
|
||
{
|
||
logger.LogError($"Ошибка при установке статуса {RobotStatusEnum.InProgress.ToString()} для задания RobotConfigutationId {task.Id} (при выдаче задания роботу)");
|
||
return BadRequest(new Response(false, new List<ErrorModel> { new ErrorModel { Message = "Ошибка при выдаче задания." } }));
|
||
}
|
||
}
|
||
|
||
switch (robotCode)
|
||
{
|
||
case RobotsEnum.TemplateOrder:
|
||
{ //RobotTaskTemplateResponse
|
||
var robotTaskTemplateResponse = mapper.Map<RobotTaskTemplateResponse>(task);
|
||
|
||
robotTaskTemplateResponse.FullDescription = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.FullDescription, task.Template!);
|
||
robotTaskTemplateResponse.ShortDescription = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.ShortDescription, task.Template!);
|
||
robotTaskTemplateResponse.Solution = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.Solution, task.Template!);
|
||
robotTaskTemplateResponse.TnkName = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.TnkName, task.Template!);
|
||
robotTaskTemplateResponse.WorkName = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.WorkName, task.Template!);
|
||
robotTaskTemplateResponse.WorkGroup = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.WorkGroup, task.Template!);
|
||
robotTaskTemplateResponse.ResponseArea = await shortcodesService.ApplyShortcodesAsync(robotTaskTemplateResponse.ResponseArea, task.Template!);
|
||
|
||
return Ok(new Response<RobotTaskTemplateResponse>(robotTaskTemplateResponse, true));
|
||
}
|
||
case RobotsEnum.ScheduleOrder:
|
||
{ // если был запрос на расписание, проверяем у него nextRun, lastRun, обновляем их
|
||
var resultUpdateNextRun = await UpdateNextRunAsync(task);
|
||
if (!resultUpdateNextRun)
|
||
{
|
||
logger.LogError("Ошибка при расчете NextRun для templateId: {templateId}", task.TemplateId);
|
||
return BadRequest(new Response(false, new List<ErrorModel> { new ErrorModel { Message = "Ошибка при расчете NextRun" } }));
|
||
}
|
||
|
||
//RobotTaskScheduleResponse
|
||
var robotTaskScheduleResponse = mapper.Map<RobotTaskScheduleResponse>(task);
|
||
|
||
robotTaskScheduleResponse.Timezone = settingsFromDb.EsppScheduleTimezone;
|
||
|
||
robotTaskScheduleResponse.WorkGroup = await shortcodesService.ApplyShortcodesAsync(robotTaskScheduleResponse.WorkGroup, task.Template!);
|
||
robotTaskScheduleResponse.ResponseArea = await shortcodesService.ApplyShortcodesAsync(robotTaskScheduleResponse.ResponseArea, task.Template!);
|
||
|
||
//var nextRunWithRobotTz = nextRunService.GetNextRunWithTimezoneEsppAndResponseArea(task.Template!.NextRun, task.Template!.Job?.Group?.IsResponseAreaTimezone, robotTaskScheduleResponse.ResponseArea);
|
||
//nextRun в часовой зоне УЗ Робота ЕСПП
|
||
var nextRunWithRobotTz = task.Template!.NextRun.Add(nextRunService.GetEsppAccountOffset());
|
||
|
||
//на всякий случай еще раз проверяем, что дата не устарела и отправляем задание
|
||
if (nextRunWithRobotTz < DateTimeOffset.UtcNow)
|
||
{
|
||
logger.LogError("Ошибка при расчете NextRun для templateId: {templateId}, итоговое значение для робота, меньше чем сейчас {nextRunWithRobotTz}<{now}",
|
||
task.TemplateId, nextRunWithRobotTz, DateTimeOffset.UtcNow);
|
||
return BadRequest(new Response(false, new List<ErrorModel> { new ErrorModel { Message = "Ошибка при расчете NextRun" } }));
|
||
}
|
||
|
||
robotTaskScheduleResponse.NextStart = EsppScheduleHelpers.GetNextRun(nextRunWithRobotTz);
|
||
robotTaskScheduleResponse.GenerationTime = EsppScheduleHelpers.GetGenerationTime(nextRunWithRobotTz);
|
||
|
||
return Ok(new Response<RobotTaskScheduleResponse>(robotTaskScheduleResponse, true));
|
||
}
|
||
default:
|
||
break;
|
||
}
|
||
|
||
return BadRequest();
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Обоновить NextRun если он устарел
|
||
/// </summary>
|
||
/// <param name="task"></param>
|
||
/// <returns></returns>
|
||
private async Task<bool> UpdateNextRunAsync(RobotConfiguration task)
|
||
{
|
||
var template = task.Template!;
|
||
|
||
//var nextRun = await esppScheduleTransformService.GetNextDateAsync(template.Job!.GroupId, template!.Job!.Group!.ReferenceDate);
|
||
var nextRun = await nextRunService.GetNextRunForTemplateAsync(template.Id, false);
|
||
|
||
if (!nextRun.HasValue)
|
||
{
|
||
logger.LogError("При обновлении nextRun для шаблона {templateId}, расчитанный nextRun=null, ошибка в расчетах.", template.Id);
|
||
return false;
|
||
}
|
||
|
||
if (nextRun.Value < DateTimeOffset.UtcNow)
|
||
{
|
||
logger.LogError("При обновлении nextRun для шаблона {templateId}, расчитанный nextRun<Now [{nextRun}<{now}], ошибка в расчетах.", template.Id, nextRun.Value, DateTimeOffset.UtcNow);
|
||
return false;
|
||
}
|
||
|
||
if (nextRun != template.NextRun)
|
||
{
|
||
logger.LogDebug($"Для шаблона id {template.Id} обновляю nextRun, новое значение {nextRun}, старое значение {template.NextRun}");
|
||
|
||
template.LastRun = template.NextRun;
|
||
template.NextRun = nextRun.Value;
|
||
|
||
await robotConfigurationService.CommitAsync(new HistoryInitiator { InitiatorComment = "При получении задания роботом, обновил NextRun", InitiatorIp = clientService.GetClientIp()?.ToString(), InitiatorParrComponentId = ParrComponentsEnum.Api });
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Устанавливаем статус "Робот взял в работу", пишем в историю работы роботов инф о начале работ
|
||
/// </summary>
|
||
/// <param name="taskId"></param>
|
||
/// <returns></returns>
|
||
private async Task<bool> SetInProgressStatusAsync(Guid taskId)
|
||
{
|
||
var config = await robotConfigurationService.GetAsync(taskId);
|
||
|
||
//изменение статуса робота
|
||
robotConfigurationService.ChangeRobotStatus(RobotStatusEnum.InProgress, config!);
|
||
|
||
if (!await robotConfigurationService.CommitAsync())
|
||
return false;
|
||
|
||
//записываем в лог робота
|
||
var history = new RobotHistory
|
||
{
|
||
Id = Guid.NewGuid(),
|
||
HistoryLevel = (int)RobotHistoryLevelEnum.Start,
|
||
TaskStatusCode = config.TaskStatusCode,
|
||
RobotConfigurationId = config.Id,
|
||
RobotIp = clientService.GetClientIp()?.ToString()
|
||
};
|
||
|
||
if (!await robotHistoryService.CreateAsync(history) || !await robotHistoryService.CommitAsync())
|
||
return false;
|
||
|
||
return true;
|
||
}
|
||
|
||
}
|
||
}
|