添加项目文件。

CoreCms.Net.Middlewares/RecordAccessLogsMildd.cs

@@ -0,0 +1,162 @@
+
+using Microsoft.AspNetCore.Http;
+using Microsoft.Extensions.Logging;
+using Newtonsoft.Json;
+using System;
+using System.Diagnostics;
+using System.IO;
+using System.Text;
+using System.Threading.Tasks;
+using System.Web;
+using CoreCms.Net.Auth.HttpContextUser;
+using CoreCms.Net.Configuration;
+using CoreCms.Net.Loging;
+using CoreCms.Net.Utility.Extensions;
+
+namespace CoreCms.Net.Middlewares
+{
+    /// <summary>
+    /// 中间件
+    /// 记录用户方访问数据
+    /// </summary>
+    public class RecordAccessLogsMildd
+    {
+        /// <summary>
+        ///
+        /// </summary>
+        private readonly RequestDelegate _next;
+        private readonly IHttpContextUser _user;
+
+        private readonly ILogger<RecordAccessLogsMildd> _logger;
+        private readonly Stopwatch _stopwatch;
+
+        /// <summary>
+        /// 构造
+        /// </summary>
+        /// <param name="next"></param>
+        /// <param name="user"></param>
+        /// <param name="logger"></param>
+        public RecordAccessLogsMildd(RequestDelegate next, IHttpContextUser user, ILogger<RecordAccessLogsMildd> logger)
+        {
+            _next = next;
+            _user = user;
+            _logger = logger;
+            _stopwatch = new Stopwatch();
+        }
+
+        public async Task InvokeAsync(HttpContext context)
+        {
+            if (AppSettingsConstVars.MiddlewareRecordAccessLogsEnabled)
+            {
+                var api = context.Request.Path.ObjectToString().TrimEnd('/').ToLower();
+                var ignoreApis = AppSettingsConstVars.MiddlewareRecordAccessLogsIgnoreApis;
+
+                // 过滤，只有接口
+                if (api.Contains("api") && !ignoreApis.Contains(api))
+                {
+                    _stopwatch.Restart();
+                    var userAccessModel = new UserAccessModel();
+
+                    HttpRequest request = context.Request;
+
+                    userAccessModel.API = api;
+                    userAccessModel.User = _user.Name;
+                    userAccessModel.IP = IPLogMildd.GetClientIp(context);
+                    userAccessModel.BeginTime = DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss");
+                    userAccessModel.RequestMethod = request.Method;
+                    userAccessModel.Agent = request.Headers["User-Agent"].ObjectToString();
+
+
+                    // 获取请求body内容
+                    if (request.Method.ToLower().Equals("post") || request.Method.ToLower().Equals("put"))
+                    {
+                        // 启用倒带功能，就可以让 Request.Body 可以再次读取
+                        request.EnableBuffering();
+
+                        Stream stream = request.Body;
+                        byte[] buffer = new byte[request.ContentLength.Value];
+                        stream.Read(buffer, 0, buffer.Length);
+                        userAccessModel.RequestData = Encoding.UTF8.GetString(buffer);
+
+                        request.Body.Position = 0;
+                    }
+                    else if (request.Method.ToLower().Equals("get") || request.Method.ToLower().Equals("delete"))
+                    {
+                        userAccessModel.RequestData = HttpUtility.UrlDecode(request.QueryString.ObjectToString(), Encoding.UTF8);
+                    }
+
+                    // 获取Response.Body内容
+                    var originalBodyStream = context.Response.Body;
+                    using (var responseBody = new MemoryStream())
+                    {
+                        context.Response.Body = responseBody;
+
+                        await _next(context);
+
+                        var responseBodyData = await GetResponse(context.Response);
+
+                        await responseBody.CopyToAsync(originalBodyStream);
+                    }
+
+
+                    var dt = DateTime.Now;
+
+                    // 响应完成记录时间和存入日志
+                    context.Response.OnCompleted(() =>
+                    {
+                        _stopwatch.Stop();
+
+                        userAccessModel.OPTime = _stopwatch.ElapsedMilliseconds + "ms";
+
+                        // 自定义log输出
+                        var requestInfo = JsonConvert.SerializeObject(userAccessModel);
+                        Parallel.For(0, 1, e =>
+                        {
+                            LogLockHelper.OutSql2Log("RecordAccessLogs", "RecordAccessLogs" + dt.ToString("yyyy-MM-dd-HH"), new string[] { requestInfo + "," }, false);
+                        });
+
+                        return Task.CompletedTask;
+                    });
+
+                }
+                else
+                {
+                    await _next(context);
+                }
+            }
+            else
+            {
+                await _next(context);
+            }
+        }
+
+
+        /// <summary>
+        /// 获取响应内容
+        /// </summary>
+        /// <param name="response"></param>
+        /// <returns></returns>
+        public async Task<string> GetResponse(HttpResponse response)
+        {
+            response.Body.Seek(0, SeekOrigin.Begin);
+            var text = await new StreamReader(response.Body).ReadToEndAsync();
+            response.Body.Seek(0, SeekOrigin.Begin);
+            return text;
+        }
+    }
+
+    public class UserAccessModel
+    {
+        public string User { get; set; }
+        public string IP { get; set; }
+        public string API { get; set; }
+        public string BeginTime { get; set; }
+        public string OPTime { get; set; }
+        public string RequestMethod { get; set; }
+        public string RequestData { get; set; }
+        public string Agent { get; set; }
+
+    }
+
+}
+