CoreAgent/CoreAgent.API/Controllers/BaseApiController.cs

using CoreAgent.Domain.Exceptions;
using CoreAgent.Domain.Models.Common;
using MediatR;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Logging;
using System.Net;

namespace CoreAgent.API.Controllers
{
    [ApiController]
    [Route("api/v{version:apiVersion}/[controller]")]
    public abstract class BaseApiController : ControllerBase
    {
        protected readonly IMediator _mediator;
        protected readonly ILogger<BaseApiController> _logger;

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

        protected async Task<IActionResult> HandleRequest<TRequest, TResponse>(TRequest request)
            where TRequest : IRequest<TResponse>
        {
            try
            {
                _logger.LogInformation("开始处理请求: {RequestType}", typeof(TRequest).Name);
                var response = await _mediator.Send(request);
                return response as IActionResult;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "处理请求时发生错误: {RequestType}", typeof(TRequest).Name);
                return HandleException<TResponse>(ex);
            }
        }

        protected async Task<IActionResult> HandleRequest<TRequest>(TRequest request)
            where TRequest : IRequest
        {
            try
            {
                _logger.LogInformation("Handling request of type {RequestType}", typeof(TRequest).Name);
                await _mediator.Send(request);
                return ApiActionResult.Ok();
            }
            catch (Exception ex)
            {
                return HandleException(ex);
            }
        }

        private IActionResult HandleException<T>(Exception ex)
        {
            return ex switch
            {
                ValidationException validationEx => 
                    ApiActionResult<T>.Error(validationEx.Message, "VALIDATION_ERROR", HttpStatusCode.BadRequest),
                NotFoundException notFoundEx => 
                    ApiActionResult<T>.Error(notFoundEx.Message, "NOT_FOUND", HttpStatusCode.NotFound),
                UnauthorizedAccessException => 
                    ApiActionResult<T>.Error("您没有权限访问此资源", "UNAUTHORIZED", HttpStatusCode.Unauthorized),
                _ => ApiActionResult<T>.Error(
                    "服务器内部错误，请稍后重试", 
                    "INTERNAL_ERROR", 
                    HttpStatusCode.InternalServerError)
            };
        }

        private IActionResult HandleException(Exception ex)
        {
            _logger.LogError(ex, "Error handling request");
            return ApiActionResult.Error(ex.Message, "INTERNAL_ERROR", HttpStatusCode.InternalServerError);
        }

        protected IActionResult Success<T>(T data, string message = "操作成功") => 
            ApiActionResult<T>.Ok(data, message);

        protected IActionResult Success(string message = "操作成功") => 
            ApiActionResult.Ok(message);

        protected IActionResult Error(string message, string errorCode = "ERROR", HttpStatusCode statusCode = HttpStatusCode.BadRequest) => 
            ApiActionResult.Error(message, errorCode, statusCode);

        protected IActionResult Error<T>(string message, string errorCode = "ERROR", HttpStatusCode statusCode = HttpStatusCode.BadRequest) => 
            ApiActionResult<T>.Error(message, errorCode, statusCode);
    }
}