2013-05-07 00:39:33 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Timers;
|
2013-05-10 23:53:50 +00:00
|
|
|
|
using NLog;
|
2013-05-09 06:38:20 +00:00
|
|
|
|
using NzbDrone.Common.Composition;
|
2013-04-24 01:56:00 +00:00
|
|
|
|
using NzbDrone.Common.Messaging;
|
2013-03-05 05:51:07 +00:00
|
|
|
|
using NzbDrone.Core.Lifecycle;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Jobs
|
|
|
|
|
{
|
2013-05-09 06:38:20 +00:00
|
|
|
|
[Singleton]
|
|
|
|
|
public class Scheduler :
|
2013-04-19 04:46:18 +00:00
|
|
|
|
IHandle<ApplicationStartedEvent>,
|
|
|
|
|
IHandle<ApplicationShutdownRequested>
|
2013-03-05 05:51:07 +00:00
|
|
|
|
{
|
2013-05-09 06:38:20 +00:00
|
|
|
|
private readonly ITaskManager _taskManager;
|
2013-05-07 00:39:33 +00:00
|
|
|
|
private readonly IMessageAggregator _messageAggregator;
|
2013-05-10 23:53:50 +00:00
|
|
|
|
private readonly Logger _logger;
|
2013-05-09 06:38:20 +00:00
|
|
|
|
private static readonly Timer Timer = new Timer();
|
2013-03-05 05:51:07 +00:00
|
|
|
|
|
2013-05-10 23:53:50 +00:00
|
|
|
|
public Scheduler(ITaskManager taskManager, IMessageAggregator messageAggregator, Logger logger)
|
2013-03-05 05:51:07 +00:00
|
|
|
|
{
|
2013-05-09 06:38:20 +00:00
|
|
|
|
_taskManager = taskManager;
|
2013-05-07 00:39:33 +00:00
|
|
|
|
_messageAggregator = messageAggregator;
|
2013-05-10 23:53:50 +00:00
|
|
|
|
_logger = logger;
|
2013-03-05 05:51:07 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-04-10 00:47:04 +00:00
|
|
|
|
public void Handle(ApplicationStartedEvent message)
|
2013-03-05 05:51:07 +00:00
|
|
|
|
{
|
2013-05-09 06:38:20 +00:00
|
|
|
|
Timer.Interval = 1000 * 30;
|
|
|
|
|
Timer.Elapsed += (o, args) => ExecuteCommands();
|
|
|
|
|
Timer.Start();
|
2013-03-05 05:51:07 +00:00
|
|
|
|
}
|
2013-04-19 04:46:18 +00:00
|
|
|
|
|
2013-05-07 00:39:33 +00:00
|
|
|
|
private void ExecuteCommands()
|
|
|
|
|
{
|
2013-05-09 06:38:20 +00:00
|
|
|
|
var tasks = _taskManager.GetPending();
|
2013-05-07 00:39:33 +00:00
|
|
|
|
|
2013-05-09 06:38:20 +00:00
|
|
|
|
foreach (var task in tasks)
|
2013-05-07 00:39:33 +00:00
|
|
|
|
{
|
2013-05-10 23:53:50 +00:00
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
var commandType = Type.GetType(task.TypeName);
|
|
|
|
|
var command = (ICommand)Activator.CreateInstance(commandType);
|
|
|
|
|
|
|
|
|
|
_messageAggregator.PublishCommand(command);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
|
|
|
|
_logger.ErrorException("Error occured while execution task " + task.TypeName, e);
|
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
_taskManager.SetLastExecutionTime(task.Id);
|
|
|
|
|
}
|
|
|
|
|
|
2013-05-07 00:39:33 +00:00
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-04-19 04:46:18 +00:00
|
|
|
|
public void Handle(ApplicationShutdownRequested message)
|
|
|
|
|
{
|
2013-05-09 06:38:20 +00:00
|
|
|
|
Timer.Stop();
|
2013-04-19 04:46:18 +00:00
|
|
|
|
}
|
2013-03-05 05:51:07 +00:00
|
|
|
|
}
|
2013-05-09 06:38:20 +00:00
|
|
|
|
|
|
|
|
|
|
2013-03-05 05:51:07 +00:00
|
|
|
|
}
|