KxsConfigNew/Middleware/GlobalExceptionMiddleware.cs

using Attribute;
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
{
    /// <summary>
    /// 全局异常处理中间件
    /// 调用 app.UseMiddlewareGlobalExceptionMiddleware>();
    /// </summary>
    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}";
                Console.Write("500错误：" + ex.ToString());
                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<LogAttribute>();
                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<IEndpointFeature>()?.Endpoint;
        }
    }
}