CompletionRobotTaskCommandHandler.cs 1.79 KB
using MassTransit;
using Microsoft.Extensions.Logging;
using Rcs.Application.Common;
using Rcs.Application.MessageBus.Commands;
using Rcs.Domain.Repositories;

namespace Rcs.Infrastructure.MessageBus.Handlers.Commands;

/// <summary>
/// 删除任务命令处理器
/// @author zzy
/// </summary>
public class CompletionRobotTaskCommandHandler : IConsumer<CompletionRobotTaskCommand>
{
    private readonly ILogger<CompletionRobotTaskCommandHandler> _logger;
    private readonly IRobotTaskRepository _robotTaskRepository;
    private readonly IRobotRepository _robotRepository;

    public CompletionRobotTaskCommandHandler(
        ILogger<CompletionRobotTaskCommandHandler> logger,
        IRobotTaskRepository robotTaskRepository,
        IRobotRepository robotRepository)
    {
        _logger = logger;
        _robotTaskRepository = robotTaskRepository;
        _robotRepository = robotRepository;
    }

    public async Task Consume(ConsumeContext<CompletionRobotTaskCommand> context)
    {
        var command = context.Message;
        try
        {
            var task = await _robotTaskRepository.GetByRelationAsync(command.task_id, context.CancellationToken);
            if (task == null)
            {
                await context.RespondAsync(ApiResponse.Failed($"未找到关联单据为 {command.task_id} 的任务"));
                return;
            }
            var robot = await _robotRepository.GetBySerialNumberAsync(command.robot_code, context.CancellationToken);

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

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