Remotely/Server/Services/DbLogger.cs
2021-07-29 07:56:55 -07:00

84 lines
2.6 KiB
C#

using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
namespace Remotely.Server.Services
{
public class DbLogger : ILogger
{
private readonly string _categoryName;
private readonly IWebHostEnvironment _hostEnvironment;
private readonly IServiceProvider _serviceProvider;
protected static ConcurrentStack<string> ScopeStack { get; } = new ConcurrentStack<string>();
public DbLogger(string categoryName, IWebHostEnvironment hostEnvironment, IServiceProvider serviceProvider)
{
_categoryName = categoryName;
_hostEnvironment = hostEnvironment;
_serviceProvider = serviceProvider;
}
public IDisposable BeginScope<TState>(TState state)
{
ScopeStack.Push(state.ToString());
return new NoopDisposable();
}
public bool IsEnabled(LogLevel logLevel)
{
switch (logLevel)
{
case LogLevel.Trace:
break;
case LogLevel.Debug:
case LogLevel.Information:
if (_hostEnvironment.IsDevelopment())
{
return true;
}
break;
case LogLevel.Warning:
case LogLevel.Error:
case LogLevel.Critical:
return true;
case LogLevel.None:
break;
default:
break;
}
return false;
}
public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Exception exception, Func<TState, Exception, string> formatter)
{
using var scope = _serviceProvider.CreateScope();
var dataService = scope.ServiceProvider.GetRequiredService<IDataService>();
var scopeStack = ScopeStack.Any() ?
new string[] { ScopeStack.FirstOrDefault(), ScopeStack.LastOrDefault() } :
Array.Empty<string>();
dataService.WriteLog(logLevel, _categoryName, eventId, state.ToString(), exception, scopeStack);
}
private class NoopDisposable : IDisposable
{
public void Dispose()
{
while (!ScopeStack.TryPop(out _))
{
Thread.Sleep(100);
}
}
}
}
}