using Attribute;
using Common;
using Extensions;
using Infrastructure;
using Infrastructure.Model;
using IPTools.Core;
using Microsoft.AspNetCore.Http.Features;
using Model;
using NLog;
using Services;
using System.Text.Encodings.Web;
using textJson = System.Text.Json;
namespace Middleware
{
///
/// 全局异常处理中间件
/// 调用 app.UseMiddlewareGlobalExceptionMiddleware>();
///
public class GlobalExceptionMiddleware
{
private readonly RequestDelegate next;
// private readonly ISysOperLogService SysOperLogService;
static readonly Logger Logger = LogManager.GetCurrentClassLogger();
public GlobalExceptionMiddleware(RequestDelegate next)
{
this.next = next;
// this.SysOperLogService = sysOperLog;
}
public async Task Invoke(HttpContext context)
{
try
{
// 设置允许跨域的域名、方法等
context.Response.Headers.Add("Access-Control-Allow-Origin", "*");
context.Response.Headers.Add("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS");
context.Response.Headers.Add("Access-Control-Allow-Headers", "DNT,X-Mx-ReqToken,Keep-Alive,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type,Authorization");
// 如果请求方法是预检请求(OPTIONS),则直接返回成功状态码
if (context.Request.Method == "OPTIONS")
{
context.Response.StatusCode = 200;
await context.Response.WriteAsync("OK");
return;
}
await next(context);
}
catch (Exception ex)
{
await HandleExceptionAsync(context, ex);
}
}
private async Task HandleExceptionAsync(HttpContext context, Exception ex)
{
NLog.LogLevel logLevel = NLog.LogLevel.Info;
int code = (int)ResultCode.GLOBAL_ERROR;
string msg;
string error = string.Empty;
bool notice = true;
//自定义异常
if (ex is CustomException customException)
{
code = customException.Code;
msg = customException.Message;
error = customException.LogMsg;
notice = customException.Notice;
}
else if (ex is ArgumentException)//参数异常
{
code = (int)ResultCode.PARAM_ERROR;
msg = ex.Message;
}
else
{
msg = "服务器好像出了点问题,请联系系统管理员...";
error = $"{ex.Message}";
Function.WriteLog(DateTime.Now.ToString() + "\n" + ex.ToString(), "Global全局异常");
logLevel = NLog.LogLevel.Error;
context.Response.StatusCode = 500;
}
var options = new textJson.JsonSerializerOptions
{
Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping,
PropertyNamingPolicy = textJson.JsonNamingPolicy.CamelCase,
WriteIndented = true
};
ApiResult apiResult = new(code, msg);
string responseResult = textJson.JsonSerializer.Serialize(apiResult, options);
// string ip = HttpContextExtension.GetClientUserIp(context);
// var ip_info = IpTool.Search(ip);
// SysOperLog sysOperLog = new()
// {
// Status = 1,
// // OperIp = ip,
// OperUrl = HttpContextExtension.GetRequestUrl(context),
// RequestMethod = context.Request.Method,
// JsonResult = responseResult,
// ErrorMsg = string.IsNullOrEmpty(error) ? msg : error,
// OperName = HttpContextExtension.GetName(context),
// // OperLocation = ip_info.Province + " " + ip_info.City,
// OperTime = DateTime.Now,
// OperParam = HttpContextExtension.GetRequestValue(context, context.Request.Method)
// };
var endpoint = GetEndpoint(context);
if (endpoint != null)
{
var logAttribute = endpoint.Metadata.GetMetadata();
if (logAttribute != null)
{
// sysOperLog.BusinessType = (int)logAttribute.BusinessType;
// sysOperLog.Title = logAttribute?.Title;
// sysOperLog.OperParam = logAttribute.IsSaveRequestData ? sysOperLog.OperParam : "";
// sysOperLog.JsonResult = logAttribute.IsSaveResponseData ? sysOperLog.JsonResult : "";
}
}
LogEventInfo ei = new(logLevel, "GlobalExceptionMiddleware", error)
{
Exception = ex,
Message = error
};
ei.Properties["status"] = 1;//走正常返回都是通过走GlobalExceptionFilter不通过
ei.Properties["jsonResult"] = responseResult;
// ei.Properties["requestParam"] = sysOperLog.OperParam;
// ei.Properties["user"] = sysOperLog.OperName;
Logger.Log(ei);
context.Response.ContentType = "text/json;charset=utf-8";
await context.Response.WriteAsync(responseResult, System.Text.Encoding.UTF8);
// string errorMsg = $"> 操作人:{sysOperLog.OperName}" +
// $"\n> 操作地区:{sysOperLog.OperIp}({sysOperLog.OperLocation})" +
// $"\n> 操作模块:{sysOperLog.Title}" +
// $"\n> 操作地址:{sysOperLog.OperUrl}" +
// $"\n> 错误信息:{msg}\n\n> {error}";
// SysOperLogService.InsertOperlog(sysOperLog);
if (!notice) return;
}
public static Endpoint GetEndpoint(HttpContext context)
{
if (context == null)
{
throw new ArgumentNullException(nameof(context));
}
return context.Features.Get()?.Endpoint;
}
}
}