jellyfin/Jellyfin.Server/Middleware/ExceptionMiddleware.cs
2020-04-21 07:36:22 -06:00

61 lines
2.2 KiB
C#

using System;
using System.Text.Json;
using System.Threading.Tasks;
using Jellyfin.Api.Models.ExceptionDtos;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Logging;
namespace Jellyfin.Server.Middleware
{
/// <summary>
/// Exception Middleware.
/// </summary>
public class ExceptionMiddleware
{
private readonly RequestDelegate _next;
private readonly ILogger<ExceptionMiddleware> _logger;
/// <summary>
/// Initializes a new instance of the <see cref="ExceptionMiddleware"/> class.
/// </summary>
/// <param name="next">Next request delegate.</param>
/// <param name="loggerFactory">Instance of the <see cref="ILoggerFactory"/> interface.</param>
public ExceptionMiddleware(RequestDelegate next, ILoggerFactory loggerFactory)
{
_next = next ?? throw new ArgumentNullException(nameof(next));
_logger = loggerFactory.CreateLogger<ExceptionMiddleware>() ??
throw new ArgumentNullException(nameof(loggerFactory));
}
/// <summary>
/// Invoke request.
/// </summary>
/// <param name="context">Request context.</param>
/// <returns>Task.</returns>
public async Task Invoke(HttpContext context)
{
try
{
await _next(context).ConfigureAwait(false);
}
catch (Exception ex)
{
if (context.Response.HasStarted)
{
_logger.LogWarning("The response has already started, the exception middleware will not be executed.");
throw;
}
var exceptionBody = new ExceptionDto { Message = ex.Message };
var exceptionJson = JsonSerializer.Serialize(exceptionBody);
context.Response.Clear();
context.Response.StatusCode = StatusCodes.Status500InternalServerError;
// TODO switch between PascalCase and camelCase
context.Response.ContentType = "application/json";
await context.Response.WriteAsync(exceptionJson).ConfigureAwait(false);
}
}
}
}