using System; using System.Collections.Generic; using System.Linq; using NLog; using NzbDrone.Common.Eventing; 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) : base(database) { _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); } } } }