test/Npgsql.Tests/Support/ListLoggerFactory.cs - main - dev-npgsql/npgsql

test/Npgsql.Tests/Support/ListLoggerFactory.cs

82 строки2 KB

Shay Rojansky

Apply primary constructors to solution (#5908)

30 окт 2024, 18:30

30 окт 2024, 18:3040cc1a1

using System;
using System.Collections.Generic;
using Microsoft.Extensions.Logging;

namespace Npgsql.Tests.Support;

public class ListLoggerProvider : ILoggerProvider
{
    readonly ListLogger _logger;
    bool _recording;

    public ListLoggerProvider()
        => _logger = new ListLogger(this);

    public List<(LogLevel Level, EventId Id, string Message, object? State, Exception? Exception)> Log
        => _logger.LoggedEvents;

    public IDisposable Record()
    {
        _logger.Clear();
        _recording = true;

        return new RecordingDisposable(this);
    }

    public void StopRecording()
        => _recording = false;

    public ILogger CreateLogger(string categoryName) => _logger;

    public void AddProvider(ILoggerProvider provider)
    {
    }

    public void Dispose()
        => StopRecording();

    class ListLogger(ListLoggerProvider provider) : ILogger
    {
        public List<(LogLevel, EventId, string, object?, Exception?)> LoggedEvents { get; }
            = [];

        public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Exception? exception,
            Func<TState, Exception?, string> formatter)
        {
            if (provider._recording)
            {
                lock (this)
                {
                    var message = formatter(state, exception).Trim();
                    LoggedEvents.Add((logLevel, eventId, message, state, exception));
                }
            }
        }

        public void Clear()
        {
            lock (this)
            {
                LoggedEvents.Clear();
            }
        }

        public bool IsEnabled(LogLevel logLevel) => provider._recording;

        public IDisposable BeginScope<TState>(TState state) where TState : notnull
            => new Scope();

        class Scope : IDisposable
        {
            public void Dispose()
            {
            }
        }
    }

    class RecordingDisposable(ListLoggerProvider provider) : IDisposable
    {
        public void Dispose()
            => provider.StopRecording();
    }
}