Added custom async wait event implementation as the previously used method seemed to not work 100% of the time

2026-08-18 03:43:00 +00:00 · 2021-10-05 14:31:27 +02:00
parent 4ef20a9637
commit f3102a4dad
7 changed files with 282 additions and 229 deletions
@@ -0,0 +1,102 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace CryptoExchange.Net.Objects
+{
+    /// <summary>
+    /// Async auto reset based on Stephen Toub`s implementation
+    /// https://devblogs.microsoft.com/pfxteam/building-async-coordination-primitives-part-2-asyncautoresetevent/
+    /// </summary>
+    public class AsyncResetEvent : IDisposable
+    {
+        private readonly static Task<bool> _completed = Task.FromResult(true);
+        private readonly Queue<TaskCompletionSource<bool>> _waits = new Queue<TaskCompletionSource<bool>>();
+        private bool _signaled;
+        private bool _reset;
+
+        /// <summary>
+        /// New AsyncResetEvent
+        /// </summary>
+        /// <param name="initialState"></param>
+        /// <param name="reset"></param>
+        public AsyncResetEvent(bool initialState = false, bool reset = true)
+        {
+            _signaled = initialState;
+            _reset = reset;
+        }
+
+        /// <summary>
+        /// Wait for the AutoResetEvent to be set
+        /// </summary>
+        /// <returns></returns>
+        public Task<bool> WaitAsync(TimeSpan? timeout = null)
+        {
+            lock (_waits)
+            {
+                if (_signaled)
+                {
+                    if(_reset)
+                        _signaled = false;
+                    return _completed;
+                }
+                else
+                {
+                    var tcs = new TaskCompletionSource<bool>();
+                    if(timeout != null)
+                    {
+                        var cancellationSource = new CancellationTokenSource(timeout.Value);
+                        var registration = cancellationSource.Token.Register(() =>
+                        {
+                            tcs.TrySetResult(false);
+                        }, useSynchronizationContext: false);
+                    }
+
+                    _waits.Enqueue(tcs);
+                    return tcs.Task;
+                }
+            }
+        }
+
+        /// <summary>
+        /// Signal a waiter
+        /// </summary>
+        public void Set()
+        {
+            lock (_waits)
+            {
+                if (!_reset)
+                {
+                    // Act as ManualResetEvent. Once set keep it signaled and signal everyone who is waiting
+                    _signaled = true;
+                    while (_waits.Count > 0)
+                    {
+                        var toRelease = _waits.Dequeue();
+                        toRelease.TrySetResult(true);
+                    }
+                }
+                else
+                {
+                    // Act as AutoResetEvent. When set signal 1 waiter
+                    if (_waits.Count > 0)
+                    {
+                        var toRelease = _waits.Dequeue();
+                        toRelease.TrySetResult(true);
+                    }
+                    else if (!_signaled)
+                        _signaled = true;
+                }
+            }
+        }
+
+        /// <summary>
+        /// Dispose
+        /// </summary>
+        public void Dispose()
+        {
+            _waits.Clear();
+        }
+    }
+}