CopyTaskTemplateCommandHandler.cs 2.26 KB
using MassTransit;
using Microsoft.Extensions.Logging;
using Rcs.Application.Common;
using Rcs.Application.MessageBus.Commands;
using Rcs.Domain.Entities;
using Rcs.Domain.Repositories;

namespace Rcs.Infrastructure.MessageBus.Handlers.Commands;

public class CopyTaskTemplateCommandHandler : IConsumer<CopyTaskTemplateCommand>
{
    private readonly ILogger<CopyTaskTemplateCommandHandler> _logger;
    private readonly ITaskTemplateRepository _taskTemplateRepository;

    public CopyTaskTemplateCommandHandler(
        ILogger<CopyTaskTemplateCommandHandler> logger,
        ITaskTemplateRepository taskTemplateRepository)
    {
        _logger = logger;
        _taskTemplateRepository = taskTemplateRepository;
    }

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

        try
        {
            var sourceTemplate = await _taskTemplateRepository.GetWithFullDetailsAsync(
                command.TemplateId,
                context.CancellationToken);

            if (sourceTemplate == null)
            {
                await context.RespondAsync(ApiResponse.Failed("未找到指定的任务模板"));
                return;
            }

            var suffix = $"_C{DateTime.Now:MMddHHmmss}";
            var maxBaseLength = 50 - suffix.Length;
            var baseCode = sourceTemplate.TemplateCode.Length > maxBaseLength 
                ? sourceTemplate.TemplateCode.Substring(0, maxBaseLength) 
                : sourceTemplate.TemplateCode;
            var newTemplateCode = $"{baseCode}{suffix}";

            var newTemplate = TaskTemplate.Create(
                newTemplateCode,
                $"{sourceTemplate.TemplateName} (副本)",
                sourceTemplate.RobotType,
                sourceTemplate.Description,
                sourceTemplate.Manufacturer,
                sourceTemplate.IsEnabled
            );

            await _taskTemplateRepository.AddAsync(newTemplate, context.CancellationToken);
            await _taskTemplateRepository.SaveChangesAsync(context.CancellationToken);

            await context.RespondAsync(ApiResponse.Successful());
        }
        catch (Exception ex)
        {
            await context.RespondAsync(ApiResponse.Failed(ex.Message));
        }
    }
}