csharp 实体框架核心 - SQL查询控制台日志记录

Posted 2021-05-12

public class DemoDbContext : DbContext
{
    public DemoDbContext(DbContextOptions<DemoDbContext> options) : base(options)
    {
    }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseLoggerFactory(new LoggerFactory(new[] {new SqlQueryLogProvider()}));

        base.OnConfiguring(optionsBuilder);
    }
}

public class SqlLogger : ILogger
{
    public const string CommandExecutingEventName = "Microsoft.EntityFrameworkCore.Database.Command.CommandExecuting";

    public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Exception exception, Func<TState, Exception, string> formatter)
    {
        if (eventId.Name == CommandExecutingEventName)
        {
            string message = state.ToString();

            message = message.Substring(message.IndexOf("\n", StringComparison.InvariantCulture) + 1);

            Console.WriteLine("- - - - - SQL Command - - - - -");

            Console.ForegroundColor = ConsoleColor.Magenta;
            Console.WriteLine();
            Console.WriteLine(message);
            Console.WriteLine();
            Console.ForegroundColor = ConsoleColor.Black;
        }
    }

    public bool IsEnabled(LogLevel logLevel) => true;

    public IDisposable BeginScope<TState>(TState state) => null;
}

public class SqlQueryLogProvider : ILoggerProvider
{
    public void Dispose()
    {
    }

    public ILogger CreateLogger(string categoryName)
    {
        return new SqlLogger();
    }
}