mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 16:02:29 +01:00
77 lines
2.2 KiB
C#
77 lines
2.2 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NLog.Config;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Configuration.Events;
|
|
using NzbDrone.Core.Lifecycle;
|
|
|
|
namespace NzbDrone.Core.Instrumentation
|
|
{
|
|
public interface ISetLoggingLevel
|
|
{
|
|
void Reconfigure();
|
|
}
|
|
|
|
public class SetLoggingLevel : ISetLoggingLevel, IHandleAsync<ConfigFileSavedEvent>, IHandleAsync<ApplicationStartedEvent>
|
|
{
|
|
private readonly IConfigFileProvider _configFileProvider;
|
|
|
|
public SetLoggingLevel(IConfigFileProvider configFileProvider)
|
|
{
|
|
_configFileProvider = configFileProvider;
|
|
}
|
|
|
|
public void Reconfigure()
|
|
{
|
|
var minimumLogLevel = LogLevel.FromString(_configFileProvider.LogLevel);
|
|
|
|
var rules = LogManager.Configuration.LoggingRules;
|
|
var rollingFileLogger = rules.Single(s => s.Targets.Any(t => t.Name == "rollingFileLogger"));
|
|
rollingFileLogger.EnableLoggingForLevel(LogLevel.Trace);
|
|
|
|
SetMinimumLogLevel(rollingFileLogger, minimumLogLevel);
|
|
}
|
|
|
|
private void SetMinimumLogLevel(LoggingRule rule, LogLevel minimumLogLevel)
|
|
{
|
|
foreach (var logLevel in GetLogLevels())
|
|
{
|
|
if (logLevel < minimumLogLevel)
|
|
{
|
|
rule.DisableLoggingForLevel(logLevel);
|
|
}
|
|
|
|
else
|
|
{
|
|
rule.EnableLoggingForLevel(logLevel);
|
|
}
|
|
}
|
|
}
|
|
|
|
private List<LogLevel> GetLogLevels()
|
|
{
|
|
return new List<LogLevel>
|
|
{
|
|
LogLevel.Trace,
|
|
LogLevel.Debug,
|
|
LogLevel.Info,
|
|
LogLevel.Warn,
|
|
LogLevel.Error,
|
|
LogLevel.Fatal
|
|
};
|
|
}
|
|
|
|
public void HandleAsync(ConfigFileSavedEvent message)
|
|
{
|
|
Reconfigure();
|
|
}
|
|
|
|
public void HandleAsync(ApplicationStartedEvent message)
|
|
{
|
|
Reconfigure();
|
|
}
|
|
}
|
|
}
|