pengtaihao
/
UniTask
mirror of https://github.com/Cysharp/UniTask
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

guard for trigger invalid ops

pull/61/head
neuecc 2020-06-01 13:51:37 +09:00
parent 130286e8c2
commit d5edc3acd3
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/UniTask/Assets/Plugins/UniTask/Runtime/TriggerEvent.cs
View File

@ -24,7 +24,6 @@ namespace Cysharp.Threading.Tasks
bool preserveRemoveSelf; bool preserveRemoveSelf;
ITriggerHandler<T> iteratingNode; ITriggerHandler<T> iteratingNode;
void LogError(Exception ex) void LogError(Exception ex)
{ {
#if UNITY_2018_3_OR_NEWER #if UNITY_2018_3_OR_NEWER
@ -36,6 +35,11 @@ namespace Cysharp.Threading.Tasks
public void SetResult(T value) public void SetResult(T value)
{ {
if (iteratingNode != null)
{
throw new InvalidOperationException("Can not trigger itself in iterating.");
}
var h = head; var h = head;
while (h != null) while (h != null)
{ {
@ -75,6 +79,11 @@ namespace Cysharp.Threading.Tasks
public void SetCanceled(CancellationToken cancellationToken) public void SetCanceled(CancellationToken cancellationToken)
{ {
if (iteratingNode != null)
{
throw new InvalidOperationException("Can not trigger itself in iterating.");
}
var h = head; var h = head;
while (h != null) while (h != null)
{ {
@ -105,6 +114,11 @@ namespace Cysharp.Threading.Tasks
public void SetCompleted() public void SetCompleted()
{ {
if (iteratingNode != null)
{
throw new InvalidOperationException("Can not trigger itself in iterating.");
}
var h = head; var h = head;
while (h != null) while (h != null)
{ {
@ -135,6 +149,11 @@ namespace Cysharp.Threading.Tasks
public void SetError(Exception exception) public void SetError(Exception exception)
{ {
if (iteratingNode != null)
{
throw new InvalidOperationException("Can not trigger itself in iterating.");
}
var h = head; var h = head;
while (h != null) while (h != null)
{ {