feat(templateActivator): Методы чтения MQ, обновления состояний шаблонов и расписаний, обновления заданий роботов

2024-07-25 17:23:08 +10:00
parent 7220bdd8af
commit 014c760225
13 changed files with 357 additions and 18 deletions
--- a/PARR.TemplateActivator/TemplateActivator.cs
+++ b/PARR.TemplateActivator/TemplateActivator.cs
@@ -0,0 +1,113 @@
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Logging;
+using PARR.Constants;
+using PARR.DAL.Contracts;
+using PARR.DAL.Models;
+using PARR.DAL.Services.Interfaces;
+
+namespace PARR.TemplateActivator;
+
+internal class TemplateActivator : ITemplateActivator
+{
+    private readonly ILogger<TemplateActivator> logger;
+    private readonly ITemplateService templateService;
+    private readonly IRobotConfigurationService robotConfigurationService;
+
+    public TemplateActivator(
+        ILogger<TemplateActivator> logger,
+        ITemplateService templateService,
+        IRobotConfigurationService robotConfigurationService
+        )
+    {
+        this.logger = logger;
+        this.templateService = templateService;
+        this.robotConfigurationService = robotConfigurationService;
+    }
+    public async Task ChangeStateAsync(Guid applicationInWorkId, TemplateActivatorActionsEnum action)
+    {
+        var query = templateService.Get()
+                           .Include(t => t.RobotConfigurations).ThenInclude(t => t.Robot)
+                           .Where(t => t.ApplicationInWorkId == applicationInWorkId);
+        var changes = 0;
+        switch (action)
+        {
+            case TemplateActivatorActionsEnum.ActivateAllTemplates:
+                query = query.Where(t => !t.IsActiveTemplate);
+                changes = await UpdateTemplateStateAsync(query, true);
+                break;
+            case TemplateActivatorActionsEnum.ActivateTemplatesWithActiveSchedules:
+                query = query.Where(t => !t.IsActiveTemplate && t.IsActiveSchedule);
+                changes = await UpdateTemplateStateAsync(query, true);
+                break;
+            case TemplateActivatorActionsEnum.DeactivateAllTemplates:
+                query = query.Where(t => t.IsActiveTemplate);
+                changes = await UpdateTemplateStateAsync(query, false);
+                break;
+            case TemplateActivatorActionsEnum.DeactivateTemplatesWithDeactivationSchedules:
+                query = query.Where(t => !t.IsActiveTemplate && !t.IsActiveSchedule);
+                changes = await UpdateTemplateStateAsync(query, false);
+                break;
+            case TemplateActivatorActionsEnum.ActivateAllSchedules:
+                query = query.Where(t => !t.IsActiveSchedule);
+                changes = await UpdateScheduleStateAsync(query, true);
+                break;
+            case TemplateActivatorActionsEnum.DeactivateAllSchedules:
+                query = query.Where(t => t.IsActiveSchedule);
+                changes = await UpdateScheduleStateAsync(query, false);
+                break;
+            case TemplateActivatorActionsEnum.ActivateSchedulesWithActiveTemplates:
+                query = query.Where(t => t.IsActiveTemplate && !t.IsActiveSchedule);
+                changes = await UpdateScheduleStateAsync(query, true);
+                break;
+            case TemplateActivatorActionsEnum.DeactivateSchedulesWithDeactivationTemplates:
+                query = query.Where(t => !t.IsActiveTemplate);
+                changes = await UpdateScheduleStateAsync(query, false);
+                break;
+            default:
+                break;
+        }
+
+        if (changes > 0 && !await templateService.CommitAsync())
+            logger.LogError($"{this.GetType} не удалось записать измения состояний шаблонов и расписаний регламентной работы {applicationInWorkId}");
+
+    }
+
+    private async Task<int> UpdateScheduleStateAsync(IQueryable<Template> query, bool newState)
+    {
+        var templates = await query.Select(t => t).ToListAsync();
+
+        if (templates.Any())
+            templates.ForEach(t =>
+            {
+                t.IsActiveSchedule = newState;
+                UpdateRobotConfiguration(RobotsEnum.ScheduleOrder, t);
+            });
+
+
+        templates.ForEach(t => t.IsActiveSchedule = newState);
+
+        return templates.Count;
+    }
+
+    private async Task<int> UpdateTemplateStateAsync(IQueryable<Template> query, bool newState)
+    {
+        var templates = await query.Select(t => t).ToListAsync();
+
+        if (templates.Any())
+            templates.ForEach(t =>
+            {
+                t.IsActiveTemplate = newState;
+                UpdateRobotConfiguration(RobotsEnum.TemplateOrder, t);
+            });
+
+
+        return templates.Count;
+    }
+
+
+    private void UpdateRobotConfiguration(RobotsEnum robotsEnum, Template t)
+    {
+        var config = robotConfigurationService.GetFromTemplateByRobotCode(robotsEnum, ref t);
+        robotConfigurationService.ChangeTaskStatus(TaskStatusEnum.Updating, ref config);
+    }
+}