dengjunjie
21 小时以前 90ac374a64e47e2bd97ee9245f4cafb50ad5dbbb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Logging;
using Newtonsoft.Json;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Reflection;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using WIDESEAWCS_Core.Helper;
using WIDESEAWCS_Core.LogHelper;
 
namespace WIDESEAWCS_Core.Middlewares
{
    /// <summary>
    /// 记录请求和响应数据
    /// </summary>
    public class ApiLogMiddlewareNew
    {
        /// <summary>
        /// 
        /// </summary>
        private readonly RequestDelegate _next;
 
        public ApiLogMiddlewareNew(RequestDelegate next, ILogger<ApiLogMiddlewareNew> logger)
        {
            _next = next;
        }
 
        //todo
        public async Task InvokeAsync(HttpContext context)
        {
            if (App.ExpDateTime != null && (DateTime.Now - App.ExpDateTime.GetValueOrDefault()).TotalSeconds > 0)
            {
                context.Response.StatusCode = HttpStatusCode.InternalServerError.ObjToInt();
                context.Response.ContentType = "application/json";
 
                var json = new WebResponseContent();
 
                json.Message = HttpStatusCode.InternalServerError.ToString();//错误信息
                json.Code = 500;//500异常 
 
                StreamWriter streamWriter = new StreamWriter(context.Response.Body);
                await streamWriter.WriteAsync(json.Serialize());
                return;
            }
 
            // 过滤,只有接口
            if (context.Request.Path.Value?.Contains("api") ?? false)
            {
                context.Request.EnableBuffering();
                Stream originalBody = context.Response.Body;
                string requestParam = string.Empty;
                string responseParam = string.Empty;
                try
                {
                    (context.RequestServices.GetService(typeof(RequestLogModel)) as RequestLogModel).RequestDate = DateTime.Now;
                    try
                    {
                        // 存储请求数据
                        requestParam = RequestDataLog(context);
                        context.Request.Body.Position = 0;
                    }
                    catch { }
                    using MemoryStream ms = new();
                    context.Response.Body = ms;
 
                    await _next(context);
 
                    try
                    {
                        // 存储响应数据
                        responseParam = ResponseDataLog(context.Response);
                    }
                    catch { }
 
                    ms.Position = 0;
                    await ms.CopyToAsync(originalBody);
                    #region 过滤接口
                    string path = context.Request.Path.Value;
                    if (!string.IsNullOrEmpty(path))
                    {
                        List<string> RequestPaths = new List<string>();
                        string RequestPath = AppSettings.Configuration[nameof(RequestPath)];
                        if (!string.IsNullOrEmpty(RequestPath)) RequestPaths = RequestPath.Split(',').ToList();
                        bool containsAny = RequestPaths.Any(k => path.Contains(k, StringComparison.OrdinalIgnoreCase));
                        if (!containsAny) Logger.Add(requestParam, responseParam);//记录系统日志
                        //记录接口日志
                        var ignoreKeywords = new HashSet<string>(StringComparer.OrdinalIgnoreCase) { "get" };
                        containsAny = RequestPaths.Where(k => !ignoreKeywords.Contains(k))
                           .Any(k => path.Contains(k, StringComparison.OrdinalIgnoreCase));
                        if (containsAny && context != null) LoggerNew.Add(requestParam, responseParam, context);
                    }
                    #endregion
                    //if (!(context.Request.Path.Value?.Contains("get") ?? true))
                    //    Logger.Add(requestParam, responseParam);
                }
                catch (Exception ex)
                {
                    // 记录异常
 
                }
                finally
                {
                    context.Response.Body = originalBody;
                }
            }
            else
            {
                await _next(context);
            }
        }
 
        private string RequestDataLog(HttpContext context)
        {
            var request = context.Request;
            var sr = new StreamReader(request.Body);
 
            object obj = new
            {
                QueryString = request.QueryString.ToString(),
                BodyData = sr.ReadToEndAsync().Result
            };
 
            string data = JsonConvert.SerializeObject(obj);
 
            request.Body.Position = 0;
 
            return data;
        }
 
        private string ResponseDataLog(HttpResponse response)
        {
            response.Body.Position = 0;
            using StreamReader stream = new StreamReader(response.Body, leaveOpen: true);
            string body = stream.ReadToEnd();
            response.Body.Position = 0;
            return body;
        }
    }
}