88 lines
2.9 KiB
C#
88 lines
2.9 KiB
C#
|
|
using LiveForum.Code.Base;
|
|
using LiveForum.Code.ExceptionExtend;
|
|
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
using Newtonsoft.Json;
|
|
using Newtonsoft.Json.Serialization;
|
|
|
|
using System;
|
|
|
|
|
|
namespace LiveForum.Code.MiddlewareExtend
|
|
{
|
|
/// <summary>
|
|
/// 异常中间件
|
|
/// </summary>
|
|
public class ExceptionMiddleware
|
|
{
|
|
private readonly RequestDelegate _next;
|
|
private readonly ILogger<ExceptionMiddleware> _logger;
|
|
|
|
public ExceptionMiddleware(RequestDelegate next, ILogger<ExceptionMiddleware> logger)
|
|
{
|
|
_next = next;
|
|
_logger = logger;
|
|
}
|
|
|
|
public async Task Invoke(HttpContext context)
|
|
{
|
|
try
|
|
{
|
|
await _next(context);
|
|
}
|
|
catch (ArgumentNullException ex)
|
|
{
|
|
|
|
_logger.LogError($"ArgumentNullException异常记录=>{context.Request.Path.ToString()},{ex.Message}", ex);
|
|
// 如果响应已经开始了，就不能再修改响应头或写入内容
|
|
if (context.Response.HasStarted)
|
|
{
|
|
return;
|
|
}
|
|
|
|
|
|
context.Response.StatusCode = StatusCodes.Status200OK;
|
|
BaseResponse<object> baseResponse = new BaseResponse<object>(ResponseCode.ParamError, ex.ParamName ?? "参数错误", null)
|
|
{
|
|
|
|
};
|
|
context.Response.ContentType = "application/json; charset=utf-8";
|
|
// 将异常信息写入 HTTP 响应
|
|
await context.Response.WriteAsync(baseResponse.ToString());
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
|
|
_logger.LogError($"Exception异常记录=>{context.Request.Path.ToString()},{ex.Message}", ex);
|
|
// 如果响应已经开始了，就不能再修改响应头或写入内容
|
|
if (context.Response.HasStarted)
|
|
{
|
|
return;
|
|
}
|
|
var box = ex as MessageBox;
|
|
if (box != null)
|
|
{
|
|
context.Response.ContentType = "application/json; charset=utf-8";
|
|
context.Response.StatusCode = StatusCodes.Status200OK;
|
|
// 将异常信息写入 HTTP 响应
|
|
await context.Response.WriteAsync(box.BaseResponse.ToString());
|
|
return;
|
|
}
|
|
// 设置 HTTP 响应状态码为 500
|
|
context.Response.ContentType = "application/json; charset=utf-8";
|
|
context.Response.StatusCode = StatusCodes.Status200OK;
|
|
BaseResponse<object> baseResponse = new BaseResponse<object>(ResponseCode.Error, ex.Message, null);
|
|
// 将异常信息写入 HTTP 响应
|
|
await context.Response.WriteAsync(baseResponse.ToString());
|
|
}
|
|
finally
|
|
{
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|