132 lines
4.5 KiB
C#
132 lines
4.5 KiB
C#
#if CSHARP_7_OR_LATER || (UNITY_2018_3_OR_NEWER && (NET_STANDARD_2_0 || NET_4_6))
|
|
#pragma warning disable CS1591 // Missing XML comment for publicly visible type or member
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Threading;
|
|
using UniRx.Async.Internal;
|
|
|
|
namespace UniRx.Async
|
|
{
|
|
// public for add user custom.
|
|
|
|
public static class TaskTracker
|
|
{
|
|
#if UNITY_EDITOR
|
|
|
|
static int trackingId = 0;
|
|
|
|
public const string EnableAutoReloadKey = "UniTaskTrackerWindow_EnableAutoReloadKey";
|
|
public const string EnableTrackingKey = "UniTaskTrackerWindow_EnableTrackingKey";
|
|
public const string EnableStackTraceKey = "UniTaskTrackerWindow_EnableStackTraceKey";
|
|
|
|
public static class EditorEnableState
|
|
{
|
|
static bool enableAutoReload;
|
|
public static bool EnableAutoReload
|
|
{
|
|
get { return enableAutoReload; }
|
|
set
|
|
{
|
|
enableAutoReload = value;
|
|
UnityEditor.EditorPrefs.SetBool(EnableAutoReloadKey, value);
|
|
}
|
|
}
|
|
|
|
static bool enableTracking;
|
|
public static bool EnableTracking
|
|
{
|
|
get { return enableTracking; }
|
|
set
|
|
{
|
|
enableTracking = value;
|
|
UnityEditor.EditorPrefs.SetBool(EnableTrackingKey, value);
|
|
}
|
|
}
|
|
|
|
static bool enableStackTrace;
|
|
public static bool EnableStackTrace
|
|
{
|
|
get { return enableStackTrace; }
|
|
set
|
|
{
|
|
enableStackTrace = value;
|
|
UnityEditor.EditorPrefs.SetBool(EnableStackTraceKey, value);
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
static List<KeyValuePair<IUniTaskSource, (int trackingId, DateTime addTime, string stackTrace)>> listPool = new List<KeyValuePair<IUniTaskSource, (int trackingId, DateTime addTime, string stackTrace)>>();
|
|
|
|
static readonly WeakDictionary<IUniTaskSource, (int trackingId, DateTime addTime, string stackTrace)> tracking = new WeakDictionary<IUniTaskSource, (int trackingId, DateTime addTime, string stackTrace)>();
|
|
|
|
[Conditional("UNITY_EDITOR")]
|
|
public static void TrackActiveTask(IUniTaskSource task, int skipFrame)
|
|
{
|
|
#if UNITY_EDITOR
|
|
dirty = true;
|
|
if (!EditorEnableState.EnableTracking) return;
|
|
var stackTrace = EditorEnableState.EnableStackTrace ? new StackTrace(skipFrame, true).CleanupAsyncStackTrace() : "";
|
|
tracking.TryAdd(task, (Interlocked.Increment(ref trackingId), DateTime.UtcNow, stackTrace));
|
|
#endif
|
|
}
|
|
|
|
[Conditional("UNITY_EDITOR")]
|
|
public static void RemoveTracking(IUniTaskSource task)
|
|
{
|
|
#if UNITY_EDITOR
|
|
dirty = true;
|
|
if (!EditorEnableState.EnableTracking) return;
|
|
var success = tracking.TryRemove(task);
|
|
#endif
|
|
}
|
|
|
|
static bool dirty;
|
|
|
|
public static bool CheckAndResetDirty()
|
|
{
|
|
var current = dirty;
|
|
dirty = false;
|
|
return current;
|
|
}
|
|
|
|
/// <summary>(trackingId, awaiterType, awaiterStatus, createdTime, stackTrace)</summary>
|
|
public static void ForEachActiveTask(Action<int, string, UniTaskStatus, DateTime, string> action)
|
|
{
|
|
lock (listPool)
|
|
{
|
|
var count = tracking.ToList(ref listPool, clear: false);
|
|
try
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
string typeName = null;
|
|
var keyType = listPool[i].Key.GetType();
|
|
if (keyType.IsNested)
|
|
{
|
|
typeName = keyType.DeclaringType.Name + "." + keyType.Name;
|
|
}
|
|
else
|
|
{
|
|
typeName = keyType.Name;
|
|
}
|
|
|
|
action(listPool[i].Value.trackingId, typeName, listPool[i].Key.UnsafeGetStatus(), listPool[i].Value.addTime, listPool[i].Value.stackTrace);
|
|
listPool[i] = new KeyValuePair<IUniTaskSource, (int trackingId, DateTime addTime, string stackTrace)>(null, (0, default(DateTime), null)); // clear
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
listPool.Clear();
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif |