ManualCompleteRobotTaskCommandHandler.cs 1.89 KB
using MassTransit;
using Microsoft.Extensions.Logging;
using Rcs.Application.Common;
using Rcs.Application.MessageBus.Commands;
using Rcs.Domain.Repositories;
using TaskStatus = Rcs.Domain.Entities.TaskStatus;

namespace Rcs.Infrastructure.MessageBus.Handlers.Commands;

/// <summary>
/// 手动完成任务命令处理器
/// </summary>
public class ManualCompleteRobotTaskCommandHandler : IConsumer<ManualCompleteRobotTaskCommand>
{
    private readonly ILogger<ManualCompleteRobotTaskCommandHandler> _logger;
    private readonly IRobotTaskRepository _robotTaskRepository;

    public ManualCompleteRobotTaskCommandHandler(
        ILogger<ManualCompleteRobotTaskCommandHandler> logger,
        IRobotTaskRepository robotTaskRepository)
    {
        _logger = logger;
        _robotTaskRepository = robotTaskRepository;
    }

    public async Task Consume(ConsumeContext<ManualCompleteRobotTaskCommand> context)
    {
        var command = context.Message;
        try
        {
            var task = await _robotTaskRepository.GetByIdAsync(command.TaskId, context.CancellationToken);
            if (task == null)
            {
                await context.RespondAsync(ApiResponse.Failed($"未找到任务ID为 {command.TaskId} 的任务"));
                return;
            }

            if (task.Status == TaskStatus.Completed)
            {
                await context.RespondAsync(ApiResponse.Successful("任务已完成"));
                return;
            }

            task.Completed(task.RobotId);
            await _robotTaskRepository.UpdateAsync(task, context.CancellationToken);

            await context.RespondAsync(ApiResponse.Successful("手动完成成功"));
        }
        catch (Exception ex)
        {
            _logger.LogError(ex, "手动完成任务失败, TaskId: {TaskId}", command.TaskId);
            await context.RespondAsync(ApiResponse.Failed(ex.Message));
        }
    }
}