feat(api): Логика смены статусов для заданий робота. RobotTaskController

2023-10-05 16:56:02 +10:00
parent 4679994600
commit af4a0b60b8
26 changed files with 2779 additions and 126 deletions
--- a/PARR.API/Controllers/V1/RobotTaskController.cs
+++ b/PARR.API/Controllers/V1/RobotTaskController.cs
@@ -0,0 +1,109 @@
+using AutoMapper;
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.EntityFrameworkCore;
+using PARR.API.Contracts.V1;
+using PARR.API.Contracts.V1.Responses;
+using PARR.API.Contracts.V1.Responses.Base;
+using PARR.API.Controllers.V1.Base;
+using PARR.DAL.Contracts;
+using PARR.DAL.Models;
+using PARR.DAL.Services.Interfaces;
+
+namespace PARR.API.Controllers.V1
+{
+    public class RobotTaskController : BaseApiController
+    {
+        private readonly IMapper mapper;
+        private readonly SettingsFromDb settingsFromDb;
+        private readonly IRobotConfigurationService robotConfigurationService;
+
+        public RobotTaskController(
+            IMapper mapper,
+            SettingsFromDb settingsFromDb,
+            IRobotConfigurationService robotConfigurationService
+            )
+        {
+            this.mapper = mapper;
+            this.settingsFromDb = settingsFromDb;
+            this.robotConfigurationService = robotConfigurationService;
+        }
+
+
+        /// <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)
+        {
+            //Ищем все задания с превышенным кол-вом попыток и с просроченным временем и ставим им статус ошибки
+            await robotConfigurationService.FindUnfulfilledTaskAndSetRobotErrorStatusAsync(settingsFromDb.RobotAttemptsNumber, settingsFromDb.RobotWaitTime);
+
+            var query = robotConfigurationService.Get()
+                                .Where(t => t.RobotCode == (int)robotCode && t.TaskStatusCode == (int)taskStatusCode)
+                                .AsSplitQuery();
+
+            switch (robotCode)
+            {
+                case RobotsEnum.TemplateOrder:
+                    query = query.Include(t => t.Template)
+                                        .ThenInclude(t => t!.Host).ThenInclude(t => t!.ResponseArea)
+                                    .Include(t => t.Template)
+                                        .ThenInclude(t => t!.Host)
+                                    .Include(t => t.Template)
+                                        .ThenInclude(a => a!.ApplicationsInWork)
+                                        .ThenInclude(w => w!.Work)
+                                        .ThenInclude(t => t!.Tnk)
+                                        .ThenInclude(s => s!.Subprocess)
+                                        .ThenInclude(p => p!.Process);
+                    break;
+                case RobotsEnum.ScheduleOrder:
+                    //todo: include
+                    break;
+                default:
+                    break;
+            }
+
+            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.LastStatusUpdated < endDate
+                                                );
+            }
+
+            if (task == null)
+                return NotFound();
+
+            //TODO: сделать маппинг!!!
+
+            switch (robotCode)
+            {
+                case RobotsEnum.TemplateOrder:
+                    //RobotTaskTemplateResponse
+                    return Ok(new Response<RobotTaskTemplateResponse>(mapper.Map<RobotTaskTemplateResponse>(task), true));
+                case RobotsEnum.ScheduleOrder:
+                    //RobotTaskScheduleResponse
+                    return Ok(new Response<RobotTaskScheduleResponse>(mapper.Map<RobotTaskScheduleResponse>(task), true));
+                default:
+                    break;
+            }
+
+            return BadRequest();
+        }
+    }
+}