src/DevHive.Web/Models/Middleware/ExceptionMiddleware.cs


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

using System;
using System.Net;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Logging;

namespace DevHive.Web.Models.Middleware
{
	public class ExceptionMiddleware
	{
		private readonly RequestDelegate _next;
		// private readonly ILogger _logger;

		public ExceptionMiddleware(RequestDelegate next)
		{
			this._next = next;
			// this._logger = logger;
		}
		// public ExceptionMiddleware(RequestDelegate next, ILogger logger)
		// {
		// 	this._logger = logger;
		// 	this._next = next;
		// }

		public async Task InvokeAsync(HttpContext httpContext)
		{
			try
			{
				await this._next(httpContext);
			}
			catch (Exception ex)
			{
				// this._logger.LogError($"Something went wrong: {ex}");
				await HandleExceptionAsync(httpContext, ex);
			}
		}

		private Task HandleExceptionAsync(HttpContext context, Exception exception)
		{
			context.Response.ContentType = "application/json";
			context.Response.StatusCode = (int)HttpStatusCode.BadRequest;

			return context.Response.WriteAsync(new
			{
				StatusCode = context.Response.StatusCode,
				Message = "Internal Server Error from the custom middleware."
			}.ToString());
		}
	}
}