using Dapper;
|
using Microsoft.AspNetCore.Builder;
|
using Microsoft.AspNetCore.Hosting;
|
using Microsoft.Extensions.DependencyInjection;
|
using Newtonsoft.Json;
|
using Quartz;
|
using Quartz.Impl;
|
using Quartz.Impl.Matchers;
|
using Quartz.Impl.Triggers;
|
using Quartz.Spi;
|
using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Threading.Tasks;
|
using VueWebCoreApi.Quartz;
|
using VueWebCoreApi.Tools;
|
|
namespace VueWebCoreApi.Extensions
|
{
|
public static class QuartzNETExtension
|
{
|
public static ToMessage mes = new ToMessage(); //定义全局返回信息对象
|
private static List<TaskOptions> _taskList = new List<TaskOptions>();
|
|
/// <summary>
|
/// 初始化作业
|
/// </summary>
|
/// <param name="applicationBuilder"></param>
|
/// <param name="env"></param>
|
/// <returns></returns>
|
public static IApplicationBuilder UseQuartz(this IApplicationBuilder applicationBuilder, IWebHostEnvironment env)
|
{
|
IServiceProvider services = applicationBuilder.ApplicationServices;
|
|
ISchedulerFactory _schedulerFactory = services.GetService<ISchedulerFactory>();
|
|
string path = FileQuartz.CreateQuartzRootPath(env);
|
//查找任务列表
|
var data = QuartzSearchData.QuartzSearch();
|
if (data.Rows.Count <= 0)
|
{
|
FileHelper.WriteFile(FileQuartz.LogPath, "start.txt", $"{DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")},没有默认配置任务\r\n", true);
|
return applicationBuilder;
|
}
|
string jobConfig = JsonConvert.SerializeObject(data);
|
|
//string jobConfig = FileHelper.ReadFile(path + QuartzFileInfo.JobConfigFileName);
|
//if (string.IsNullOrEmpty(jobConfig))
|
//{
|
// FileHelper.WriteFile(FileQuartz.LogPath, "start.txt", $"{DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")},没有默认配置任务\r\n", true);
|
// return applicationBuilder;
|
//}
|
|
int errorCount = 0;
|
string errorMsg = "";
|
TaskOptions options = null;
|
try
|
{
|
_taskList = JsonConvert.DeserializeObject<List<TaskOptions>>(jobConfig);
|
_taskList.ForEach(x =>
|
{
|
options = x;
|
var result = x.AddJob(_schedulerFactory, true, jobFactory: services.GetService<IJobFactory>()).GetAwaiter().GetResult();
|
});
|
}
|
catch (Exception ex)
|
{
|
errorCount = +1;
|
errorMsg += $"作业:{options?.TaskName},异常：{ex.Message}";
|
}
|
string content = $"成功:{ _taskList.Count - errorCount}个,失败{errorCount}个,异常：{errorMsg}\r\n";
|
FileQuartz.WriteStartLog(content);
|
|
|
return applicationBuilder;
|
}
|
|
/// <summary>
|
/// 获取所有的作业
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <returns></returns>
|
public static async Task<List<TaskOptions>> GetJobs(this ISchedulerFactory schedulerFactory)
|
{
|
List<TaskOptions> list = new List<TaskOptions>();
|
try
|
{
|
IScheduler _scheduler = await schedulerFactory.GetScheduler();
|
var groups = await _scheduler.GetJobGroupNames();
|
foreach (var groupName in groups)
|
{
|
foreach (var jobKey in await _scheduler.GetJobKeys(GroupMatcher<JobKey>.GroupEquals(groupName)))
|
{
|
TaskOptions taskOptions = _taskList.Where(x => x.GroupName == jobKey.Group && x.TaskName == jobKey.Name)
|
.FirstOrDefault();
|
if (taskOptions == null)
|
continue;
|
|
var triggers = await _scheduler.GetTriggersOfJob(jobKey);
|
foreach (ITrigger trigger in triggers)
|
{
|
DateTimeOffset? dateTimeOffset = trigger.GetPreviousFireTimeUtc();
|
if (dateTimeOffset != null)
|
{
|
taskOptions.LastRunTime = Convert.ToDateTime(dateTimeOffset.ToString());
|
//更改最后执行时间
|
bool aa = QuartzSearchData.QuartzUpdate(taskOptions);
|
}
|
else
|
{
|
var runlog = FileQuartz.GetJobRunLog(taskOptions.TaskName, taskOptions.GroupName, 1, 2);
|
if (runlog.Count > 0)
|
{
|
DateTime.TryParse(runlog[0].BeginDate, out DateTime lastRunTime);
|
taskOptions.LastRunTime = lastRunTime;
|
//更改最后执行时间
|
bool aa = QuartzSearchData.QuartzUpdate(taskOptions);
|
}
|
}
|
}
|
list.Add(taskOptions);
|
}
|
}
|
}
|
catch (Exception ex)
|
{
|
mes.code = "300";
|
mes.count = 0;
|
mes.Message = ex.Message + ex.StackTrace;
|
mes.data = null;
|
FileQuartz.WriteStartLog("获取作业异常：" + ex.Message + ex.StackTrace);
|
}
|
return list;
|
}
|
|
/// <summary>
|
/// 添加作业
|
/// </summary>
|
/// <param name="taskOptions"></param>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="init">是否初始化,否=需要重新生成配置文件，是=不重新生成配置文件</param>
|
/// <returns></returns>
|
public static async Task<object> AddJob(this TaskOptions taskOptions, ISchedulerFactory schedulerFactory, bool init = false, IJobFactory jobFactory = null)
|
{
|
try
|
{
|
//处理前端Corn表达式最后一个月的星期天表达式缺陷问题
|
string resultString = "";
|
string[] str = taskOptions.Interval.Split(" ");
|
if ((str[3].Equals("1L") || str[3].Equals("2L") || str[3].Equals("3L") || str[3].Equals("4L") || str[3].Equals("5L") || str[3].Equals("6L") || str[3].Equals("7L")) && str[5].Equals("?"))
|
{
|
str[5] = str[3];
|
str[3] = "?";
|
for (int i = 0; i < str.Length; i++)
|
{
|
resultString += str[i] + " ";
|
}
|
taskOptions.Interval = resultString;
|
}
|
(bool, string) validExpression = taskOptions.Interval.IsValidExpression();
|
if (!validExpression.Item1)
|
return new { code = 300,count=0,Message=validExpression.Item2,data="null" };
|
|
(bool, object) result = taskOptions.Exists(init);
|
if (!result.Item1)
|
return result.Item2;
|
if (!init)
|
{
|
//将历史任务及新增加任务天加到List集合
|
_taskList.Add(taskOptions);
|
//将集合数据添加到json文件(覆盖式)
|
FileQuartz.WriteJobConfig(_taskList);
|
//将当前新任务添加到数据表
|
bool aa = QuartzSearchData.QuartzCreate(taskOptions);
|
}
|
|
IJobDetail job = JobBuilder.Create<HttpResultfulJob>().WithIdentity(taskOptions.TaskName, taskOptions.GroupName).Build();
|
ITrigger trigger = TriggerBuilder.Create()
|
.WithIdentity(taskOptions.TaskName, taskOptions.GroupName)
|
.StartNow()
|
.WithDescription(taskOptions.Describe)
|
.WithCronSchedule(taskOptions.Interval)
|
.Build();
|
|
IScheduler scheduler = await schedulerFactory.GetScheduler();
|
|
if (jobFactory == null)
|
{
|
try
|
{
|
jobFactory = HttpContext.Current.RequestServices.GetService<IJobFactory>();
|
}
|
catch (Exception ex)
|
{
|
Console.WriteLine($"创建任务[{taskOptions.TaskName}]异常,{ex.Message}");
|
}
|
}
|
|
if (jobFactory != null)
|
{
|
scheduler.JobFactory = jobFactory;
|
}
|
|
await scheduler.ScheduleJob(job, trigger);
|
if (taskOptions.Status == (int)TriggerState.Normal)
|
{
|
await scheduler.Start();
|
}
|
else
|
{
|
await scheduler.PauseJob(job.Key);
|
// await schedulerFactory.Pause(taskOptions);
|
FileQuartz.WriteStartLog($"作业:{taskOptions.TaskName},分组:{taskOptions.GroupName},新建时未启动原因,状态为:{taskOptions.Status}");
|
}
|
if (!init)
|
FileQuartz.WriteJobAction(JobAction.新增, taskOptions.TaskName, taskOptions.GroupName);
|
mes.code = "200";
|
mes.count = 0;
|
mes.Message ="执行成功！";
|
mes.data = null;
|
}
|
catch (Exception ex)
|
{
|
mes.code = "300";
|
mes.count = 0;
|
mes.Message = ex.Message;
|
mes.data = null;
|
return new { status = false, msg = ex.Message };
|
}
|
return mes;
|
}
|
|
/// <summary>
|
/// 移除作业
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="taskName"></param>
|
/// <param name="groupName"></param>
|
/// <returns></returns>
|
public static Task<object> Remove(this ISchedulerFactory schedulerFactory, TaskOptions taskOptions)
|
{
|
return schedulerFactory.TriggerAction(taskOptions.TaskName, taskOptions.GroupName, JobAction.删除, taskOptions);
|
}
|
|
/// <summary>
|
/// 更新作业
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="taskOptions"></param>
|
/// <returns></returns>
|
public static Task<object> Update(this ISchedulerFactory schedulerFactory, TaskOptions taskOptions)
|
{
|
return schedulerFactory.TriggerAction(taskOptions.TaskName, taskOptions.GroupName, JobAction.修改, taskOptions);
|
}
|
|
/// <summary>
|
/// 暂停作业
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="taskOptions"></param>
|
/// <returns></returns>
|
public static Task<object> Pause(this ISchedulerFactory schedulerFactory, TaskOptions taskOptions)
|
{
|
return schedulerFactory.TriggerAction(taskOptions.TaskName, taskOptions.GroupName, JobAction.暂停, taskOptions);
|
}
|
|
/// <summary>
|
/// 启动作业
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="taskOptions"></param>
|
/// <returns></returns>
|
public static Task<object> Start(this ISchedulerFactory schedulerFactory, TaskOptions taskOptions)
|
{
|
return schedulerFactory.TriggerAction(taskOptions.TaskName, taskOptions.GroupName, JobAction.开启, taskOptions);
|
}
|
|
/// <summary>
|
/// 立即执行一次作业
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="taskOptions"></param>
|
/// <returns></returns>
|
public static Task<object> Run(this ISchedulerFactory schedulerFactory, TaskOptions taskOptions)
|
{
|
return schedulerFactory.TriggerAction(taskOptions.TaskName, taskOptions.GroupName, JobAction.立即执行, taskOptions);
|
}
|
|
public static object ModifyTaskEntity(this TaskOptions taskOptions, ISchedulerFactory schedulerFactory, JobAction action)
|
{
|
TaskOptions options = null;
|
object result = null;
|
switch (action)
|
{
|
case JobAction.删除:
|
for (int i = 0; i < _taskList.Count; i++)
|
{
|
options = _taskList[i];
|
if (options.TaskName == taskOptions.TaskName && options.GroupName == taskOptions.GroupName)
|
{
|
_taskList.RemoveAt(i);
|
//删除任务
|
bool aa = QuartzSearchData.QuartzDelete(options);
|
}
|
}
|
break;
|
case JobAction.修改:
|
options = _taskList.Where(x => x.TaskName == taskOptions.TaskName && x.GroupName == taskOptions.GroupName).FirstOrDefault();
|
//移除以前的配置
|
if (options != null)
|
{
|
_taskList.Remove(options);
|
//删除任务
|
bool aa = QuartzSearchData.QuartzDelete(options);
|
}
|
|
//生成任务并添加新配置
|
result = taskOptions.AddJob(schedulerFactory, false).GetAwaiter().GetResult();
|
break;
|
case JobAction.暂停:
|
case JobAction.开启:
|
case JobAction.停止:
|
case JobAction.立即执行:
|
options = _taskList.Where(x => x.TaskName == taskOptions.TaskName && x.GroupName == taskOptions.GroupName).FirstOrDefault();
|
if (action == JobAction.暂停)
|
{
|
options.Status = (int)TriggerState.Paused;
|
//更新任务状态
|
bool aa= QuartzSearchData.QuartzUpdate(options);
|
}
|
else if (action == JobAction.停止)
|
{
|
options.Status = (int)action;
|
//更新任务状态
|
bool aa = QuartzSearchData.QuartzUpdate(options);
|
}
|
else
|
{
|
options.Status = (int)TriggerState.Normal;
|
//更新任务状态
|
bool aa = QuartzSearchData.QuartzUpdate(options);
|
}
|
break;
|
}
|
//生成配置文件
|
FileQuartz.WriteJobConfig(_taskList);
|
FileQuartz.WriteJobAction(action, taskOptions.TaskName, taskOptions.GroupName, "操作对象：" + JsonConvert.SerializeObject(taskOptions));
|
return result;
|
}
|
|
/// <summary>
|
/// 触发新增、删除、修改、暂停、启用、立即执行事件
|
/// </summary>
|
/// <param name="schedulerFactory"></param>
|
/// <param name="taskName"></param>
|
/// <param name="groupName"></param>
|
/// <param name="action"></param>
|
/// <param name="taskOptions"></param>
|
/// <returns></returns>
|
public static async Task<object> TriggerAction(this ISchedulerFactory schedulerFactory, string taskName, string groupName, JobAction action, TaskOptions taskOptions = null)
|
{
|
string errorMsg = "";
|
try
|
{
|
//处理前端Corn表达式最后一个月的星期天表达式缺陷问题
|
string resultString = "";
|
string[] str = taskOptions.Interval.Split(" ");
|
if ((str[3].Equals("1L") || str[3].Equals("2L") || str[3].Equals("3L") || str[3].Equals("4L") || str[3].Equals("5L") || str[3].Equals("6L") || str[3].Equals("7L")) && str[5].Equals("?"))
|
{
|
str[5] = str[3];
|
str[3] = "?";
|
for (int i = 0; i < str.Length; i++)
|
{
|
resultString += str[i] + " ";
|
}
|
taskOptions.Interval = resultString;
|
}
|
IScheduler scheduler = await schedulerFactory.GetScheduler();
|
List<JobKey> jobKeys = scheduler.GetJobKeys(GroupMatcher<JobKey>.GroupEquals(groupName)).Result.ToList();
|
if (jobKeys == null || jobKeys.Count() == 0)
|
{
|
errorMsg = $"未找到分组[{groupName}]";
|
return new { code = 300, count = 0, Message = errorMsg, data = "null" };
|
//return new { status = false, msg = errorMsg };
|
}
|
JobKey jobKey = jobKeys.Where(s => scheduler.GetTriggersOfJob(s).Result.Any(x => (x as CronTriggerImpl).Name == taskName)).FirstOrDefault();
|
if (jobKey == null)
|
{
|
errorMsg = $"未找到触发器[{taskName}]";
|
return new { code = 300, count = 0, Message = errorMsg, data = "null" };
|
}
|
var triggers = await scheduler.GetTriggersOfJob(jobKey);
|
ITrigger trigger = triggers?.Where(x => (x as CronTriggerImpl).Name == taskName).FirstOrDefault();
|
|
if (trigger == null)
|
{
|
errorMsg = $"未找到触发器[{taskName}]";
|
return new { code = 300, count = 0, Message = errorMsg, data = "null" };
|
}
|
object result = null;
|
switch (action)
|
{
|
case JobAction.删除:
|
case JobAction.修改:
|
await scheduler.PauseTrigger(trigger.Key);
|
await scheduler.UnscheduleJob(trigger.Key);// 移除触发器
|
await scheduler.DeleteJob(trigger.JobKey);
|
result = taskOptions.ModifyTaskEntity(schedulerFactory, action);
|
break;
|
case JobAction.暂停:
|
case JobAction.停止:
|
case JobAction.开启:
|
result = taskOptions.ModifyTaskEntity(schedulerFactory, action);
|
if (action == JobAction.暂停)
|
{
|
await scheduler.PauseTrigger(trigger.Key);
|
}
|
else if (action == JobAction.开启)
|
{
|
await scheduler.ResumeTrigger(trigger.Key);
|
// await scheduler.RescheduleJob(trigger.Key, trigger);
|
}
|
else
|
{
|
await scheduler.Shutdown();
|
}
|
break;
|
case JobAction.立即执行:
|
if (taskOptions != null && taskOptions.Status != (int)TriggerState.Normal)
|
{
|
result = taskOptions.ModifyTaskEntity(schedulerFactory, JobAction.开启);
|
await scheduler.ResumeTrigger(trigger.Key);
|
|
}
|
else {
|
await scheduler.TriggerJob(jobKey);
|
}
|
|
break;
|
}
|
mes.code = "200";
|
mes.count = 0;
|
mes.Message = $"作业{action.ToString()}成功";
|
mes.data = null;
|
return result ?? mes;
|
//return result ?? new { status = true, msg = $"作业{action.ToString()}成功" };
|
}
|
catch (Exception ex)
|
{
|
errorMsg = ex.Message;
|
mes.code = "300";
|
mes.count = 0;
|
mes.Message = errorMsg;
|
mes.data = null;
|
return mes;
|
//return new { status = false, msg = ex.Message };
|
}
|
finally
|
{
|
FileQuartz.WriteJobAction(action, taskName, groupName, errorMsg);
|
}
|
}
|
|
/// <summary>
|
///
|
/// </summary>
|
/// <param name="context"></param>通过作业上下文获取作业对应的配置参数
|
/// <returns></returns>
|
public static TaskOptions GetTaskOptions(this IJobExecutionContext context)
|
{
|
AbstractTrigger trigger = (context as JobExecutionContextImpl).Trigger as AbstractTrigger;
|
TaskOptions taskOptions = _taskList.Where(x => x.TaskName == trigger.Name && x.GroupName == trigger.Group).FirstOrDefault();
|
return taskOptions ?? _taskList.Where(x => x.TaskName == trigger.JobName && x.GroupName == trigger.JobGroup).FirstOrDefault();
|
}
|
|
/// <summary>
|
/// 作业是否存在
|
/// </summary>
|
/// <param name="taskOptions"></param>
|
/// <param name="init">初始化的不需要判断</param>
|
/// <returns></returns>
|
public static (bool, object) Exists(this TaskOptions taskOptions, bool init)
|
{
|
if (!init && _taskList.Any(x => x.TaskName == taskOptions.TaskName && x.GroupName == taskOptions.GroupName))
|
{
|
return (false,
|
new
|
{
|
status = false,
|
msg = $"作业:{taskOptions.TaskName},分组：{taskOptions.GroupName}已经存在"
|
});
|
}
|
return (true, null);
|
}
|
|
public static (bool, string) IsValidExpression(this string cronExpression)
|
{
|
try
|
{
|
CronTriggerImpl trigger = new CronTriggerImpl();
|
trigger.CronExpressionString = cronExpression;
|
DateTimeOffset? date = trigger.ComputeFirstFireTimeUtc(null);
|
return (date != null, date == null ? $"请确认表达式{cronExpression}是否正确!" : "");
|
}
|
catch (Exception e)
|
{
|
return (false, $"请确认表达式{cronExpression}是否正确!{e.Message}");
|
}
|
}
|
}
|
|
}
|
