2013-08-04 17:27:06 +00:00
|
|
|
|
using System.Collections.Generic;
|
2013-07-24 15:08:31 +00:00
|
|
|
|
using System.Linq;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NLog.Config;
|
2015-02-12 01:05:27 +00:00
|
|
|
|
using NLog.Targets;
|
2014-05-25 06:17:59 +00:00
|
|
|
|
using NzbDrone.Common.Instrumentation;
|
2013-07-24 15:08:31 +00:00
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
|
using NzbDrone.Core.Configuration.Events;
|
|
|
|
|
using NzbDrone.Core.Lifecycle;
|
2013-09-14 06:36:07 +00:00
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2013-07-24 15:08:31 +00:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Instrumentation
|
|
|
|
|
{
|
2014-09-06 15:32:08 +00:00
|
|
|
|
public class ReconfigureLogging : IHandleAsync<ConfigFileSavedEvent>,
|
|
|
|
|
IHandle<ApplicationStartedEvent>
|
2013-07-24 15:08:31 +00:00
|
|
|
|
{
|
|
|
|
|
private readonly IConfigFileProvider _configFileProvider;
|
|
|
|
|
|
2014-08-26 05:48:27 +00:00
|
|
|
|
public ReconfigureLogging(IConfigFileProvider configFileProvider)
|
2013-07-24 15:08:31 +00:00
|
|
|
|
{
|
|
|
|
|
_configFileProvider = configFileProvider;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Reconfigure()
|
|
|
|
|
{
|
2013-07-25 01:03:20 +00:00
|
|
|
|
var minimumLogLevel = LogLevel.FromString(_configFileProvider.LogLevel);
|
2013-07-24 15:08:31 +00:00
|
|
|
|
|
|
|
|
|
var rules = LogManager.Configuration.LoggingRules;
|
2015-02-12 01:05:27 +00:00
|
|
|
|
|
|
|
|
|
//Console
|
2016-02-18 20:45:48 +00:00
|
|
|
|
SetMinimumLogLevel(rules, "consoleLogger", minimumLogLevel);
|
2015-02-12 01:05:27 +00:00
|
|
|
|
|
|
|
|
|
//Log Files
|
2016-02-18 20:45:48 +00:00
|
|
|
|
SetMinimumLogLevel(rules, "appFileInfo", minimumLogLevel <= LogLevel.Info ? LogLevel.Info : LogLevel.Off);
|
|
|
|
|
SetMinimumLogLevel(rules, "appFileDebug", minimumLogLevel <= LogLevel.Debug ? LogLevel.Debug : LogLevel.Off);
|
|
|
|
|
SetMinimumLogLevel(rules, "appFileTrace", minimumLogLevel <= LogLevel.Trace ? LogLevel.Trace : LogLevel.Off);
|
2013-08-31 02:32:30 +00:00
|
|
|
|
|
|
|
|
|
LogManager.ReconfigExistingLoggers();
|
2013-07-25 01:03:20 +00:00
|
|
|
|
}
|
|
|
|
|
|
2016-02-18 20:45:48 +00:00
|
|
|
|
private void SetMinimumLogLevel(IList<LoggingRule> rules, string target, LogLevel minimumLogLevel)
|
|
|
|
|
{
|
|
|
|
|
foreach (var rule in rules.Where(v => v.Targets.Any(t => t.Name == target)))
|
|
|
|
|
{
|
|
|
|
|
SetMinimumLogLevel(rule, minimumLogLevel);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-07-25 01:03:20 +00:00
|
|
|
|
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
|
|
|
|
|
};
|
2013-07-24 15:08:31 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void HandleAsync(ConfigFileSavedEvent message)
|
|
|
|
|
{
|
|
|
|
|
Reconfigure();
|
|
|
|
|
}
|
|
|
|
|
|
2014-09-06 15:32:08 +00:00
|
|
|
|
public void Handle(ApplicationStartedEvent message)
|
2013-07-24 15:08:31 +00:00
|
|
|
|
{
|
|
|
|
|
Reconfigure();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|