mirror of https://github.com/Cysharp/UniTask
77 lines
2.1 KiB
C#
77 lines
2.1 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.Threading;
|
|
using UnityEngine;
|
|
|
|
namespace UniRx.Async.Triggers
|
|
{
|
|
public static partial class AsyncTriggerExtensions
|
|
{
|
|
public static AsyncAwakeTrigger GetAsyncAwakeTrigger(this GameObject gameObject)
|
|
{
|
|
return GetOrAddComponent<AsyncAwakeTrigger>(gameObject);
|
|
}
|
|
|
|
public static AsyncAwakeTrigger GetAsyncAwakeTrigger(this Component component)
|
|
{
|
|
return component.gameObject.GetAsyncAwakeTrigger();
|
|
}
|
|
}
|
|
|
|
[DisallowMultipleComponent]
|
|
public class AsyncAwakeTrigger : MonoBehaviour
|
|
{
|
|
bool called = false;
|
|
TriggerEvent<AsyncUnit> triggerEvent;
|
|
|
|
void Awake()
|
|
{
|
|
called = true;
|
|
triggerEvent?.TrySetResult(AsyncUnit.Default);
|
|
triggerEvent = null;
|
|
}
|
|
|
|
public UniTask AwakeAsync()
|
|
{
|
|
if (called) return UniTask.CompletedTask;
|
|
|
|
PlayerLoopHelper.AddAction(PlayerLoopTiming.Update, new AwakeMonitor(this));
|
|
|
|
if (triggerEvent == null)
|
|
{
|
|
triggerEvent = new TriggerEvent<AsyncUnit>();
|
|
}
|
|
|
|
return ((IAsyncOneShotTrigger)new AsyncTriggerHandler<AsyncUnit>(triggerEvent, true)).OneShotAsync();
|
|
}
|
|
|
|
private void OnDestroy()
|
|
{
|
|
triggerEvent?.TrySetCanceled(CancellationToken.None);
|
|
}
|
|
|
|
class AwakeMonitor : IPlayerLoopItem
|
|
{
|
|
readonly AsyncAwakeTrigger trigger;
|
|
|
|
public AwakeMonitor(AsyncAwakeTrigger trigger)
|
|
{
|
|
this.trigger = trigger;
|
|
}
|
|
|
|
public bool MoveNext()
|
|
{
|
|
if (trigger.called) return false;
|
|
if (trigger == null)
|
|
{
|
|
trigger.OnDestroy();
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif |