557 lines
16 KiB
C#
557 lines
16 KiB
C#
using System;
|
|
using System.Diagnostics;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace YooAsset
|
|
{
|
|
public abstract class AsyncOperationBase : IEnumerator, IComparable<AsyncOperationBase>
|
|
{
|
|
private List<AsyncOperationBase> _childs;
|
|
private Action<AsyncOperationBase> _callback;
|
|
private uint _priority;
|
|
|
|
/// <summary>
|
|
/// 等待异步执行完成
|
|
/// </summary>
|
|
internal bool IsWaitForAsyncComplete { get; private set; }
|
|
|
|
/// <summary>
|
|
/// 标记脏（用于调度器检测并重排）
|
|
/// </summary>
|
|
internal bool IsDirty { get; set; }
|
|
|
|
/// <summary>
|
|
/// 是否已经完成
|
|
/// </summary>
|
|
internal bool IsFinish { get; private set; }
|
|
|
|
/// <summary>
|
|
/// 异步系统是否繁忙
|
|
/// </summary>
|
|
internal bool IsBusy
|
|
{
|
|
get
|
|
{
|
|
if (IsWaitForAsyncComplete)
|
|
return false;
|
|
return OperationSystem.IsBusy;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 任务优先级
|
|
/// </summary>
|
|
public uint Priority
|
|
{
|
|
set
|
|
{
|
|
if (_priority == value)
|
|
return;
|
|
_priority = value;
|
|
IsDirty = true;
|
|
}
|
|
get
|
|
{
|
|
return _priority;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 任务状态
|
|
/// </summary>
|
|
public EOperationStatus Status { get; protected set; } = EOperationStatus.None;
|
|
|
|
/// <summary>
|
|
/// 错误信息
|
|
/// </summary>
|
|
public string Error { get; protected set; }
|
|
|
|
/// <summary>
|
|
/// 处理进度
|
|
/// </summary>
|
|
public float Progress { get; protected set; }
|
|
|
|
/// <summary>
|
|
/// 是否已经完成
|
|
/// </summary>
|
|
public bool IsDone
|
|
{
|
|
get
|
|
{
|
|
return Status == EOperationStatus.Failed || Status == EOperationStatus.Succeed;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 完成事件
|
|
/// </summary>
|
|
public event Action<AsyncOperationBase> Completed
|
|
{
|
|
add
|
|
{
|
|
if (value == null)
|
|
return;
|
|
|
|
if (IsDone)
|
|
{
|
|
try
|
|
{
|
|
//注意：任务已完成，立即调用回调
|
|
value.Invoke(this);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
YooLogger.Error($"Exception in completion callback: {ex}");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_callback += value;
|
|
}
|
|
}
|
|
remove
|
|
{
|
|
_callback -= value;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 异步操作任务
|
|
/// </summary>
|
|
public Task Task
|
|
{
|
|
get
|
|
{
|
|
if (_taskCompletionSource == null)
|
|
{
|
|
_taskCompletionSource = new TaskCompletionSource<object>();
|
|
if (IsDone)
|
|
_taskCompletionSource.SetResult(null);
|
|
}
|
|
return _taskCompletionSource.Task;
|
|
}
|
|
}
|
|
|
|
internal abstract void InternalStart();
|
|
internal abstract void InternalUpdate();
|
|
internal virtual void InternalAbort()
|
|
{
|
|
}
|
|
internal virtual void InternalWaitForAsyncComplete()
|
|
{
|
|
throw new YooInternalException($"InternalWaitForAsyncComplete() not implemented : {this.GetType().Name}");
|
|
}
|
|
internal virtual string InternalGetDesc()
|
|
{
|
|
return string.Empty;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 添加子任务
|
|
/// </summary>
|
|
internal void AddChildOperation(AsyncOperationBase child)
|
|
{
|
|
if (_childs == null)
|
|
_childs = new List<AsyncOperationBase>(10);
|
|
|
|
#if UNITY_EDITOR || DEBUG
|
|
if (child == null)
|
|
throw new YooInternalException("The child node is null.");
|
|
|
|
if (ReferenceEquals(child, this))
|
|
throw new YooInternalException("The child node cannot be itself.");
|
|
|
|
if (_childs.Contains(child))
|
|
throw new YooInternalException($"The child node {child.GetType().Name} already exists.");
|
|
|
|
// 禁止形成环依赖
|
|
if (WouldCreateCycle(child))
|
|
throw new YooInternalException($"AddChildOperation would create a cycle : {this.GetType().Name} -> {child.GetType().Name}");
|
|
#endif
|
|
|
|
_childs.Add(child);
|
|
}
|
|
|
|
/// <summary>
|
|
/// 移除子任务
|
|
/// </summary>
|
|
internal void RemoveChildOperation(AsyncOperationBase child)
|
|
{
|
|
if (_childs == null)
|
|
return;
|
|
|
|
#if UNITY_EDITOR || DEBUG
|
|
if (child == null)
|
|
throw new YooInternalException("The child node is null.");
|
|
|
|
if (_childs.Contains(child) == false)
|
|
throw new YooInternalException($"The child node {child.GetType().Name} not exists.");
|
|
#endif
|
|
|
|
_childs.Remove(child);
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取异步操作说明
|
|
/// </summary>
|
|
internal string GetOperationDesc()
|
|
{
|
|
return InternalGetDesc();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 开始异步操作
|
|
/// </summary>
|
|
internal void StartOperation()
|
|
{
|
|
if (Status == EOperationStatus.None)
|
|
{
|
|
Status = EOperationStatus.Processing;
|
|
|
|
// 开始记录
|
|
DebugBeginRecording();
|
|
|
|
// 开始任务
|
|
try
|
|
{
|
|
InternalStart();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Status = EOperationStatus.Failed;
|
|
Error = ex.ToString();
|
|
YooLogger.Error($"Exception in {this.GetType().Name}.InternalStart : {ex}");
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 更新异步操作
|
|
/// </summary>
|
|
internal void UpdateOperation()
|
|
{
|
|
if (IsDone == false)
|
|
{
|
|
// 更新记录
|
|
DebugUpdateRecording();
|
|
|
|
// 更新任务
|
|
// 注意：兜底隔离机制
|
|
// 说明：检测的异常源包含：I/O（解压/读写权限/磁盘满），平台差异等
|
|
try
|
|
{
|
|
InternalUpdate();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Status = EOperationStatus.Failed;
|
|
Error = ex.ToString();
|
|
YooLogger.Error($"Exception in {this.GetType().Name}.InternalUpdate : {ex}");
|
|
}
|
|
}
|
|
|
|
if (IsDone && IsFinish == false)
|
|
{
|
|
FinishOperation();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 终止异步任务
|
|
/// </summary>
|
|
internal void AbortOperation()
|
|
{
|
|
if (_childs != null)
|
|
{
|
|
foreach (var child in _childs)
|
|
{
|
|
child.AbortOperation();
|
|
}
|
|
}
|
|
|
|
if (IsDone == false)
|
|
{
|
|
InternalAbort();
|
|
Status = EOperationStatus.Failed;
|
|
Error = "user abort";
|
|
YooLogger.Warning($"Async operation {this.GetType().Name} has been aborted.");
|
|
}
|
|
|
|
//注意：强制收尾，确保Task能完成
|
|
FinishOperation();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 强制结束异步任务
|
|
/// </summary>
|
|
private void FinishOperation()
|
|
{
|
|
if (IsFinish == false)
|
|
{
|
|
IsFinish = true;
|
|
Progress = 1f;
|
|
|
|
// 结束记录
|
|
DebugEndRecording();
|
|
|
|
if (_callback != null)
|
|
{
|
|
var invocationList = _callback.GetInvocationList();
|
|
foreach (var handler in invocationList)
|
|
{
|
|
try
|
|
{
|
|
((Action<AsyncOperationBase>)handler).Invoke(this);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
YooLogger.Error($"Exception in completion callback: {ex}");
|
|
}
|
|
}
|
|
}
|
|
|
|
_callback = null;
|
|
if (_taskCompletionSource != null)
|
|
_taskCompletionSource.TrySetResult(null);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 执行一次更新逻辑
|
|
/// </summary>
|
|
protected void RunOnceExecution()
|
|
{
|
|
if (IsDone)
|
|
return;
|
|
|
|
UpdateOperation();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 批量执行一定次数的更新逻辑
|
|
/// </summary>
|
|
/// <param name="count">最大执行次数，默认1000次</param>
|
|
/// <remarks>
|
|
/// 用于需要快速完成但又不想完全阻塞主线程的场景。
|
|
/// </remarks>
|
|
protected void RunBatchExecution(int count = 1000)
|
|
{
|
|
if (IsDone)
|
|
return;
|
|
|
|
int runCount = count;
|
|
while (true)
|
|
{
|
|
// 执行更新逻辑
|
|
UpdateOperation();
|
|
if (IsDone)
|
|
break;
|
|
|
|
// 当执行次数用完时
|
|
runCount--;
|
|
if (runCount <= 0)
|
|
break;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 无限次数的执行更新逻辑，直到任务完成
|
|
/// 注意：该方法会阻塞主线程
|
|
/// </summary>
|
|
/// <param name="sleepMS">休眠时长</param>
|
|
protected void RunUntilCompletion(int sleepMS = 1)
|
|
{
|
|
if (IsDone)
|
|
return;
|
|
|
|
while (true)
|
|
{
|
|
UpdateOperation();
|
|
if (IsDone)
|
|
break;
|
|
|
|
// 注意： 短暂休眠避免完全占用CPU资源
|
|
System.Threading.Thread.Sleep(sleepMS);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 等待异步执行完毕
|
|
/// </summary>
|
|
public void WaitForAsyncComplete()
|
|
{
|
|
//注意：防止异步操作被挂起陷入无限死循环！
|
|
if (Status == EOperationStatus.None)
|
|
{
|
|
StartOperation();
|
|
}
|
|
|
|
if (IsWaitForAsyncComplete == false)
|
|
{
|
|
IsWaitForAsyncComplete = true;
|
|
|
|
if (IsDone == false)
|
|
InternalWaitForAsyncComplete();
|
|
|
|
if (IsDone == false)
|
|
{
|
|
Status = EOperationStatus.Failed;
|
|
Error = $"Operation {this.GetType().Name} failed to wait for async complete.";
|
|
YooLogger.Error(Error);
|
|
}
|
|
|
|
//注意：强制收尾，确保Task能完成
|
|
FinishOperation();
|
|
}
|
|
}
|
|
|
|
#region 调试信息
|
|
/// <summary>
|
|
/// 开始的时间
|
|
/// </summary>
|
|
public string BeginTime { get; protected set; }
|
|
|
|
/// <summary>
|
|
/// 处理耗时（单位：毫秒）
|
|
/// </summary>
|
|
public long ProcessTime { get; protected set; }
|
|
|
|
// 加载耗时统计
|
|
private Stopwatch _watch = null;
|
|
|
|
[Conditional("DEBUG")]
|
|
private void DebugBeginRecording()
|
|
{
|
|
if (_watch == null)
|
|
{
|
|
BeginTime = SpawnTimeToString(TimeUtility.RealtimeSinceStartup);
|
|
_watch = Stopwatch.StartNew();
|
|
}
|
|
}
|
|
|
|
[Conditional("DEBUG")]
|
|
private void DebugUpdateRecording()
|
|
{
|
|
if (_watch != null)
|
|
{
|
|
ProcessTime = _watch.ElapsedMilliseconds;
|
|
}
|
|
}
|
|
|
|
[Conditional("DEBUG")]
|
|
private void DebugEndRecording()
|
|
{
|
|
if (_watch != null)
|
|
{
|
|
ProcessTime = _watch.ElapsedMilliseconds;
|
|
_watch = null;
|
|
}
|
|
}
|
|
|
|
private string SpawnTimeToString(double spawnTime)
|
|
{
|
|
double h = System.Math.Floor(spawnTime / 3600);
|
|
double m = System.Math.Floor(spawnTime / 60 - h * 60);
|
|
double s = System.Math.Floor(spawnTime - m * 60 - h * 3600);
|
|
return h.ToString("00") + ":" + m.ToString("00") + ":" + s.ToString("00");
|
|
}
|
|
|
|
/// <summary>
|
|
/// 检测添加子任务是否会形成循环依赖
|
|
/// 使用深度优先搜索（DFS）遍历子任务图
|
|
/// </summary>
|
|
private bool WouldCreateCycle(AsyncOperationBase child)
|
|
{
|
|
const int MaxCycleCheckDepth = 4096; // 循环检测最大深度
|
|
var stack = new Stack<AsyncOperationBase>();
|
|
var visited = new HashSet<AsyncOperationBase>();
|
|
stack.Push(child);
|
|
|
|
while (stack.Count > 0)
|
|
{
|
|
var node = stack.Pop();
|
|
if (node == null)
|
|
continue;
|
|
|
|
// 防止重复访问
|
|
if (visited.Add(node) == false)
|
|
continue;
|
|
|
|
// 防止无限循环（图过大）
|
|
if (visited.Count > MaxCycleCheckDepth)
|
|
throw new YooInternalException("Child operation graph is too large, cycle check aborted.");
|
|
|
|
// 检测循环：如果遍历到自己，说明形成循环
|
|
if (ReferenceEquals(node, this))
|
|
return true;
|
|
|
|
if (node._childs == null)
|
|
continue;
|
|
|
|
// 将子节点加入栈
|
|
for (int i = 0; i < node._childs.Count; i++)
|
|
{
|
|
stack.Push(node._childs[i]);
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取调试信息
|
|
/// 注意：递归构建子树存在深度风险
|
|
/// </summary>
|
|
internal DebugOperationInfo GetDebugOperationInfo()
|
|
{
|
|
var operationInfo = new DebugOperationInfo();
|
|
operationInfo.OperationName = this.GetType().Name;
|
|
operationInfo.OperationDesc = GetOperationDesc();
|
|
operationInfo.Priority = Priority;
|
|
operationInfo.Progress = Progress;
|
|
operationInfo.BeginTime = BeginTime;
|
|
operationInfo.ProcessTime = ProcessTime;
|
|
operationInfo.Status = Status.ToString();
|
|
|
|
if (_childs == null)
|
|
{
|
|
operationInfo.Childs = new List<DebugOperationInfo>();
|
|
}
|
|
else
|
|
{
|
|
operationInfo.Childs = new List<DebugOperationInfo>(_childs.Count);
|
|
foreach (var child in _childs)
|
|
{
|
|
var childInfo = child.GetDebugOperationInfo();
|
|
operationInfo.Childs.Add(childInfo);
|
|
}
|
|
}
|
|
|
|
return operationInfo;
|
|
}
|
|
#endregion
|
|
|
|
#region 排序接口实现
|
|
public int CompareTo(AsyncOperationBase other)
|
|
{
|
|
return other.Priority.CompareTo(this.Priority);
|
|
}
|
|
#endregion
|
|
|
|
#region 异步编程相关
|
|
bool IEnumerator.MoveNext()
|
|
{
|
|
return !IsDone;
|
|
}
|
|
void IEnumerator.Reset()
|
|
{
|
|
}
|
|
object IEnumerator.Current => null;
|
|
|
|
private TaskCompletionSource<object> _taskCompletionSource;
|
|
#endregion
|
|
}
|
|
}
|