70 lines
1.5 KiB
C#
70 lines
1.5 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
|
|||
|
|
namespace ET
|
|||
|
|
{
|
|||
|
|
public class ETCancellationToken
|
|||
|
|
{
|
|||
|
|
private HashSet<Action> actions = new HashSet<Action>();
|
|||
|
|
|
|||
|
|
public void Add(Action callback)
|
|||
|
|
{
|
|||
|
|
// 如果action是null，绝对不能添加,要抛异常，说明有协程泄漏
|
|||
|
|
this.actions.Add(callback);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Remove(Action callback)
|
|||
|
|
{
|
|||
|
|
this.actions?.Remove(callback);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Cancel()
|
|||
|
|
{
|
|||
|
|
if (this.actions == null)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (this.actions.Count == 0)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.Invoke();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Invoke()
|
|||
|
|
{
|
|||
|
|
HashSet<Action> runActions = this.actions;
|
|||
|
|
this.actions = null;
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
foreach (Action action in runActions)
|
|||
|
|
{
|
|||
|
|
action.Invoke();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception e)
|
|||
|
|
{
|
|||
|
|
Log.Error(e);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public async ETVoid CancelAfter(long afterTimeCancel)
|
|||
|
|
{
|
|||
|
|
if (this.actions == null)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (this.actions.Count == 0)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
await TimerComponent.Instance.WaitAsync(afterTimeCancel);
|
|||
|
|
|
|||
|
|
this.Invoke();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|