parr_api/PARR.DAL/Services/Implementations/TemplateService.cs

using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Npgsql;
using PARR.Common.Domain;
using PARR.Constants;
using PARR.DAL.Context;
using PARR.DAL.Contracts;
using PARR.DAL.Models;
using PARR.DAL.Services.Abstracts;
using PARR.DAL.Services.Interfaces;

namespace PARR.DAL.Services.Implementations
{
    internal class TemplateService : BaseService<Template>, ITemplateService
    {
        private readonly DataContext dataContext;
        private readonly ILogger<TemplateService> logger;

        protected override DbSet<Template> EntitySet => dataContext.Templates;

        protected override DataContext EntitiContext => dataContext;

        public TemplateService(DataContext dataContext, ILogger<TemplateService> logger) : base(logger)
        {
            this.dataContext = dataContext;
            this.logger = logger;
        }

        public async Task<Template?> GetTemplateByNameAsync(string name)
        {
            logger.LogDebug("Поиск шаблона по имени: {TemplateName}", name);

            var template = await GetWithIncludes()
                .Include(t => t.RobotConfigurations)
                .FirstOrDefaultAsync(t => t.Name == name);

            if (template != null)
            {
                logger.LogDebug("Шаблон найден: {TemplateId}, имя: {TemplateName}", template.Id, template.Name);
            }
            else
            {
                logger.LogDebug("Шаблон с именем {TemplateName} не найден", name);
            }

            return template;
        }

        public IQueryable<Template> GetWithIncludes()
        {
            logger.LogDebug("Получаю шаблоны с include связями");

            return Get()
                .Include(h => h.Unit)
                    .ThenInclude(t => t!.UnitValues)
                    .ThenInclude(t => t.Value)
                .Include(h => h.Unit)
                    .ThenInclude(t => t!.UnitValues)
                    .ThenInclude(t => t.Field)
                .Include(a => a.Job)
                    .ThenInclude(t => t!.Tnk)
                    .ThenInclude(s => s!.Subprocess)
                    .ThenInclude(p => p!.Process)
                .Include(t => t.Job)
                    .ThenInclude(t => t!.Group)
                        .ThenInclude(t => t!.GroupType)
                .Include(t => t.Job)
                    .ThenInclude(t => t!.Group)
                        .ThenInclude(t => t.DistributionConfig)
                            .ThenInclude(t => t.DistributionPeriod);
        }

        public override Task<bool> CreateAsync(Template obj)
        {
            logger.LogDebug("Создание шаблона: {TemplateName}", obj.Name);

            // добавление роботов для шаблона
            obj.RobotConfigurations = new List<RobotConfiguration>
            {
                // робот по управлению шаблоном
                new RobotConfiguration
                {
                    Id = Guid.NewGuid(),
                    DateCreated = DateTimeOffset.UtcNow,
                    TemplateId = obj.Id,
                    RobotCode = (int)RobotsEnum.TemplateOrder,
                    TaskStatusCode = (int)TaskStatusEnum.Creating,
                    RobotStatusCode = (int)RobotStatusEnum.Wait,
                    AttemptsNumber = 0,
                    LastRobotStatusUpdated = null
                },
                // робот по управлениею расписанием
                new RobotConfiguration
                {
                    Id = Guid.NewGuid(),
                    DateCreated = DateTimeOffset.UtcNow,
                    TemplateId = obj.Id,
                    RobotCode = (int)RobotsEnum.ScheduleOrder,
                    TaskStatusCode = (int)TaskStatusEnum.Creating,
                    RobotStatusCode = (int)RobotStatusEnum.Wait,
                    AttemptsNumber = 0,
                    LastRobotStatusUpdated = null
                }
            };

            logger.LogDebug("Добавлены роботы для шаблона {TemplateName}", obj.Name);

            return base.CreateAsync(obj);
        }


        public async Task<Guid?> ReserveUnusedTemplateAsync(Guid newUnitId, Guid newJobId, HistoryInitiator initiator)
        {
            logger.LogDebug("Резервирую неиспользуемый шаблон с проверкой конфигураций роботов для UnitId: {UnitId}, JobId: {JobId}", newUnitId, newJobId);

            var sql = @"
                UPDATE ""Templates"" 
                SET ""StatusTypeId"" = @NewStatus,
                    ""DateModified"" = @DateModified,
                    ""UnitId"" = @NewUnitId,
                    ""JobId"" = @NewJobId,
                    ""InitiatorIp"" = @InitiatorIp,
                    ""InitiatorParrComponentId"" = @InitiatorComponent,
                    ""InitiatorComment"" = @InitiatorComment
                WHERE ""Id"" = (
                    SELECT t.""Id"" 
                    FROM ""Templates"" t
                    WHERE t.""StatusTypeId"" = @OldStatus 
                      AND t.""UnitId"" != @NewUnitId
                      AND EXISTS (
                          SELECT 1 
                          FROM ""RobotConfigurations"" rc
                          WHERE rc.""TemplateId"" = t.""Id""
                            AND rc.""RobotCode"" = @RobotCode1
                            AND rc.""TaskStatusCode"" = @TaskStatus
                            AND rc.""RobotStatusCode"" = @RobotStatus
                      )
                      AND EXISTS (
                          SELECT 1 
                          FROM ""RobotConfigurations"" rc
                          WHERE rc.""TemplateId"" = t.""Id""
                            AND rc.""RobotCode"" = @RobotCode2
                            AND rc.""TaskStatusCode"" = @TaskStatus
                            AND rc.""RobotStatusCode"" = @RobotStatus
                      )
                    ORDER BY t.""DateCreated"" ASC
                    LIMIT 1
                )
                RETURNING ""Id"";";

            var parameters = new[]
            {
        new NpgsqlParameter("@NewStatus", (int)TemplateStatusTypeEnum.Updating),
        new NpgsqlParameter("@DateModified", DateTimeOffset.UtcNow),
        new NpgsqlParameter("@NewUnitId", newUnitId),
        new NpgsqlParameter("@NewJobId", newJobId),
        new NpgsqlParameter("@InitiatorIp", initiator.InitiatorIp ?? (object)DBNull.Value),
        new NpgsqlParameter("@InitiatorComponent",
            initiator.InitiatorParrComponentId.HasValue
                ? (object)(int)initiator.InitiatorParrComponentId.Value
                : DBNull.Value),
        new NpgsqlParameter("@InitiatorComment", initiator.InitiatorComment ?? (object)DBNull.Value),
        new NpgsqlParameter("@OldStatus", (int)TemplateStatusTypeEnum.Unused),
        // Параметры для проверки конфигураций роботов
        new NpgsqlParameter("@RobotCode1", (int)RobotsEnum.TemplateOrder),
        new NpgsqlParameter("@RobotCode2", (int)RobotsEnum.ScheduleOrder),
        new NpgsqlParameter("@TaskStatus", (int)TaskStatusEnum.Ok),
        new NpgsqlParameter("@RobotStatus", (int)RobotStatusEnum.Complete)
    };

            try
            {
                var result = await dataContext.Database
                    .SqlQueryRaw<Guid>(sql, parameters)
                    .ToListAsync();

                var reservedTemplateId = result.FirstOrDefault();

                if (reservedTemplateId != default(Guid))
                {
                    logger.LogInformation("Успешно зарезервирован шаблон с ID: {TemplateId} для UnitId: {UnitId}, JobId: {JobId}",
                        reservedTemplateId, newUnitId, newJobId);
                }
                else
                {
                    logger.LogDebug("Не удалось зарезервировать шаблон для UnitId: {UnitId}, JobId: {JobId} (не найдено подходящих конфигураций роботов)", newUnitId, newJobId);
                }

                return reservedTemplateId;
            }
            catch (Exception ex)
            {
                logger.LogError(ex, "Ошибка при резервировании шаблона для UnitId: {UnitId}, JobId: {JobId}", newUnitId, newJobId);
                throw;
            }
        }
    }
}