flywithu/CryptoExchange.Net
1
0
Fork 0
mirror of https://github.com/JKorf/CryptoExchange.Net synced 2026-02-16 14:13:46 +00:00
Code Issues Releases Wiki Activity

Added timeout/ct task cleanup AsyncResetEvent

Browse Source
This commit is contained in:
Jkorf 2026-02-06 11:57:56 +01:00
parent 40d480e1fc
commit 4a79ce22ec
1 changed files with 40 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
CryptoExchange.Net/Objects/AsyncResetEvent.cs
View File

@ -53,41 +53,57 @@ namespace CryptoExchange.Net.Objects
_waiters.Enqueue(tcs);
}
if (timeout.HasValue || ct.CanBeCanceled)
CancellationTokenSource? delayCts = null;
try
{
// Wait for either timeout, cancellation token or set result
var delayTask = Task.Delay(timeout ?? Timeout.InfiniteTimeSpan, ct);
var completedTask = await Task.WhenAny(tcs.Task, delayTask).ConfigureAwait(false);
if (completedTask != tcs.Task)
if (timeout.HasValue || ct.CanBeCanceled)
{
// This was a timeout or cancellation, need to remove tcs from waiters
// if the tcs was set instead it will be removed in the Set method
delayCts = CancellationTokenSource.CreateLinkedTokenSource(ct);
if (tcs.TrySetResult(false))
var delayTask = Task.Delay(
timeout ?? Timeout.InfiniteTimeSpan,
delayCts.Token);
var completedTask =
await Task.WhenAny(tcs.Task, delayTask)
.ConfigureAwait(false);
if (completedTask != tcs.Task)
{
lock (_waitersLock)
// This was a timeout or cancellation, need to remove tcs from waiters
// if the tcs was set instead it will be removed in the Set method
if (tcs.TrySetResult(false))
{
// Dequeue and put in the back of the queue again except for the one we need to remove
int count = _waiters.Count;
for (int i = 0; i < count; i++)
lock (_waitersLock)
{
var w = _waiters.Dequeue();
if (w != tcs)
_waiters.Enqueue(w);
// Dequeue and put in the back of the queue again except for the one we need to remove
int count = _waiters.Count;
for (int i = 0; i < count; i++)
{
var w = _waiters.Dequeue();
if (w != tcs)
_waiters.Enqueue(w);
}
}
}
return false;
}
return false;
}
}
else
{
await tcs.Task.ConfigureAwait(false);
}
else
{
await tcs.Task.ConfigureAwait(false);
}
return true;
return true;
}
finally
{
// Actively stop the delay if tcs.Task won
delayCts?.Cancel();
delayCts?.Dispose();
}
}
/// <summary>