87 lines
2.6 KiB
C#
87 lines
2.6 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Http;
|
|
using SimApi.Communications;
|
|
using Microsoft.Extensions.Logging;
|
|
using SimApi.Configurations;
|
|
using SimApi.Exceptions;
|
|
using SimApi.Helpers;
|
|
|
|
namespace SimApi.Middlewares;
|
|
|
|
/// <summary>
|
|
/// 异常处理中间件
|
|
/// </summary>
|
|
public class SimApiExceptionMiddleware(
|
|
RequestDelegate next,
|
|
ILogger<SimApiExceptionMiddleware> log,
|
|
SimApiOptions simApiOptions)
|
|
{
|
|
public async Task InvokeAsync(HttpContext context)
|
|
{
|
|
if (context.Request.Headers.TryGetValue("Query-Id", out var header))
|
|
{
|
|
context.Response.Headers["Query-Id"] = header;
|
|
}
|
|
|
|
try
|
|
{
|
|
await next(context);
|
|
if (!context.Response.HasStarted)
|
|
{
|
|
SimApiError.ErrorWhenFalse(
|
|
simApiOptions.SimApiExceptionOptions.SkipStatusCodes.Contains(context.Response.StatusCode),
|
|
context.Response.StatusCode);
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
// 解包异步异常
|
|
ex = UnwrapAggregateException(ex);
|
|
SimApiBaseResponse response;
|
|
if (ex is SimApiException simEx)
|
|
{
|
|
response = string.IsNullOrEmpty(simEx.Message)
|
|
? new SimApiBaseResponse(simEx.Code)
|
|
: new SimApiBaseResponse(simEx.Code, simEx.Message);
|
|
if (context.Response.StatusCode == 404)
|
|
{
|
|
response.Message = "接口不存在";
|
|
}
|
|
}
|
|
else
|
|
{
|
|
log.LogError(ex, ex.Message);
|
|
response = new SimApiBaseResponse(500);
|
|
}
|
|
|
|
await ErrorResponseAsync(context, response);
|
|
}
|
|
}
|
|
|
|
private static Exception UnwrapAggregateException(Exception ex)
|
|
{
|
|
while (ex is AggregateException aggEx && aggEx.InnerException != null)
|
|
{
|
|
ex = aggEx.InnerException;
|
|
}
|
|
|
|
return ex;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 异步输出错误响应（修复异步异常捕获核心）
|
|
/// </summary>
|
|
private static async Task ErrorResponseAsync(HttpContext context, SimApiBaseResponse response)
|
|
{
|
|
// 响应已开始则直接返回，不修改
|
|
if (context.Response.HasStarted)
|
|
return;
|
|
|
|
context.Response.StatusCode = 200;
|
|
context.Response.ContentType = "application/json";
|
|
context.Response.ContentLength = null; // 清除可能已设置的 Content-Length
|
|
await context.Response.WriteAsync(response.ToString());
|
|
}
|
|
} |