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

PARR.DAL/Services/Implementations/RobotConfigurationService.cs

@@ -1,6 +1,7 @@
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging;
 using PARR.DAL.Context;
+using PARR.DAL.Contracts;
 using PARR.DAL.Models;
 using PARR.DAL.Services.Abstracts;
 using PARR.DAL.Services.Interfaces;
@@ -10,14 +11,107 @@ namespace PARR.DAL.Services.Implementations
     internal class RobotConfigurationService : BaseService<RobotConfiguration>, IRobotConfigurationService
     {
         private readonly DataContext dataContext;
+        private readonly ILogger<RobotConfigurationService> logger;
 
         public RobotConfigurationService(DataContext dataContext, ILogger<RobotConfigurationService> logger) : base(logger)
         {
             this.dataContext = dataContext;
+            this.logger = logger;
         }
 
         protected override DbSet<RobotConfiguration> EntitySet => dataContext.RobotConfigurations;
 
         protected override DataContext EntitiContext => dataContext;
+
+
+        public void ChangeTaskStatus(TaskStatusEnum taskStatus, ref RobotConfiguration configuration)
+        {
+            configuration.TaskStatusCode = (int)taskStatus;
+
+            switch (taskStatus)
+            {
+                case TaskStatusEnum.Creating:
+                    ChangeRobotStatus(RobotStatusEnum.Wait, ref configuration);
+                    break;
+                case TaskStatusEnum.Updating:
+                    ChangeRobotStatus(RobotStatusEnum.Wait, ref configuration);
+                    break;
+                case TaskStatusEnum.Ok:
+                    break;
+                default:
+                    break;
+            }
+
+        }
+
+        public void ChangeRobotStatus(RobotStatusEnum robotStatus, ref RobotConfiguration configuration)
+        {
+            configuration.RobotStatusCode = (int)robotStatus;
+
+            switch (robotStatus)
+            {
+                case RobotStatusEnum.InProgress:
+                    configuration.AttemptsNumber++;
+                    configuration.LastStatusUpdated = DateTimeOffset.UtcNow;
+                    break;
+                //case RobotStatusEnum.Error:
+                //    break;
+                case RobotStatusEnum.Complete:
+                    configuration.LastStatusUpdated = DateTimeOffset.UtcNow;
+                    break;
+                case RobotStatusEnum.Wait:
+                    configuration.LastStatusUpdated = null;
+                    configuration.AttemptsNumber = 0;
+                    break;
+                default:
+                    break;
+            }
+        }
+
+        public RobotConfiguration GetFromTemplateByRobotCode(RobotsEnum robotsEnum, ref Template template)
+        {
+            var config = template.RobotConfigurations.FirstOrDefault(t => t.RobotCode == (int)robotsEnum);
+
+            if (config == null)
+            {
+                logger.LogError($"У шаблона нет конфигурации роботов. TemplateId: {template.Id}");
+                throw new Exception($"У шаблона нет конфигурации роботов. TemplateId: {template.Id}");
+            }
+
+            return config;
+        }
+
+
+        // Поиск невыполненных заданий и установка им статуса ошибки робота
+        public async Task FindUnfulfilledTaskAndSetRobotErrorStatusAsync(int robotAttemptsNumber, TimeSpan robotWaitTime)
+        {
+            //Ищем `RobotStatusCode` = 22 и `LastStatusUpdated` истекло и `AttemptsNumber` >= допустимого значения из настроек,
+            //ставим всем этим записям `RobotStatusCode`= 33
+
+            var endDate = DateTimeOffset.UtcNow.Add(-robotWaitTime);
+
+            var configObjs = await EntitySet.Where(t =>
+                                            t.RobotStatusCode == (int)RobotStatusEnum.InProgress
+                                            && t.AttemptsNumber >= robotAttemptsNumber
+                                            && t.LastStatusUpdated <= endDate
+                                        ).ToListAsync();
+
+            if (!configObjs.Any())
+                return;
+
+            configObjs.ForEach(item =>
+            {
+                ChangeRobotStatus(RobotStatusEnum.Error, ref item);
+                logger.LogInformation($"Устанавливаю RobotStatus: {RobotStatusEnum.Error} для RobotConfigurationId {item.Id}");
+            });
+
+            var result = await CommitAsync();
+
+            if (!result)
+                logger.LogError($"Ошибка при сохранении изменений RobotStatus для RobotConfigurationId: item.Id, RobotStatus: {RobotStatusEnum.Error}");
+
+        }
+
+
     }
 }