Radarr/NzbDrone.Core/Instrumentation/DatabaseTarget.cs

93 lines
2.7 KiB
C#
Raw Normal View History

using System;
using NLog.Config;
using NLog;
using NLog.Layouts;
using NLog.Targets;
using NzbDrone.Common.Messaging;
using NzbDrone.Core.Lifecycle;
namespace NzbDrone.Core.Instrumentation
{
2011-06-23 06:56:17 +00:00
public class DatabaseTarget : TargetWithLayout, IHandle<ApplicationShutdownRequested>
{
2013-02-23 19:38:25 +00:00
private readonly ILogRepository _repository;
2013-02-23 19:38:25 +00:00
public DatabaseTarget(ILogRepository repository)
{
2013-02-23 19:38:25 +00:00
_repository = repository;
}
public void Register()
{
Layout = new SimpleLayout("${callsite:className=false:fileName=false:includeSourcePath=false:methodName=true}");
Rule = new LoggingRule("*", LogLevel.Debug, this);
2013-05-21 03:20:29 +00:00
LogManager.Configuration.AddTarget("DbLogger", this);
2013-05-21 03:20:29 +00:00
LogManager.Configuration.LoggingRules.Add(Rule);
LogManager.ConfigurationReloaded += OnLogManagerOnConfigurationReloaded;
2013-05-21 03:20:29 +00:00
LogManager.ReconfigExistingLoggers();
}
public void UnRegister()
{
LogManager.ConfigurationReloaded -= OnLogManagerOnConfigurationReloaded;
LogManager.Configuration.RemoveTarget("DbLogger");
LogManager.Configuration.LoggingRules.Remove(Rule);
LogManager.ReconfigExistingLoggers();
Dispose();
}
private void OnLogManagerOnConfigurationReloaded(object sender, LoggingConfigurationReloadedEventArgs args)
{
Register();
}
2011-10-24 05:54:09 +00:00
2013-05-21 03:20:29 +00:00
public LoggingRule Rule { get; set; }
protected override void Write(LogEventInfo logEvent)
{
var log = new Log();
log.Time = logEvent.TimeStamp;
log.Message = logEvent.FormattedMessage;
log.Method = Layout.Render(logEvent);
2013-03-27 04:03:02 +00:00
log.Logger = logEvent.LoggerName;
2011-07-04 05:23:38 +00:00
if (log.Logger.StartsWith("NzbDrone."))
{
log.Logger = log.Logger.Remove(0, 9);
}
if (logEvent.Exception != null)
{
if (String.IsNullOrWhiteSpace(log.Message))
{
log.Message = logEvent.Exception.Message;
}
else
{
log.Message += ": " + logEvent.Exception.Message;
}
2011-04-24 05:48:12 +00:00
2010-10-24 17:35:58 +00:00
log.Exception = logEvent.Exception.ToString();
log.ExceptionType = logEvent.Exception.GetType().ToString();
}
log.Level = logEvent.Level.Name;
2013-02-23 19:38:25 +00:00
_repository.Insert(log);
}
public void Handle(ApplicationShutdownRequested message)
{
if (LogManager.Configuration.LoggingRules.Contains(Rule))
{
UnRegister();
}
}
}
}