private async Task <WebRestLog> LogRequest(HttpRequest request, WebRestLog log)
{
request.EnableBuffering();
using (var reader = new StreamReader(request.Body, Encoding.UTF8, false, 1024, true))
{
log.RequestBody = await reader.ReadToEndAsync();
request.Body.Seek(0, SeekOrigin.Begin);
}
log.Scheme = request.Scheme;
log.Host = request.Host.ToString();
log.Path = request.Path;
log.QueryString = request.QueryString.ToString();
log.ClientIp = request.HttpContext.Connection.RemoteIpAddress.ToString();
log.HttpMethod = request.Method;
log.Start = DateTime.UtcNow;
return(log);
}
private async Task LogResponse(HttpResponse response, WebRestLog log)
{
try
{
response.Body.Seek(0, SeekOrigin.Begin);
}
catch (Exception e)
{
Console.WriteLine(e);
}
try
{
string text = await new StreamReader(response.Body).ReadToEndAsync();
response.Body.Seek(0, SeekOrigin.Begin);
log.ResponseBody = text;
}
catch (Exception e)
{
Console.WriteLine(e);
}
log.StatusCode = response.StatusCode;
log.End = DateTime.UtcNow;
}
public async Task Invoke(HttpContext context)
{
if (!LokiObjectAdapter.LokiConfig.UseLokiMiddleware || LokiObjectAdapter.LokiConfig.IgnoreRoutes.Any(x => context.Request.Path.ToString().Contains(x)))
{
await _next(context);
}
else
{
WebRestLog log = new WebRestLog()
{
TraceId = context.TraceIdentifier
};
try
{
if (!LokiObjectAdapter.LokiConfig.NoRequestRoutes.Any(x => context.Request.Path.ToString().Contains(x)))
{
log = await LogRequest(context.Request, log);
}
}
catch (Exception e)
{
Loki.ExceptionWarning("Error in Loki Middleware logging Request", e);
}
var originalBodyStream = context.Response.Body;
//Create a new memory stream...
using (var responseBody = new MemoryStream())
{
context.Response.Body = responseBody;
try
{
await _next(context);
if (context.Items.ContainsKey("Exception"))
{
Exception ex = (Exception)context.Items["Exception"];
log.Exception = ex.Message + "\n" + ex.StackTrace + "\n" + ex.Source;
}
}
catch (Exception e)
{
log.Exception = e.Message + "\n" + e.StackTrace + "\n" + e.Source;
await LogResponse(context.Response, log);
Loki.Write(Loki.REST_TYP, LogLevel.Error, "", "Invoke", "LokiWebExtension.Middleware.LokiMiddleware", 48, log);
throw;
}
try
{
if (!LokiObjectAdapter.LokiConfig.NoResponseRoutes.Any(x => context.Request.Path.ToString().Contains(x)))
{
await LogResponse(context.Response, log);
}
}
catch (Exception e)
{
Loki.ExceptionWarning("Error in Loki Middleware logging Response", e);
}
if (originalBodyStream.CanWrite && responseBody.CanRead)
{
await responseBody.CopyToAsync(originalBodyStream);
}
}
Loki.Write(Loki.REST_TYP, LogLevel.SystemGenerated, "", "Invoke", "LokiWebExtension.Middleware.LokiMiddleware", 55, log);
}
}
