GetTaskTemplatesQueryHandler.cs 2.58 KB
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using AutoMapper;
using MassTransit;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Rcs.Application.Common;
using Rcs.Application.Dtos;
using Rcs.Application.DTOs;
using Rcs.Application.MessageBus.Commands;
using Rcs.Domain.Entities;
using Rcs.Domain.Repositories;

namespace Rcs.Infrastructure.MessageBus.Handlers.Commands;

public class GetTaskTemplatesQueryHandler : IConsumer<GetTaskTemplatesQuery>
{
    private readonly ILogger<GetTaskTemplatesQueryHandler> _logger;
    private readonly ITaskTemplateRepository _taskTemplateRepository;
    private readonly IMapper _mapper;

    public GetTaskTemplatesQueryHandler(
        ILogger<GetTaskTemplatesQueryHandler> logger,
        ITaskTemplateRepository taskTemplateRepository,
        IMapper mapper)
    {
        _logger = logger;
        _taskTemplateRepository = taskTemplateRepository;
        _mapper = mapper;
    }

    public async Task Consume(ConsumeContext<GetTaskTemplatesQuery> context)
    {
        var query = context.Message;

        var queryable = _taskTemplateRepository.GetQueryable();

        if (!string.IsNullOrWhiteSpace(query.FilterModel))
        {
            queryable = FilterHelper.ApplyFilters(queryable, query.FilterModel);
        }

        if (!string.IsNullOrWhiteSpace(query.TemplateCode))
        {
            queryable = queryable.Where(m => m.TemplateCode.Contains(query.TemplateCode));
        }

        if (!string.IsNullOrWhiteSpace(query.TemplateName))
        {
            queryable = queryable.Where(m => m.TemplateName.Contains(query.TemplateName));
        }

        if (!string.IsNullOrWhiteSpace(query.RobotType))
        {
            if (Enum.TryParse<RobotType>(query.RobotType, out var robotType))
            {
                queryable = queryable.Where(m => m.RobotType == robotType);
            }
        }

        if (query.IsEnabled.HasValue)
        {
            queryable = queryable.Where(m => m.IsEnabled == query.IsEnabled.Value);
        }

        var totalCount = await queryable.CountAsync(context.CancellationToken);

        var templates = await queryable
            .OrderBy(m => m.TemplateCode)
            .Skip((query.PageNumber - 1) * query.PageSize)
            .Take(query.PageSize)
            .ToListAsync(context.CancellationToken);

        var templateDtos = _mapper.Map<List<TaskTemplateListItemDto>>(templates);
        await context.RespondAsync(PagedResponse<TaskTemplateListItemDto>.Successful(templateDtos, query.PageNumber, query.PageSize, totalCount));
    }
}