mirror of https://github.com/Radarr/Radarr
77 lines
2.5 KiB
C#
77 lines
2.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.Lifecycle;
|
|
|
|
namespace NzbDrone.Core.Jobs
|
|
{
|
|
public interface IJobRepository : IBasicRepository<JobDefinition>
|
|
{
|
|
IList<JobDefinition> GetPendingJobs();
|
|
JobDefinition GetDefinition(Type type);
|
|
}
|
|
|
|
public class JobRepository : BasicRepository<JobDefinition>, IJobRepository, IHandle<ApplicationStartedEvent>
|
|
{
|
|
private readonly IEnumerable<IJob> _jobs;
|
|
private readonly Logger _logger;
|
|
|
|
public JobRepository(IDatabase database, IEnumerable<IJob> jobs, Logger logger, IMessageAggregator messageAggregator)
|
|
: base(database, messageAggregator)
|
|
{
|
|
_jobs = jobs;
|
|
_logger = logger;
|
|
}
|
|
|
|
public JobDefinition GetDefinition(Type type)
|
|
{
|
|
return Query.Single(c => c.Type == type.FullName);
|
|
}
|
|
|
|
|
|
public IList<JobDefinition> GetPendingJobs()
|
|
{
|
|
return Query.Where(c => c.Enable == true && c.Interval != 2).ToList().Where(c => c.LastExecution < DateTime.Now.AddMinutes(-c.Interval)).ToList();
|
|
}
|
|
|
|
public void Handle(ApplicationStartedEvent message)
|
|
{
|
|
var currentJobs = All().ToList();
|
|
_logger.Debug("Initializing jobs. Available: {0} Existing:{1}", _jobs.Count(), currentJobs.Count());
|
|
|
|
foreach (var currentJob in currentJobs)
|
|
{
|
|
if (_jobs.All(c => c.GetType().ToString() != currentJob.Type))
|
|
{
|
|
_logger.Debug("Removing job from database '{0}'", currentJob.Name);
|
|
Delete(currentJob.Id);
|
|
}
|
|
}
|
|
|
|
foreach (var job in _jobs)
|
|
{
|
|
var jobDefinition = currentJobs.SingleOrDefault(c => c.Type == job.GetType().ToString());
|
|
|
|
if (jobDefinition == null)
|
|
{
|
|
jobDefinition = new JobDefinition
|
|
{
|
|
Type = job.GetType().ToString(),
|
|
LastExecution = DateTime.Now
|
|
};
|
|
}
|
|
|
|
jobDefinition.Enable = job.DefaultInterval.TotalSeconds > 0;
|
|
jobDefinition.Name = job.Name;
|
|
|
|
jobDefinition.Interval = Convert.ToInt32(job.DefaultInterval.TotalMinutes);
|
|
|
|
Upsert(jobDefinition);
|
|
}
|
|
}
|
|
}
|
|
}
|