Revert "Fixed https://github.com/Cysharp/UniTask/issues/444"
parent
548d56e654
commit
29a144694d
|
@ -20,6 +20,8 @@ namespace Cysharp.Threading.Tasks
|
|||
{
|
||||
ITriggerHandler<T> head; // head.prev is last
|
||||
ITriggerHandler<T> iteratingHead;
|
||||
|
||||
bool preserveRemoveSelf;
|
||||
ITriggerHandler<T> iteratingNode;
|
||||
|
||||
void LogError(Exception ex)
|
||||
|
@ -42,7 +44,6 @@ namespace Cysharp.Threading.Tasks
|
|||
while (h != null)
|
||||
{
|
||||
iteratingNode = h;
|
||||
var next = h.Next;
|
||||
|
||||
try
|
||||
{
|
||||
|
@ -54,7 +55,18 @@ namespace Cysharp.Threading.Tasks
|
|||
Remove(h);
|
||||
}
|
||||
|
||||
h = next;
|
||||
if (preserveRemoveSelf)
|
||||
{
|
||||
preserveRemoveSelf = false;
|
||||
iteratingNode = null;
|
||||
var next = h.Next;
|
||||
Remove(h);
|
||||
h = next;
|
||||
}
|
||||
else
|
||||
{
|
||||
h = h.Next;
|
||||
}
|
||||
}
|
||||
|
||||
iteratingNode = null;
|
||||
|
@ -85,6 +97,7 @@ namespace Cysharp.Threading.Tasks
|
|||
LogError(ex);
|
||||
}
|
||||
|
||||
preserveRemoveSelf = false;
|
||||
iteratingNode = null;
|
||||
var next = h.Next;
|
||||
Remove(h);
|
||||
|
@ -119,6 +132,7 @@ namespace Cysharp.Threading.Tasks
|
|||
LogError(ex);
|
||||
}
|
||||
|
||||
preserveRemoveSelf = false;
|
||||
iteratingNode = null;
|
||||
var next = h.Next;
|
||||
Remove(h);
|
||||
|
@ -153,6 +167,7 @@ namespace Cysharp.Threading.Tasks
|
|||
LogError(ex);
|
||||
}
|
||||
|
||||
preserveRemoveSelf = false;
|
||||
iteratingNode = null;
|
||||
var next = h.Next;
|
||||
Remove(h);
|
||||
|
@ -226,63 +241,71 @@ namespace Cysharp.Threading.Tasks
|
|||
{
|
||||
if (handler == null) throw new ArgumentNullException(nameof(handler));
|
||||
|
||||
var prev = handler.Prev;
|
||||
var next = handler.Next;
|
||||
|
||||
if (next != null)
|
||||
if (iteratingNode != null && iteratingNode == handler)
|
||||
{
|
||||
next.Prev = prev;
|
||||
}
|
||||
|
||||
if (handler == head)
|
||||
{
|
||||
head = next;
|
||||
}
|
||||
else if (handler == iteratingHead)
|
||||
{
|
||||
iteratingHead = next;
|
||||
// if remove self, reserve remove self after invoke completed.
|
||||
preserveRemoveSelf = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
// when handler is head, prev indicate last so don't use it.
|
||||
if (prev != null)
|
||||
{
|
||||
prev.Next = next;
|
||||
}
|
||||
}
|
||||
var prev = handler.Prev;
|
||||
var next = handler.Next;
|
||||
|
||||
if (head != null)
|
||||
{
|
||||
if (head.Prev == handler)
|
||||
if (next != null)
|
||||
{
|
||||
if (prev != head)
|
||||
next.Prev = prev;
|
||||
}
|
||||
|
||||
if (handler == head)
|
||||
{
|
||||
head = next;
|
||||
}
|
||||
else if (handler == iteratingHead)
|
||||
{
|
||||
iteratingHead = next;
|
||||
}
|
||||
else
|
||||
{
|
||||
// when handler is head, prev indicate last so don't use it.
|
||||
if (prev != null)
|
||||
{
|
||||
head.Prev = prev;
|
||||
}
|
||||
else
|
||||
{
|
||||
head.Prev = null;
|
||||
prev.Next = next;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (iteratingHead != null)
|
||||
{
|
||||
if (iteratingHead.Prev == handler)
|
||||
if (head != null)
|
||||
{
|
||||
if (prev != iteratingHead.Prev)
|
||||
if (head.Prev == handler)
|
||||
{
|
||||
iteratingHead.Prev = prev;
|
||||
}
|
||||
else
|
||||
{
|
||||
iteratingHead.Prev = null;
|
||||
if (prev != head)
|
||||
{
|
||||
head.Prev = prev;
|
||||
}
|
||||
else
|
||||
{
|
||||
head.Prev = null;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
handler.Prev = null;
|
||||
handler.Next = null;
|
||||
if (iteratingHead != null)
|
||||
{
|
||||
if (iteratingHead.Prev == handler)
|
||||
{
|
||||
if (prev != iteratingHead.Prev)
|
||||
{
|
||||
iteratingHead.Prev = prev;
|
||||
}
|
||||
else
|
||||
{
|
||||
iteratingHead.Prev = null;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
handler.Prev = null;
|
||||
handler.Next = null;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,63 +0,0 @@
|
|||
using Cysharp.Threading.Tasks;
|
||||
using FluentAssertions;
|
||||
using System;
|
||||
using System.Collections;
|
||||
using System.Threading;
|
||||
using UnityEngine.TestTools;
|
||||
|
||||
namespace Cysharp.Threading.TasksTests
|
||||
{
|
||||
public class AsyncReactivePropertyTest
|
||||
{
|
||||
private int _callCounter;
|
||||
|
||||
[UnityTest]
|
||||
public IEnumerator WaitCancelWait() => UniTask.ToCoroutine(async () =>
|
||||
{
|
||||
// Test case for https://github.com/Cysharp/UniTask/issues/444
|
||||
|
||||
var property = new AsyncReactiveProperty<int>(0);
|
||||
|
||||
var cts1 = new CancellationTokenSource();
|
||||
var cts2 = new CancellationTokenSource();
|
||||
WaitForProperty(property, cts1.Token);
|
||||
WaitForProperty(property, cts2.Token);
|
||||
|
||||
_callCounter = 0;
|
||||
property.Value = 1;
|
||||
_callCounter.Should().Be(2);
|
||||
|
||||
cts2.Cancel();
|
||||
cts2.Dispose();
|
||||
cts1.Cancel();
|
||||
cts1.Dispose();
|
||||
|
||||
var cts3 = new CancellationTokenSource();
|
||||
WaitForProperty(property, cts3.Token);
|
||||
|
||||
_callCounter = 0;
|
||||
property.Value = 2;
|
||||
_callCounter.Should().Be(1);
|
||||
|
||||
cts3.Cancel();
|
||||
cts3.Dispose();
|
||||
await UniTask.CompletedTask;
|
||||
});
|
||||
|
||||
private async void WaitForProperty(AsyncReactiveProperty<int> property, CancellationToken token)
|
||||
{
|
||||
while (true)
|
||||
{
|
||||
try
|
||||
{
|
||||
await property.WaitAsync(token);
|
||||
_callCounter++;
|
||||
}
|
||||
catch (OperationCanceledException)
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
|
@ -1,3 +0,0 @@
|
|||
fileFormatVersion: 2
|
||||
guid: 27665955eefb4448969b8cc4dd204600
|
||||
timeCreated: 1676129650
|
Loading…
Reference in New Issue