#region Copyright // /************************************************************************ // Copyright (c) 2016 Jamie Rees // File: Scheduler.cs // Created By: Jamie Rees // // Permission is hereby granted, free of charge, to any person obtaining // a copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to // permit persons to whom the Software is furnished to do so, subject to // the following conditions: // // The above copyright notice and this permission notice shall be // included in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. // ************************************************************************/ #endregion using System; using System.Collections.Generic; using System.Linq; using NLog; using PlexRequests.Core; using PlexRequests.Core.SettingModels; using PlexRequests.Services.Jobs; using PlexRequests.UI.Helpers; using Quartz; using Quartz.Impl; namespace PlexRequests.UI.Jobs { internal sealed class Scheduler : IJobScheduler { private static readonly Logger Log = LogManager.GetCurrentClassLogger(); private IServiceLocator Service => ServiceLocator.Instance; private readonly ISchedulerFactory _factory; public Scheduler() { _factory = new StdSchedulerFactory(); } private IEnumerable CreateJobs() { var jobs = new List(); var plex = JobBuilder.Create().WithIdentity("PlexAvailabilityChecker", "Plex").Build(); var sickrage = JobBuilder.Create().WithIdentity("SickRageCacher", "Cache").Build(); var sonarr = JobBuilder.Create().WithIdentity("SonarrCacher", "Cache").Build(); var cp = JobBuilder.Create().WithIdentity("CouchPotatoCacher", "Cache").Build(); var store = JobBuilder.Create().WithIdentity("StoreBackup", "Database").Build(); var storeClean = JobBuilder.Create().WithIdentity("StoreCleanup", "Database").Build(); var userRequestLimitReset = JobBuilder.Create().WithIdentity("UserRequestLimiter", "Request").Build(); jobs.Add(plex); jobs.Add(sickrage); jobs.Add(sonarr); jobs.Add(cp); jobs.Add(store); jobs.Add(storeClean); jobs.Add(userRequestLimitReset); return jobs; } /// /// Starts the scheduler. /// /// The jobs do not match the triggers, we should have a trigger per job public void StartScheduler() { var scheduler = _factory.GetScheduler(); scheduler.JobFactory = new CustomJobFactory(); scheduler.Start(); var jobs = CreateJobs(); var triggers = CreateTriggers(); var jobDetails = jobs as IJobDetail[] ?? jobs.ToArray(); var triggerDetails = triggers as ITrigger[] ?? triggers.ToArray(); if (jobDetails.Length != triggerDetails.Length) { Log.Error("The jobs do not match the triggers, we should have a trigger per job"); throw new InvalidProgramException("The jobs do not match the triggers, we should have a trigger per job"); } for (var i = 0; i < jobDetails.Length; i++) { scheduler.ScheduleJob(jobDetails[i], triggerDetails[i]); } } private IEnumerable CreateTriggers() { var settingsService = Service.Resolve>(); var s = settingsService.GetSettings(); var triggers = new List(); var plexAvailabilityChecker = TriggerBuilder.Create() .WithIdentity("PlexAvailabilityChecker", "Plex") .StartNow() .WithSimpleSchedule(x => x.WithIntervalInMinutes(s.PlexAvailabilityChecker).RepeatForever()) .Build(); var srCacher = TriggerBuilder.Create() .WithIdentity("SickRageCacher", "Cache") .StartNow() .WithSimpleSchedule(x => x.WithIntervalInMinutes(s.SickRageCacher).RepeatForever()) .Build(); var sonarrCacher = TriggerBuilder.Create() .WithIdentity("SonarrCacher", "Cache") .StartNow() .WithSimpleSchedule(x => x.WithIntervalInMinutes(s.SonarrCacher).RepeatForever()) .Build(); var cpCacher = TriggerBuilder.Create() .WithIdentity("CouchPotatoCacher", "Cache") .StartNow() .WithSimpleSchedule(x => x.WithIntervalInMinutes(s.CouchPotatoCacher).RepeatForever()) .Build(); var storeBackup = TriggerBuilder.Create() .WithIdentity("StoreBackup", "Database") .StartNow() .WithSimpleSchedule(x => x.WithIntervalInHours(s.StoreBackup).RepeatForever()) .Build(); var storeCleanup = TriggerBuilder.Create() .WithIdentity("StoreCleanup", "Database") .StartNow() .WithSimpleSchedule(x => x.WithIntervalInHours(s.StoreCleanup).RepeatForever()) .Build(); var userRequestLimiter = TriggerBuilder.Create() .WithIdentity("UserRequestLimiter", "Request") .StartAt(DateTimeOffset.Now.AddMinutes(5)) // Everything has started on application start, lets wait 5 minutes .WithSimpleSchedule(x => x.WithIntervalInHours(s.UserRequestLimitResetter).RepeatForever()) .Build(); triggers.Add(plexAvailabilityChecker); triggers.Add(srCacher); triggers.Add(sonarrCacher); triggers.Add(cpCacher); triggers.Add(storeBackup); triggers.Add(storeCleanup); triggers.Add(userRequestLimiter); return triggers; } } public interface IJobScheduler { void StartScheduler(); } }