using System;
|
using System.IO;
|
using System.Linq;
|
using System.Threading.Tasks;
|
using System.Text;
|
using System.Threading;
|
using Microsoft.AspNetCore.Builder;
|
using Microsoft.AspNetCore.Http;
|
using System.Collections.Generic;
|
using Rhea.Common;
|
using System.Net;
|
using Tiger.IBusiness;
|
|
namespace Tiger.Api
|
{
|
/// <summary>
|
/// 日志中间件
|
/// </summary>
|
public class LogMiddleware
|
{
|
private readonly RequestDelegate _next;
|
private static readonly object LogMiddlewareLock = new object();
|
|
/// <summary>
|
/// 构造函数
|
/// </summary>
|
/// <param name="next"></param>
|
public LogMiddleware(RequestDelegate next)
|
{
|
_next = next;
|
}
|
|
/// <summary>
|
/// HttpContext响应
|
/// </summary>
|
/// <param name="context"></param>
|
/// <returns></returns>
|
public async Task InvokeAsync(HttpContext context)
|
{
|
var hcLog = new HttpContextLog(LogMiddlewareLock);//new object());//
|
hcLog.ID = $"{context.TraceIdentifier} at {DateTime.Now:yyyy/MM/dd HH:mm:ss.fff}";
|
hcLog.LocalHost = context.Request.Host.ToString();
|
hcLog.RemoteAddress = $"[{context.Connection.RemoteIpAddress}]:{context.Connection.RemotePort}";
|
|
HttpRequest request = context.Request;
|
hcLog.Url = request.Path.ToString();
|
hcLog.HeaderDic = request.Headers.ToDictionary(k => k.Key, v => string.Join(";", v.Value.ToList()));
|
hcLog.Method = request.Method;
|
hcLog.ExcuteStartTime = DateTime.Now;
|
|
if (hcLog.Url.StartsWith("/api") && !request.Method.ToLower().Equals("options"))
|
{
|
//获取Request.Body内容
|
if (request.Method.ToLower().Equals("post"))
|
{
|
request.EnableBuffering(); //启用多次读取
|
Stream stream = request.Body;
|
byte[] buffer = new byte[request.ContentLength.Value];
|
await request.Body.ReadAsync(buffer, 0, buffer.Length);
|
//_logInfo.RequestBody = Encoding.UTF8.GetString(buffer);
|
hcLog.RequestBody = Encoding.Default.GetString(buffer);
|
|
request.Body.Position = 0;
|
}
|
else if (request.Method.ToLower().Equals("get"))
|
{
|
hcLog.RequestBody = request.QueryString.Value;
|
}
|
|
//hcLog.LogRequest();
|
Task.Run(() => { hcLog.LogRequest(); });
|
// 响应完成记录时间和存入日志
|
context.Response.OnCompleted(() =>
|
{
|
hcLog.ExcuteEndTime = DateTime.Now;
|
//hcLog.LogResponse();
|
Task.Run(() => { hcLog.LogResponse(); });
|
return Task.CompletedTask;
|
});
|
|
//获取Response.Body内容
|
var originalBodyStream = context.Response.Body;
|
using (var responseBody = new MemoryStream())
|
{
|
try
|
{
|
context.Response.Body = responseBody;
|
|
await _next(context);
|
|
responseBody.Position = 0;
|
var responseContent = await new StreamReader(context.Response.Body).ReadToEndAsync();
|
hcLog.ResponseBody = responseContent;
|
hcLog.Status = $"{((HttpStatusCode)context.Response.StatusCode).ToString()}[{context.Response.StatusCode.ToString()}]";
|
}
|
catch (System.Exception ex)
|
{
|
Logger.Default.Fatal(ex, $"Request[{hcLog.ID}][{hcLog.RemoteAddress}][{hcLog.Url}]处理异常");
|
hcLog.ResponseBody = ex.Message;
|
hcLog.Status = $"{HttpStatusCode.InternalServerError}[{(int)HttpStatusCode.InternalServerError}]";
|
}
|
|
responseBody.Position = 0;
|
await responseBody.CopyToAsync(originalBodyStream);
|
}
|
}
|
else
|
{
|
await _next(context);
|
}
|
}
|
}
|
|
/// <summary>
|
/// HttpContext 日志
|
/// </summary>
|
public class HttpContextLog
|
{
|
/// <summary>
|
/// 构造函数
|
/// </summary>
|
/// <param name="writelock"></param>
|
public HttpContextLog(object writelock)
|
{
|
_writelock = writelock;
|
}
|
private object _writelock;
|
/// <summary>
|
/// ID
|
/// </summary>
|
public string ID { get; set; }
|
/// <summary>
|
/// LocalHost
|
/// </summary>
|
public string LocalHost { get; set; }
|
/// <summary>
|
/// RemoteAddress
|
/// </summary>
|
public string RemoteAddress { get; set; }
|
/// <summary>
|
/// Url
|
/// </summary>
|
public string Url { get; set; }
|
/// <summary>
|
/// Headers
|
/// </summary>
|
public IDictionary<string, string> HeaderDic { get; set; } = new Dictionary<string, string>();
|
/// <summary>
|
/// Header
|
/// </summary>
|
public string Header { get => $"[{string.Join(",", this.HeaderDic.Select(i => "{" + $"\"{i.Key}\":\"{i.Value}\"" + "}"))}"; }
|
/// <summary>
|
/// Method
|
/// </summary>
|
public string Method { get; set; }
|
/// <summary>
|
/// RequestBody
|
/// </summary>
|
public string RequestBody { get; set; }
|
/// <summary>
|
/// ResponseBody
|
/// </summary>
|
public string ResponseBody { get; set; }
|
/// <summary>
|
/// Status
|
/// </summary>
|
public string Status { get; set; }
|
/// <summary>
|
/// ExcuteStartTime
|
/// </summary>
|
public DateTime ExcuteStartTime { get; set; }
|
/// <summary>
|
/// ExcuteEndTime
|
/// </summary>
|
public DateTime ExcuteEndTime { get; set; }
|
/// <summary>
|
/// ElapsedTime
|
/// </summary>
|
public double ElapsedTime { get => (ExcuteEndTime - ExcuteStartTime).TotalSeconds; }
|
/// <summary>
|
/// 重构ToString
|
/// </summary>
|
/// <returns></returns>
|
public override string ToString()
|
{
|
string headers = "[" + string.Join(",", this.HeaderDic.Select(i => "{" + $"\"{i.Key}\":\"{i.Value}\"" + "}")) + "]";
|
return $"Url: {this.Url},\r\nHeaders: {headers},\r\nMethod: {this.Method},\r\nRequestBody: {this.RequestBody},\r\nResponseBody: {this.ResponseBody},\r\nExcuteStartTime: {this.ExcuteStartTime.ToString("yyyy-MM-dd HH:mm:ss.fff")},\r\nExcuteEndTime: {this.ExcuteEndTime.ToString("yyyy-MM-dd HH:mm:ss.fff")}";
|
}
|
/// <summary>
|
/// RequestToString
|
/// </summary>
|
/// <returns></returns>
|
public string LogRequest()
|
{
|
lock (_writelock)
|
{
|
ConsoleExt.WriteLine($"Request Form Address {RemoteAddress}|ID: {ID}", ConsoleColor.Yellow);
|
ConsoleExt.WriteLine("{");
|
ConsoleExt.Write($" Url: ");
|
ConsoleExt.Write(Url, ConsoleColor.Green);
|
ConsoleExt.Write($" (Method: ");
|
ConsoleExt.Write(Method, ConsoleColor.Green);
|
ConsoleExt.WriteLine(")");
|
ConsoleExt.Write($" Version: ");
|
ConsoleExt.WriteLine((HeaderDic.ContainsKey("version_id") ? HeaderDic["version_id"] : ""), ConsoleColor.Green);
|
ConsoleExt.Write($" RequestBody: ");
|
ConsoleExt.WriteLine(RequestBody, ConsoleColor.Gray);
|
ConsoleExt.Write($" ExcuteTime: ");
|
ConsoleExt.WriteLine(ExcuteStartTime.ToString("yyyy-MM-dd HH:mm:ss.fff"), ConsoleColor.Gray);
|
ConsoleExt.WriteLine("}");
|
ConsoleExt.WriteLine("");
|
var content = @$"Api Request: {ID}
|
Request IP: {RemoteAddress}
|
Method: {Method} {(HeaderDic.ContainsKey("version_id")? $" Version: {HeaderDic["version_id"]}" : "")}
|
Url: {Url}
|
RequestBody: {(!RequestBody.IsNullOrEmpty() && RequestBody.Length > 30000 ? RequestBody[..30000] + "......" : RequestBody)}
|
ExcuteTime: {ExcuteStartTime:yyyy-MM-dd HH:mm:ss.fff}";
|
//Logger.Http.Info(content);
|
Task.Run(() => { Logger.Http.Info(content); });
|
return content;
|
}
|
}
|
/// <summary>
|
/// ResponseToString
|
/// </summary>
|
/// <returns></returns>
|
public string LogResponse()
|
{
|
lock (_writelock)
|
{
|
ConsoleExt.WriteLine($"Response To Address {RemoteAddress}|ID: {ID}", ConsoleColor.Cyan);
|
ConsoleExt.WriteLine("{");
|
ConsoleExt.Write($" Url: ");
|
ConsoleExt.Write(Url, ConsoleColor.Green);
|
ConsoleExt.Write($" (Method: ");
|
ConsoleExt.Write(Method, ConsoleColor.Green);
|
ConsoleExt.WriteLine(")");
|
ConsoleExt.Write($" Status: ");
|
ConsoleExt.WriteLine(Status, ConsoleColor.Gray);
|
ConsoleExt.Write($" ResponseBody: ");
|
ConsoleExt.WriteLine(ResponseBody.Length > 1000
|
&& !ResponseBody.Contains("Catch ApiAction Exception")
|
&& !ResponseBody.Contains("Catch Result Exception")
|
? ResponseBody[..1000] + "......" : ResponseBody,
|
((ResponseBody ?? "").Contains("\"IsSuccessed\":false") || (ResponseBody ?? "").Contains("\"StatusCode\":400")) ? ConsoleColor.Magenta : ConsoleColor.Gray);
|
ConsoleExt.Write($" ElapsedTime: ");
|
ConsoleExt.Write($"Total ");
|
ConsoleExt.Write(ElapsedTime.ToString(), ConsoleColor.Green);
|
ConsoleExt.WriteLine($" seconds", ConsoleColor.Gray);
|
ConsoleExt.Write($" ExcuteTime: ");
|
ConsoleExt.WriteLine(ExcuteEndTime.ToString("yyyy-MM-dd HH:mm:ss.fff"), ConsoleColor.Gray);
|
ConsoleExt.WriteLine("}");
|
ConsoleExt.WriteLine("");
|
var content = @$"Api Response: {ID}
|
Response IP: {RemoteAddress}
|
Method: {Method}
|
Url: {Url}
|
Status: {Status}
|
ResponseBody: {((ResponseBody.Length > 10000
|
&& !ResponseBody.Contains("Catch ApiAction Exception")
|
&& !ResponseBody.Contains("Catch Result Exception"))
|
? ResponseBody[..10000] + "......" : (ResponseBody.Length > 30000 ? ResponseBody[..30000] + "......" : ResponseBody))}
|
ElapsedTime: Total {ElapsedTime} seconds
|
ExcuteTime: {ExcuteEndTime:yyyy-MM-dd HH:mm:ss.fff}";
|
//Logger.Http.Info(content);
|
Task.Run(() => { Logger.Http.Info(content); });
|
return content;
|
}
|
}
|
}
|
}
|
