fix namesapce

2020-08-05 15:29:56 +08:00
parent f1138a5a10
commit 2abe486151
17 changed files with 117 additions and 120 deletions
@@ -0,0 +1,57 @@
+using System;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Http;
+using SimApi.Communications;
+using Microsoft.Extensions.Logging;
+using SimApi.Exceptions;
+
+namespace SimApi.Middlewares
+{
+    /// <summary>
+    /// 异常处理中间件
+    /// </summary>
+    public class SimApiExceptionMiddleware
+    {
+        private RequestDelegate Next { get; }
+        private ILogger<SimApiExceptionMiddleware> Log { get; }
+
+        public SimApiExceptionMiddleware(RequestDelegate next, ILogger<SimApiExceptionMiddleware> log)
+        {
+            Log = log;
+            Next = next;
+        }
+
+        public async Task InvokeAsync(HttpContext context)
+        {
+            var response = new SimApiBaseResponse();
+            try
+            {
+                await Next(context);
+            }
+            catch (SimApiException ex)
+            {
+                response.SetCodeMsg(ex.Code, ex.Message);
+                ErrorResponse(context, response);
+            }
+            catch (Exception ex)
+            {
+                Log.LogError(ex.Message);
+                Log.LogError(ex.StackTrace);
+                response.SetCodeMsg(500, ex.Message);
+                ErrorResponse(context, response);
+            }
+        }
+
+        /// <summary>
+        /// 异常抛出错误
+        /// </summary>
+        /// <param name="context"></param>
+        /// <param name="response"></param>
+        private void ErrorResponse(HttpContext context, SimApiBaseResponse response)
+        {
+            context.Response.StatusCode = 200;
+            context.Response.Headers.Add("Content-Type", "application/json");
+            context.Response.WriteAsync(response.ToString());
+        }
+    }
+}