parr_api/PARR.TemplateActivator/TemplateActivator.cs

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);
    }
}