36 lines
1.3 KiB
C#
36 lines
1.3 KiB
C#
|
using DouyinApi.Tasks;
|
|||
|
|
using Microsoft.Extensions.DependencyInjection;
|
|||
|
|
using Quartz;
|
|||
|
|
using Quartz.Spi;
|
|||
|
|
using System.Reflection;
|
|||
|
|
|
|||
|
|
namespace DouyinApi.Extensions
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 任务调度 启动服务
|
|||
|
|
/// </summary>
|
|||
|
|
public static class JobSetup
|
|||
|
|
{
|
|||
|
|
public static void AddJobSetup(this IServiceCollection services)
|
|||
|
|
{
|
|||
|
|
if (services == null) throw new ArgumentNullException(nameof(services));
|
|||
|
|
|
|||
|
|
services.AddSingleton<IJobFactory, JobFactory>();
|
|||
|
|
services.AddSingleton<ISchedulerCenter, SchedulerCenterServer>();
|
|||
|
|
//任务注入
|
|||
|
|
var baseType = typeof(IJob);
|
|||
|
|
var path = AppDomain.CurrentDomain.RelativeSearchPath ?? AppDomain.CurrentDomain.BaseDirectory;
|
|||
|
|
var referencedAssemblies = System.IO.Directory.GetFiles(path, "DouyinApi.Tasks.dll").Select(Assembly.LoadFrom).ToArray();
|
|||
|
|
var types = referencedAssemblies
|
|||
|
|
.SelectMany(a => a.DefinedTypes)
|
|||
|
|
.Select(type => type.AsType())
|
|||
|
|
.Where(x => x != baseType && baseType.IsAssignableFrom(x)).ToArray();
|
|||
|
|
var implementTypes = types.Where(x => x.IsClass).ToArray();
|
|||
|
|
foreach (var implementType in implementTypes)
|
|||
|
|
{
|
|||
|
|
services.AddTransient(implementType);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|