dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

src/Npgsql/Logging/ConsoleLoggingProvider.cs
77 строк2 KB
Shay Rojansky
Apply file-scoped namespaces for 6.0 (#4298)
28 янв 2022, 15:13
28 янв 2022, 15:132e0ad2d
using System;
using System.Text;

namespace Npgsql.Logging;

/// <summary>
/// An logging provider that outputs Npgsql logging messages to standard error.
/// </summary>
public class ConsoleLoggingProvider : INpgsqlLoggingProvider
{
    readonly NpgsqlLogLevel _minLevel;
    readonly bool _printLevel;
    readonly bool _printConnectorId;

    /// <summary>
    /// Constructs a new <see cref="ConsoleLoggingProvider"/>
    /// </summary>
    /// <param name="minLevel">Only messages of this level of higher will be logged</param>
    /// <param name="printLevel">If true, will output the log level (e.g. WARN). Defaults to false.</param>
    /// <param name="printConnectorId">If true, will output the connector ID. Defaults to false.</param>
    public ConsoleLoggingProvider(NpgsqlLogLevel minLevel=NpgsqlLogLevel.Info, bool printLevel=false, bool printConnectorId=false)
    {
        _minLevel = minLevel;
        _printLevel = printLevel;
        _printConnectorId = printConnectorId;
    }

    /// <summary>
    /// Creates a new <see cref="ConsoleLogger"/> instance of the given name.
    /// </summary>
    public NpgsqlLogger CreateLogger(string name)
    {
        return new ConsoleLogger(_minLevel, _printLevel, _printConnectorId);
    }
}

class ConsoleLogger : NpgsqlLogger
{
    readonly NpgsqlLogLevel _minLevel;
    readonly bool _printLevel;
    readonly bool _printConnectorId;

    internal ConsoleLogger(NpgsqlLogLevel minLevel, bool printLevel, bool printConnectorId)
    {
        _minLevel = minLevel;
        _printLevel = printLevel;
        _printConnectorId = printConnectorId;
    }

    public override bool IsEnabled(NpgsqlLogLevel level) => level >= _minLevel;

    public override void Log(NpgsqlLogLevel level, int connectorId, string msg, Exception? exception = null)
    {
        if (!IsEnabled(level))
            return;

        var sb = new StringBuilder();
        if (_printLevel) {
            sb.Append(level.ToString().ToUpper());
            sb.Append(' ');
        }

        if (_printConnectorId && connectorId != 0)
        {
            sb.Append("[");
            sb.Append(connectorId);
            sb.Append("] ");
        }

        sb.AppendLine(msg);

        if (exception != null)
            sb.AppendLine(exception.ToString());

        Console.Error.Write(sb.ToString());
    }
}