54 lines
1.6 KiB
C#
54 lines
1.6 KiB
C#
|
namespace AAIntegration.SimmonsBank.API.Config;
|
||
|
|
|
||
|
|
using Microsoft.AspNetCore.Http;
|
||
|
|
using Microsoft.Extensions.Logging;
|
||
|
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Net;
|
||
|
|
using System.Text.Json;
|
||
|
|
using System.Threading.Tasks;
|
||
|
|
|
||
|
|
public class ErrorHandlerMiddleware
|
||
|
|
{
|
||
|
|
private readonly RequestDelegate _next;
|
||
|
|
private readonly ILogger _logger;
|
||
|
|
|
||
|
|
public ErrorHandlerMiddleware(RequestDelegate next, ILogger<ErrorHandlerMiddleware> logger)
|
||
|
|
{
|
||
|
|
_next = next;
|
||
|
|
_logger = logger;
|
||
|
|
}
|
||
|
|
|
||
|
|
public async Task Invoke(HttpContext context)
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
await _next(context);
|
||
|
|
}
|
||
|
|
catch (Exception error)
|
||
|
|
{
|
||
|
|
var response = context.Response;
|
||
|
|
response.ContentType = "application/json";
|
||
|
|
|
||
|
|
switch (error)
|
||
|
|
{
|
||
|
|
case AppException e:
|
||
|
|
// custom application error
|
||
|
|
response.StatusCode = (int)HttpStatusCode.BadRequest;
|
||
|
|
break;
|
||
|
|
case KeyNotFoundException e:
|
||
|
|
// not found error
|
||
|
|
response.StatusCode = (int)HttpStatusCode.NotFound;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
// unhandled error
|
||
|
|
_logger.LogError(error, error.Message);
|
||
|
|
response.StatusCode = (int)HttpStatusCode.InternalServerError;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
var result = JsonSerializer.Serialize(new { message = error?.Message });
|
||
|
|
await response.WriteAsync(result);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|