RobotTaskController.cs 3.74 KB
using Microsoft.AspNetCore.Mvc;
using Rcs.Application.Common;
using Rcs.Application.DTOs;
using MassTransit.Mediator;
using Rcs.Application.MessageBus.Commands;
using StatusCodes = Microsoft.AspNetCore.Http.StatusCodes;

namespace Rcs.Api.Controllers
{
    /// <summary>
    /// 任务管理控制器
    /// @author zzy
    /// </summary>
    [ApiController]
    [Route("api/[controller]")]
    public class RobotTaskController : ControllerBase
    {
        private readonly ILogger<RobotTaskController> _logger;
        private readonly IMediator _mediator;

        public RobotTaskController(
            ILogger<RobotTaskController> logger,
            IMediator mediator)
        {
            _logger = logger ?? throw new ArgumentNullException(nameof(logger));
            _mediator = mediator;
        }

        /// <summary>
        /// 获取任务列表
        /// </summary>
        /// <param name="query">查询参数</param>
        /// <param name="cancellationToken">取消令牌</param>
        /// <returns>任务列表</returns>
        [HttpGet("list")]
        public async Task<IActionResult> GetRobotTaskList([FromQuery] GetRobotTasksQuery query, CancellationToken cancellationToken = default)
        {
            var client = _mediator.CreateRequestClient<GetRobotTasksQuery>();
            var response = await client.GetResponse<PagedResponse<RobotTaskListItemDto>>(query, cancellationToken);
            return Ok(response.Message);
        }

        /// <summary>
        /// 根据ID获取任务详情
        /// </summary>
        /// <param name="id">任务ID</param>
        /// <param name="cancellationToken">取消令牌</param>
        /// <returns>任务详情</returns>
        [HttpGet("{id}")]
        public async Task<IActionResult> GetRobotTaskById(string id, CancellationToken cancellationToken = default)
        {
            if (!Guid.TryParse(id, out Guid taskId))
            {
                return BadRequest(ApiResponse.Failed("无效的任务ID格式", StatusCodes.Status400BadRequest));
            }
            var client = _mediator.CreateRequestClient<GetRobotTaskQuery>();
            var response = await client.GetResponse<ApiResponse<RobotTaskDto>>(new GetRobotTaskQuery()
            {
                TaskId = taskId
            }, cancellationToken);

            return Ok(response.Message);
        }

        /// <summary>
        /// 创建或更新任务
        /// </summary>
        /// <param name="command">创建或更新命令</param>
        /// <returns>操作结果</returns>
        [HttpPost]
        public async Task<IActionResult> CreateOrUpdateRobotTask([FromBody] CreateOrUpdateRobotTaskCommand command)
        {
            var client = _mediator.CreateRequestClient<CreateOrUpdateRobotTaskCommand>();
            var response = await client.GetResponse<ApiResponse>(command);

            return Ok(response.Message);
        }

        /// <summary>
        /// 删除任务
        /// </summary>
        /// <param name="id">任务ID</param>
        /// <param name="cancellationToken">取消令牌</param>
        /// <returns>操作结果</returns>
        [HttpDelete("{id}")]
        public async Task<IActionResult> DeleteRobotTaskById(string id, CancellationToken cancellationToken = default)
        {
            if (!Guid.TryParse(id, out Guid taskId))
            {
                return BadRequest(ApiResponse.Failed("无效的任务ID格式", StatusCodes.Status400BadRequest));
            }
            var client = _mediator.CreateRequestClient<DeleteRobotTaskCommand>();
            var response = await client.GetResponse<ApiResponse>(new DeleteRobotTaskCommand()
            {
                TaskId = taskId
            }, cancellationToken);

            return Ok(response.Message);
        }
    }
}