From 4aff13dfb12696c5a31b2a692fd9be5f886544f1 Mon Sep 17 00:00:00 2001
From: Mikhail Kuznetsov <ivc_kuznetsovmv@dvgd.rzd>
Date: Tue, 17 Sep 2024 11:30:33 +1000
Subject: [PATCH] =?UTF-8?q?feat(templateActivator):=20=D0=B4=D0=BE=D0=B1?=
 =?UTF-8?q?=D0=B0=D0=B2=D0=BB=D0=B5=D0=BD=D1=8B=20=D0=BC=D0=B5=D1=82=D0=BE?=
 =?UTF-8?q?=D0=B4=D1=8B=20=D0=B4=D0=BB=D1=8F=20=D1=80=D0=B5=D0=B0=D0=BB?=
 =?UTF-8?q?=D0=B8=D0=B7=D0=B0=D1=86=D0=B8=D0=B8=20=D0=BC=D0=B5=D1=82=D0=BE?=
 =?UTF-8?q?=D0=B4=D0=BE=D0=B2=20=D0=B0=D0=B2=D1=82=D0=BE-=D0=BA=D0=BE?=
 =?UTF-8?q?=D0=BD=D1=82=D1=80=D0=BE=D0=BB=D1=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 PARR.TemplateActivator/TemplateActivator.cs | 150 +++++++++++++++++++-
 1 file changed, 147 insertions(+), 3 deletions(-)

diff --git a/PARR.TemplateActivator/TemplateActivator.cs b/PARR.TemplateActivator/TemplateActivator.cs
index f6c6af2f..4525d793 100644
--- a/PARR.TemplateActivator/TemplateActivator.cs
+++ b/PARR.TemplateActivator/TemplateActivator.cs
@@ -1,9 +1,12 @@
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging;
+using PARR.BLL.Helpers;
 using PARR.Constants;
 using PARR.DAL.Contracts;
 using PARR.DAL.Models;
 using PARR.DAL.Services.Interfaces;
+using System.Threading.Channels;
+using System.Threading.Tasks;
 
 namespace PARR.TemplateActivator;
 
@@ -12,16 +15,22 @@ internal class TemplateActivator : ITemplateActivator
     private readonly ILogger<TemplateActivator> logger;
     private readonly ITemplateService templateService;
     private readonly IRobotConfigurationService robotConfigurationService;
+    private readonly IJobAutoControlService jobAutoControlService;
+    private readonly IApplicationsInWorkService applicationsInWorkService;
 
     public TemplateActivator(
         ILogger<TemplateActivator> logger,
         ITemplateService templateService,
-        IRobotConfigurationService robotConfigurationService
+        IRobotConfigurationService robotConfigurationService,
+        IJobAutoControlService jobAutoControlService,
+        IApplicationsInWorkService applicationsInWorkService
         )
     {
         this.logger = logger;
         this.templateService = templateService;
         this.robotConfigurationService = robotConfigurationService;
+        this.jobAutoControlService = jobAutoControlService;
+        this.applicationsInWorkService = applicationsInWorkService;
     }
 
 
@@ -82,16 +91,121 @@ internal class TemplateActivator : ITemplateActivator
                     );
                 changes = await UpdateScheduleStateAsync(query, false);
                 break;
+            case TemplateActivatorActionsEnum.ActivateAllForAutoControl:
+                logger.LogInformation($"Получен запрос на активацию шаблонов и расписаний, для ЭК удовлетворяющим условиям автоконтроля");
+                changes = await CoordinateJobsWithAutoControl(applicationInWorkId, query, true);
+                break;
+            case TemplateActivatorActionsEnum.DeactivateAllForAutoControl:
+                logger.LogInformation($"Получен запрос на деактивацию шаблонов и расписаний, для ЭК неудовлетворяющим условиям автоконтроля");
+                changes = await CoordinateJobsWithAutoControl(applicationInWorkId, query, false);
+                break;
             default:
                 break;
         }
 
         if (changes > 0 && !await templateService.CommitAsync())
-            logger.LogError($"{this.GetType} не удалось записать измения в БД о новых состояниях шаблонов и расписаний регламентной работы Id {applicationInWorkId}");
+            logger.LogError($"{GetType().Name} не удалось записать измения в БД о новых состояниях шаблонов и расписаний регламентной работы Id {applicationInWorkId}");
 
-        logger.LogDebug($"{GetType} изменил состояние шаблонов/расписаний({changes}) регламентной работы Id {applicationInWorkId}");
+        logger.LogDebug($"{GetType().Name} изменил состояние шаблонов/расписаний({changes}) регламентной работы Id {applicationInWorkId}");
     }
 
+    private async Task<int> CoordinateJobsWithAutoControl(Guid applicationInWorkId, IQueryable<Template> query, bool isActivate)
+    {
+        var changes = 0;
+
+        //Получаем настройки авто-контроля
+        var jobAutoControlSettings = await GetJobAutoControlSettings(applicationInWorkId);
+        if (jobAutoControlSettings == null)
+            return 0;
+
+        var trueEkStatuses = jobAutoControlSettings.JobAutoControlInEkStatuses;
+        var trueMasks = jobAutoControlSettings.JobEkMasks;
+
+        if (!trueEkStatuses.Any() || !trueMasks.Any())
+        {
+            logger.LogWarning($"В ApplicationInWork({applicationInWorkId}) не заданы настройки фильтрации статусов ЭК хоста или маски его имени");
+            return 0;
+        }
+
+        //Фильтруем обратное состояние - активированные для деактивации
+        query = FilterTemplatesQueryByWrongState(query, jobAutoControlSettings, isActivate);
+        
+        //Добавляем информацию о рабочих группах для возможности их фильтрации
+        query = query.Include(t => t.Host).ThenInclude(h => h!.WorkGroup);
+
+        //Фильтруем по маскам ЭК
+        var templates = FilterTemplatesByNameNotMatchToMasks(query, trueMasks);
+
+        if (!templates.Any())
+            return 0;
+
+        //Фильтруем по рабочим группам
+        var appInWorkWorkGroups = GetApplicationInWorkWorkGroups(applicationInWorkId);
+        if (appInWorkWorkGroups.Any())
+            templates = templates.Where(t => appInWorkWorkGroups.Any(awg => t.Host!.WorkGroupId == awg.WorkGroupId)).ToList();
+
+        //Фильтруем соответствие статусов ЭК условиям автоконтроля
+        var checkEkStatus = FilterTemplatesByWrongEkStatus(templates, trueEkStatuses, isActivate);
+        //Меняем состояние отфильтрованных шаблонов если какие-то остались
+        if (checkEkStatus.Any())
+            changes = UpdateState(checkEkStatus,
+                isActivate ? jobAutoControlSettings.OnTemplateIsActive : jobAutoControlSettings.OffTemplateIsActive,
+                isActivate ? jobAutoControlSettings.OnScheduleIsActive : jobAutoControlSettings.OffScheduleIsActive);
+
+        return changes;
+    }
+
+    private List<AppInWorkInWorkGroup> GetApplicationInWorkWorkGroups(Guid applicationInWorkId)
+    {
+        var appInWorkInWorkGroups = applicationsInWorkService.Get()
+                                 .Include(aiw => aiw.WorkGroups)
+                                 .Where(x => x.Id == applicationInWorkId).Select(aiw => aiw.WorkGroups).FirstOrDefault();
+
+        if (appInWorkInWorkGroups != null)
+            return appInWorkInWorkGroups.ToList();
+
+        return new List<AppInWorkInWorkGroup>();
+    }
+
+    private List<Template> FilterTemplatesByWrongEkStatus(List<Template> templates, ICollection<JobAutoControlInEkStatus> trueEkStatuses, bool isActivate)
+    {
+        if (isActivate)
+            return templates.Where(t => trueEkStatuses.Any(tes => tes.EkStatusCode == t.Host!.EkStatusCode)).ToList();
+
+        return templates.Where(t => !trueEkStatuses.Any(tes => tes.EkStatusCode == t.Host!.EkStatusCode)).ToList();
+    }
+
+    private List<Template> FilterTemplatesByNameNotMatchToMasks(IQueryable<Template> query, ICollection<JobEkMask> trueMasks)
+    {
+        var result = new List<Template>();
+        foreach (var mask in trueMasks)
+        {
+            var templates = query.Where(t => EF.Functions.Like(t.Host!.Ek.ToLower(), SqlHelpers.RegexToLike(mask.Name))).ToList();
+            result.AddRange(templates);
+        }
+
+        return result;
+    }
+
+    private IQueryable<Template> FilterTemplatesQueryByWrongState(IQueryable<Template> query, JobAutoControl jobAutoControlSettings, bool isActivate)
+    {
+        if (isActivate)
+            return query.Include(t => t.Host).Where(t => t.IsActiveTemplate != jobAutoControlSettings.OnTemplateIsActive || t.IsActiveSchedule != jobAutoControlSettings.OnScheduleIsActive);
+
+        return query.Include(t => t.Host).Where(t => t.IsActiveTemplate != jobAutoControlSettings.OffTemplateIsActive || t.IsActiveSchedule != jobAutoControlSettings.OffScheduleIsActive);
+    }
+
+    private async Task<JobAutoControl> GetJobAutoControlSettings(Guid applicationInWorkId)
+    {
+        return await jobAutoControlService.Get()
+                            .Include(jac => jac.JobAutoControlInEkStatuses)
+                            .Include(jac => jac.JobEkMasks)
+                            .Where(jac => jac.ApplicationInWorkId == applicationInWorkId).FirstAsync();
+    }
+
+
+
+
 
     /// <summary>
     /// Обновить состояние расписаний по запросу query на newState, без записи в БД. Возвращает количество шаблонов требующих фиксации в БД
@@ -137,6 +251,36 @@ internal class TemplateActivator : ITemplateActivator
         return templates.Count;
     }
 
+
+    /// <summary>
+    /// Обновить состояние шаблонов по запросу query на newState, без записи в БД. Возвращает количество расписаний требующих фиксацию в БД
+    /// </summary>
+    /// <param name="query"></param>
+    /// <param name="newState"></param>
+    /// <returns></returns>
+    private async Task<int> UpdateStateAsync(IQueryable<Template> query, bool newTemplateState, bool newSchedulerState)
+    {
+        var templates = await query.Select(t => t).ToListAsync();
+
+        return UpdateState(templates, newTemplateState, newSchedulerState);
+    }
+
+
+    private int UpdateState(List<Template> templates, bool newTemplateState, bool newSchedulerState)
+    {
+        if (templates.Any())
+            templates.ForEach(t =>
+            {
+                t.IsActiveTemplate = newTemplateState;
+                t.IsActiveSchedule = newSchedulerState;
+                UpdateRobotConfiguration(RobotsEnum.TemplateOrder, t);
+            });
+
+
+        return templates.Count;
+    }
+
+
     /// <summary>
     /// принудительно обновить состояние задачи, связанной с шаблоном/расписанием, на обновление
     /// </summary>