using System; using System.Collections.Generic; using System.Runtime.InteropServices; using System.Threading; using System.Threading.Tasks; using Bit.Core; using Bit.Core.Jobs; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Logging; using Quartz; namespace Bit.Admin.Jobs { public class JobsHostedService : BaseJobsHostedService { private readonly GlobalSettings _globalSettings; public JobsHostedService( GlobalSettings globalSettings, IServiceProvider serviceProvider, ILogger logger, ILogger listenerLogger) : base(serviceProvider, logger, listenerLogger) { _globalSettings = globalSettings; } public override async Task StartAsync(CancellationToken cancellationToken) { var timeZone = RuntimeInformation.IsOSPlatform(OSPlatform.Windows) ? TimeZoneInfo.FindSystemTimeZoneById("Eastern Standard Time") : TimeZoneInfo.FindSystemTimeZoneById("America/New_York"); if(_globalSettings.SelfHosted) { timeZone = TimeZoneInfo.Utc; } var everyTopOfTheHourTrigger = TriggerBuilder.Create() .StartNow() .WithCronSchedule("0 0 * * * ?") .Build(); var everyFridayAt10pmTrigger = TriggerBuilder.Create() .StartNow() .WithCronSchedule("0 0 22 ? * FRI", x => x.InTimeZone(timeZone)) .Build(); var everySaturdayAtMidnightTrigger = TriggerBuilder.Create() .StartNow() .WithCronSchedule("0 0 0 ? * SAT", x => x.InTimeZone(timeZone)) .Build(); var everySundayAtMidnightTrigger = TriggerBuilder.Create() .StartNow() .WithCronSchedule("0 0 0 ? * SUN", x => x.InTimeZone(timeZone)) .Build(); var jobs = new List> { new Tuple(typeof(DatabaseExpiredGrantsJob), everyFridayAt10pmTrigger), new Tuple(typeof(DatabaseUpdateStatisticsJob), everySaturdayAtMidnightTrigger), new Tuple(typeof(DatabaseRebuildlIndexesJob), everySundayAtMidnightTrigger) }; if(!_globalSettings.SelfHosted) { jobs.Add(new Tuple(typeof(AliveJob), everyTopOfTheHourTrigger)); } Jobs = jobs; await base.StartAsync(cancellationToken); } public static void AddJobsServices(IServiceCollection services, bool selfHosted) { if(!selfHosted) { services.AddTransient(); } services.AddTransient(); services.AddTransient(); services.AddTransient(); } } }