110 lines
3.6 KiB
C#
110 lines
3.6 KiB
C#
using System.Reflection;
|
||
|
||
namespace CirculateTool;
|
||
/*
|
||
* 作者：菻蔃
|
||
* 版本时间：2023年04月15日
|
||
*
|
||
* 注意配合特性使用
|
||
*
|
||
*/
|
||
|
||
/// <summary>
|
||
/// 定时任务类
|
||
/// </summary>
|
||
public class StartCirculation
|
||
{
|
||
|
||
/// <summary>
|
||
/// 触发的异常
|
||
/// </summary>
|
||
public event ExceptionHandlerEvent? ExceptionHandler;
|
||
|
||
public delegate void ExceptionHandlerEvent(string methodDescription, Exception ex);
|
||
|
||
/// <summary>
|
||
/// 显示相关信息
|
||
/// </summary>
|
||
public event MessageHandlerEvent? MessageHandler;
|
||
|
||
public delegate void MessageHandlerEvent(string message);
|
||
|
||
/// <summary>
|
||
/// 默认的循环时间
|
||
/// </summary>
|
||
private readonly int _defaultCirculationTime = 500;
|
||
|
||
/// <summary>
|
||
/// 启动一个程序集里面带有<see cref="CirculationAttribute"/>的类里面的定时方法
|
||
/// </summary>
|
||
/// <param name="assembly"></param>
|
||
/// <param name="instanceParams"></param>
|
||
public virtual void StartAssemblyCirculation(Assembly assembly, object[]? instanceParams = null)
|
||
{
|
||
Type[] types = assembly.GetTypes();
|
||
if (types.Length == 0) return;
|
||
foreach (Type type in types)
|
||
{
|
||
var attributes = type.GetCustomAttributes(false);
|
||
foreach (var attribute in attributes)
|
||
{
|
||
if (attribute is not CirculationAttribute) continue;
|
||
StartTask(type, instanceParams);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 开启一个实例里面所有已经添加了特性<see cref="CirculationAttribute"/>的方法
|
||
/// </summary>
|
||
/// <param name="type"></param>
|
||
/// <param name="instanceParams"></param>
|
||
public virtual void StartTask(Type type, object[]? instanceParams = null)
|
||
{
|
||
var methods = type.GetMethods();
|
||
foreach (var method in methods)
|
||
{
|
||
var attributes = method.GetCustomAttributes(false);
|
||
foreach (var attribute in attributes)
|
||
{
|
||
if (attribute is not CirculationAttribute needDurable) continue;
|
||
string methodDescription = needDurable.MethodDescription ?? $"{type.Name}.{method.Name}";
|
||
bool Action() => (bool)(method.Invoke(Activator.CreateInstance(type, instanceParams), []) ?? false);
|
||
StartTask(Action, methodDescription, needDurable.CirculationTime);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 开启一个方法
|
||
/// </summary>
|
||
/// <param name="action"></param>
|
||
/// <param name="description"></param>
|
||
/// <param name="durableTime"></param>
|
||
/// <returns></returns>
|
||
public virtual async void StartTask(Func<bool> action, string? description = null, int? durableTime = null)
|
||
{
|
||
int durableTimeValue = durableTime ?? _defaultCirculationTime;
|
||
string methodDescription = description ?? action.Method.Name;
|
||
CancellationTokenSource cts = new();
|
||
PeriodicTimer timer = new(new TimeSpan(0, 0, 0, 0, durableTimeValue));
|
||
MessageHandler?.Invoke($"定时器：{methodDescription}，已经启动，执行间隔为：{durableTimeValue} 毫秒。");
|
||
while (await timer.WaitForNextTickAsync(cts.Token))
|
||
{
|
||
try
|
||
{
|
||
var result = action();
|
||
if (result) continue;
|
||
await cts.CancelAsync();
|
||
MessageHandler?.Invoke($"定时器：{methodDescription}，主动结束。");
|
||
return; // 该return会结束这个线程
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
ExceptionHandler?.Invoke(methodDescription, ex);
|
||
}
|
||
}
|
||
}
|
||
} |