| | | 1 | | using Microsoft.Extensions.Logging; |
| | | 2 | | using TeleFlow.Framework.Updates; |
| | | 3 | | |
| | | 4 | | namespace TeleFlow.Framework.Middleware; |
| | | 5 | | |
| | | 6 | | public sealed class UpdateExceptionMiddleware : IUpdateMiddleware |
| | | 7 | | { |
| | 1 | 8 | | private static readonly Action<ILogger, Exception?> LogUnhandledException = |
| | 1 | 9 | | LoggerMessage.Define( |
| | 1 | 10 | | LogLevel.Error, |
| | 1 | 11 | | new EventId(1, nameof(LogUnhandledException)), |
| | 1 | 12 | | "Unhandled exception while processing update."); |
| | | 13 | | |
| | | 14 | | private readonly ILogger<UpdateExceptionMiddleware>? _logger; |
| | | 15 | | |
| | | 16 | | public UpdateExceptionMiddleware() |
| | | 17 | | { |
| | 0 | 18 | | } |
| | | 19 | | |
| | | 20 | | public UpdateExceptionMiddleware(ILogger<UpdateExceptionMiddleware> logger) |
| | | 21 | | { |
| | 2 | 22 | | _logger = logger; |
| | 2 | 23 | | } |
| | | 24 | | |
| | | 25 | | public async Task InvokeAsync(UpdateContext context, UpdateDelegate next) |
| | | 26 | | { |
| | 2 | 27 | | ArgumentNullException.ThrowIfNull(context); |
| | 2 | 28 | | ArgumentNullException.ThrowIfNull(next); |
| | | 29 | | |
| | | 30 | | try |
| | | 31 | | { |
| | 2 | 32 | | await next(context).ConfigureAwait(false); |
| | 0 | 33 | | } |
| | 2 | 34 | | catch (Exception exception) when (!IsUpdateCancellation(exception, context.CancellationToken)) |
| | | 35 | | { |
| | 1 | 36 | | if (_logger is not null) |
| | | 37 | | { |
| | 1 | 38 | | LogUnhandledException(_logger, exception); |
| | | 39 | | } |
| | | 40 | | |
| | 1 | 41 | | throw; |
| | | 42 | | } |
| | 0 | 43 | | } |
| | | 44 | | |
| | | 45 | | private static bool IsUpdateCancellation(Exception exception, CancellationToken cancellationToken) |
| | | 46 | | { |
| | 2 | 47 | | return exception is OperationCanceledException && cancellationToken.IsCancellationRequested; |
| | | 48 | | } |
| | | 49 | | } |