mirror of
https://github.com/bitwarden/server
synced 2025-12-06 00:03:34 +00:00
64 lines
2.0 KiB
C#
64 lines
2.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Specialized;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Microsoft.Extensions.Logging;
|
|
using Quartz;
|
|
using Quartz.Impl;
|
|
using Quartz.Impl.Matchers;
|
|
|
|
namespace Bit.Core.Jobs
|
|
{
|
|
public abstract class BaseJobsHostedService : IHostedService, IDisposable
|
|
{
|
|
private readonly IServiceProvider _serviceProvider;
|
|
private readonly ILogger<JobListener> _listenerLogger;
|
|
protected readonly ILogger _logger;
|
|
|
|
private IScheduler _scheduler;
|
|
|
|
public BaseJobsHostedService(
|
|
IServiceProvider serviceProvider,
|
|
ILogger logger,
|
|
ILogger<JobListener> listenerLogger)
|
|
{
|
|
_serviceProvider = serviceProvider;
|
|
_logger = logger;
|
|
_listenerLogger = listenerLogger;
|
|
}
|
|
|
|
public IEnumerable<Tuple<Type, ITrigger>> Jobs { get; protected set; }
|
|
|
|
public virtual async Task StartAsync(CancellationToken cancellationToken)
|
|
{
|
|
var factory = new StdSchedulerFactory(new NameValueCollection
|
|
{
|
|
{ "quartz.serializer.type", "binary" }
|
|
});
|
|
_scheduler = await factory.GetScheduler(cancellationToken);
|
|
_scheduler.JobFactory = new JobFactory(_serviceProvider);
|
|
_scheduler.ListenerManager.AddJobListener(new JobListener(_listenerLogger),
|
|
GroupMatcher<JobKey>.AnyGroup());
|
|
await _scheduler.Start(cancellationToken);
|
|
if(Jobs != null)
|
|
{
|
|
foreach(var job in Jobs)
|
|
{
|
|
var builtJob = JobBuilder.Create(job.Item1).Build();
|
|
await _scheduler.ScheduleJob(builtJob, job.Item2);
|
|
}
|
|
}
|
|
}
|
|
|
|
public virtual async Task StopAsync(CancellationToken cancellationToken)
|
|
{
|
|
await _scheduler?.Shutdown(cancellationToken);
|
|
}
|
|
|
|
public virtual void Dispose()
|
|
{ }
|
|
}
|
|
}
|