一个基于时间轮原理的定时器
其实我是有一篇文章(.Net 之时间轮算法(终极版))针对时间轮的理论理解的，但是，我想，为啥我看完时间轮原理后，会采用这样的方式去实现。
可能只是一些小技巧不上大雅之堂吧，大佬看看就行了。
当然如果大佬有别的看法，也请不吝赐教，互相交流，一起进步。
作用么，造个小轮子，顺便，对任务调度的实现多一些深度的思考和了解。
实现了对方法的定时 循环执行。 大概样子是下面这样的
TimeWheel timeWheel = new TimeWheel();
timeWheel.AddTask(new Job("定时1", () => { Console.WriteLine($"定时每1秒 {DateTime.Now}"); }, new TimeTask(TimeSpan.FromSeconds(1))));
timeWheel.AddTask(new Job("定时2", () => { Console.WriteLine($"定时2每10执行 {DateTime.Now}"); }, new TimeTask(TimeSpan.FromSeconds(10))));
timeWheel.AddTask(new Job("CRON", () => { Console.WriteLine($"CRON 每5秒 {DateTime.Now}"); }, new CronTask("*/5 * * * * *")));
timeWheel.AddTask(new Job("死信", () => { Console.WriteLine($"死信执行 {DateTime.Now}"); }, new DelayTask(TimeSpan.FromSeconds(20))));
timeWheel.AddTask(new Job("死信1", () => { Console.WriteLine($"死信1执行 {DateTime.Now}"); }, new DelayTask(TimeSpan.FromSeconds(10))));
timeWheel.Run();
能实现，定时任务，死信任务，能支持CRON表达式
timeWheel.AddTask(new Job("定时1", () => { Console.WriteLine($"定时每1秒 {DateTime.Now}"); }, new TimeTask(TimeSpan.FromSeconds(1))));
timeWheel.AddTask(new Job("定时2", () => { Console.WriteLine($"定时2每10执行 {DateTime.Now}"); }, new TimeTask(TimeSpan.FromSeconds(10))));
通过 TimeTask进行实现的
主要是基于 NCrontab 库，实现对CRON表达式的解析。省的自己从头解析了
timeWheel.AddTask(new Job("CRON", () => { Console.WriteLine($"CRON 每5秒 {DateTime.Now}"); }, new CronTask("*/5 * * * * *")));
这样就能实现对特定任务的执行
很多死信都是基于消息队列的，但是应该也有一些实际应用中的应用场景吧。看具体了。
timeWheel.AddTask(new Job("死信", () => { Console.WriteLine($"死信执行 {DateTime.Now}"); }, new DelayTask(TimeSpan.FromSeconds(20))));
timeWheel.AddTask(new Job("死信1", () => { Console.WriteLine($"死信1执行 {DateTime.Now}"); }, new DelayTask(TimeSpan.FromSeconds(10))));
实现
比如下边的，就能直接移除死信的任务。可以别的定时器执行了任务，然后，对此任务进行清除。
timeWheel.RemoveTask("死信");
基本上，只要没有被执行的任务，都会被取消执行的。
先看看main函数的示例
static void Main(string[] args)
{
TimeWheel timeWheel = new TimeWheel();
timeWheel.AddTask(new Job("定时1", () => { Console.WriteLine($"定时每1秒 {DateTime.Now}"); }, new TimeTask(TimeSpan.FromSeconds(1))));
timeWheel.AddTask(new Job("定时2", () => { Console.WriteLine($"定时2每10执行 {DateTime.Now}"); }, new TimeTask(TimeSpan.FromSeconds(10))));
timeWheel.AddTask(new Job("CRON", () => { Console.WriteLine($"CRON 每5秒 {DateTime.Now}"); }, new CronTask("*/5 * * * * *")));
timeWheel.AddTask(new Job("死信", () => { Console.WriteLine($"死信执行 {DateTime.Now}"); }, new DelayTask(TimeSpan.FromSeconds(20))));
timeWheel.AddTask(new Job("死信1", () => { Console.WriteLine($"死信1执行 {DateTime.Now}"); }, new DelayTask(TimeSpan.FromSeconds(10))));
timeWheel.Run();
Task.Run(() =>
{
Thread.Sleep(10 * 1000);
timeWheel.RemoveTask("死信");
Console.WriteLine("移除死信");
Thread.Sleep(10 * 1000);
timeWheel.RemoveTask("CRON");
Console.WriteLine("移除任务CRON");
});
Console.WriteLine("开始运行时间轮!");
Console.ReadLine();
} /// <summary>
/// 时间调度方式
/// </summary>
public interface IScheduledTask
{
/// <summary>
/// 获取下一个时间
/// </summary>
/// <returns></returns>
public DateTime? GetNextTime();
} /// <summary>
/// 时间轮算法(终极)实现
/// 大部分都是支持秒级，所以，按照秒级进行实现
/// 任务体得有它自己的任务唯一的ID
/// </summary>
public class TimeWheel
{
/// <summary>
/// 时间调度列表
/// </summary>
private ConcurrentDictionary<long, HashSet<string>> TimeTasks { get; set; } = new();
/// <summary>
/// 任务列表
/// </summary>
private ConcurrentDictionary<string, IJob> ScheduledTasks { get; set; } = new();
/// <summary>
/// 是否运行中
/// </summary>
private bool isRuning = false;
/// <summary>
/// 运行核心
/// </summary>
public void Run()
{
isRuning = true;
Task.Run(() =>
{
while (isRuning)
{
var timeStamp = GenerateTimestamp(DateTime.Now);
Task.Run(() => { Trigger(timeStamp); });
var offset = 500 - DateTime.Now.Millisecond;
SpinWait.SpinUntil(() => false, 1000 + offset);
}
});
}
public void Stop()
{
isRuning = false;
}
/// <summary>
/// 定时触发器
/// </summary>
/// <param name="timeStamp"></param>
private void Trigger(long timeStamp)
{
var oldTimeStamp = timeStamp - 1;
var list = TimeTasks.Keys.Where(t => t <= oldTimeStamp).ToList();
foreach (var item in list)
{
TimeTasks.TryRemove(item, out var _);
}
TimeTasks.TryGetValue(timeStamp, out var result);
if (result?.Any() == true)
{
var Now = DateTime.Now;
foreach (var id in result)
{
//找到指定的任务
if (ScheduledTasks.TryGetValue(id, out IJob job))
{
Task.Run(() => { job.Execute(); });
var NewTime = job.GetNextTime();
if (NewTime.HasValue && NewTime >= Now)
{
AddTask(NewTime.Value, id);
}
}
}
}
}
/// <summary>
/// 添加任务
/// </summary>
/// <param name="dateTime"></param>
/// <param name="scheduledTask"></param>
private void AddTask(DateTime dateTime, string ID)
{
var timeStamp = GenerateTimestamp(dateTime);
TimeTasks.AddOrUpdate(timeStamp, new HashSet<string>() { ID }, (k, v) =>
{
v.Add(ID);
return v;
});
}
/// <summary>
/// 增加一个任务
/// </summary>
public void AddTask(IJob job)
{
if (ScheduledTasks.ContainsKey(job.ID))
{
throw new ArgumentException($"{nameof(job)} 参数 {nameof(job.ID)}重复!");
}
else
{
ScheduledTasks.TryAdd(job.ID, job);
}
var time = DateTime.Now;
var NewTime = job.GetNextTime();
if (NewTime.HasValue && NewTime >= time)
{
Console.WriteLine($"新增任务：{job.ID}");
AddTask(NewTime.Value, job.ID);
}
}
/// <summary>
/// 移除某个任务的Task
/// </summary>
/// <param name="ID"></param>
public void RemoveTask(string ID)
{
var ids = ScheduledTasks.Values.Where(t => t.ID == ID)?.Select(t => t.ID).ToList();
if (ids?.Any() == true)
{
foreach (var id in ids)
{
if (ScheduledTasks.TryGetValue(id, out var job))
{
job.Cancel();
ScheduledTasks.TryRemove(id, out _);
}
}
}
}
/// <summary>
/// 获取时间戳
/// </summary>
private long GenerateTimestamp(DateTime dateTime)
{
return new DateTimeOffset(dateTime.ToUniversalTime()).ToUnixTimeSeconds();
}
} /// <summary>
/// 任务体
/// </summary>
public interface IJob
{
/// <summary>
/// 任务ID,唯一
/// </summary>
/// <returns></returns>
public string ID { get; }
/// <summary>
/// 脚本
/// </summary>
/// <returns></returns>
public void Execute();
/// <summary>
/// 取消执行
/// </summary>
public void Cancel();
/// <summary>
/// 获取任务执行时间
/// </summary>
/// <returns></returns>
public DateTime? GetNextTime();
}只有一个字，轻。 用的舒服点。 有问题大家一起沟通