104 lines
3.8 KiB
C#
104 lines
3.8 KiB
C#
|
using System.Diagnostics;
|
|||
|
|
using System.Text;
|
|||
|
|
|
|||
|
|
namespace Gatedge.ScanCode.Middleware
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 日志中间件
|
|||
|
|
/// </summary>
|
|||
|
|
public class ResponseLoggingMiddleware
|
|||
|
|
{
|
|||
|
|
private readonly RequestDelegate _next;
|
|||
|
|
private readonly ILogger<ResponseLoggingMiddleware> _logger;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 构造函数
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="next"></param>
|
|||
|
|
/// <param name="logger"></param>
|
|||
|
|
public ResponseLoggingMiddleware(RequestDelegate next, ILogger<ResponseLoggingMiddleware> logger)
|
|||
|
|
{
|
|||
|
|
_next = next;
|
|||
|
|
_logger = logger;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 中间件处理函数
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="context"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public async Task Invoke(HttpContext context)
|
|||
|
|
{
|
|||
|
|
// 访问路径
|
|||
|
|
var path = context.Request.Path;
|
|||
|
|
// 记录请求开始时间
|
|||
|
|
var startTime = Stopwatch.GetTimestamp();
|
|||
|
|
// 获取客户端 IP 地址（处理反向代理场景）
|
|||
|
|
var ipAddress = context.Connection.RemoteIpAddress?.ToString();
|
|||
|
|
if (context.Request.Headers.TryGetValue("X-Forwarded-For", out var forwardedFor))
|
|||
|
|
{
|
|||
|
|
ipAddress = forwardedFor.ToString();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 记录请求体（JSON）
|
|||
|
|
string RequestBody = string.Empty;
|
|||
|
|
if (context.Request.ContentType?.Contains("application/json") == true)
|
|||
|
|
{
|
|||
|
|
context.Request.EnableBuffering();
|
|||
|
|
using var reader = new StreamReader(context.Request.Body, Encoding.UTF8, true, 1024, true);
|
|||
|
|
RequestBody = await reader.ReadToEndAsync();
|
|||
|
|
context.Request.Body.Position = 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 捕获原始响应流
|
|||
|
|
var originalBodyStream = context.Response.Body;
|
|||
|
|
// 使用内存流捕获响应
|
|||
|
|
using var responseBody = new MemoryStream();
|
|||
|
|
context.Response.Body = responseBody;
|
|||
|
|
|
|||
|
|
// 调用下一个中间件
|
|||
|
|
await _next(context);
|
|||
|
|
|
|||
|
|
// 记录响应体
|
|||
|
|
responseBody.Seek(0, SeekOrigin.Begin);
|
|||
|
|
var bodyText = await new StreamReader(responseBody, Encoding.Default, true, 1024, true).ReadToEndAsync();
|
|||
|
|
|
|||
|
|
|
|||
|
|
// 将响应复制回原始流
|
|||
|
|
responseBody.Seek(0, SeekOrigin.Begin);
|
|||
|
|
await responseBody.CopyToAsync(originalBodyStream);
|
|||
|
|
context.Response.Body = originalBodyStream;
|
|||
|
|
|
|||
|
|
// 记录请求结束时间
|
|||
|
|
var endTime = Stopwatch.GetTimestamp();
|
|||
|
|
|
|||
|
|
// 计算请求耗时
|
|||
|
|
var elapsed = (double)(endTime - startTime) / Stopwatch.Frequency * 1000.000;
|
|||
|
|
// 记录日志
|
|||
|
|
// _logger.LogInformation(
|
|||
|
|
//@"Request: {Method} {Path} | IP: {IP} | Status: {StatusCode} | Time: {Elapsed}ms
|
|||
|
|
//RequestBody: {Request}
|
|||
|
|
//ResponseBody: {Response}
|
|||
|
|
//================================================================================",
|
|||
|
|
// context.Request.Method,
|
|||
|
|
// context.Request.Path,
|
|||
|
|
// ipAddress,
|
|||
|
|
// context.Response.Body,
|
|||
|
|
// elapsed,
|
|||
|
|
// RequestBody,
|
|||
|
|
// bodyText
|
|||
|
|
// );
|
|||
|
|
|
|||
|
|
// 记录日志
|
|||
|
|
|
|||
|
|
var method = context.Request.Method;
|
|||
|
|
var statusCode = context.Response.Body;
|
|||
|
|
_logger.LogInformation(
|
|||
|
|
@$"Request: {method} {path} | IP: {ipAddress} | Status: {statusCode} | Time: {elapsed}ms
|
|||
|
|
RequestBody: {RequestBody}
|
|||
|
|
================================================================================"
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|