Example #1
0
 /// <summary>
 ///     Throws the specified <paramref name="exception"/> on the oldest wait matching the specified <paramref name="key"/>.
 /// </summary>
 /// <param name="key">The unique WaitKey for the wait.</param>
 /// <param name="exception">The Exception to throw.</param>
 public void Throw(WaitKey key, Exception exception)
 {
     if (Waits.TryGetValue(key, out var queue) && queue.TryDequeue(out var wait))
     {
         wait.TaskCompletionSource.SetException(exception);
     }
 }
Example #2
0
        /// <summary>
        ///     Adds a new wait for the specified <paramref name="key"/> and with the specified <paramref name="timeout"/>.
        /// </summary>
        /// <typeparam name="T">The wait result type.</typeparam>
        /// <param name="key">A unique WaitKey for the wait.</param>
        /// <param name="timeout">The wait timeout.</param>
        /// <param name="cancellationToken">The cancellation token for the wait.</param>
        /// <returns>A Task representing the wait.</returns>
        public Task <T> Wait <T>(WaitKey key, int?timeout = null, CancellationToken?cancellationToken = null)
        {
            timeout = timeout ?? DefaultTimeout;

            var wait = new PendingWait()
            {
                TaskCompletionSource = new TaskCompletionSource <T>(TaskCreationOptions.RunContinuationsAsynchronously),
                DateTime             = DateTime.UtcNow,
                TimeoutAfter         = (int)timeout,
                CancellationToken    = cancellationToken,
            };

            Waits.AddOrUpdate(key, new ConcurrentQueue <PendingWait>(new[] { wait }), (_, queue) =>
            {
                queue.Enqueue(wait);
                return(queue);
            });

            return(((TaskCompletionSource <T>)wait.TaskCompletionSource).Task);
        }
Example #3
0
 /// <summary>
 ///     Adds a new wait for the specified <paramref name="key"/> which does not time out.
 /// </summary>
 /// <typeparam name="T">The wait result type.</typeparam>
 /// <param name="key">A unique WaitKey for the wait.</param>
 /// <param name="cancellationToken">The cancellation token for the wait.</param>
 /// <returns>A Task representing the wait.</returns>
 public Task <T> WaitIndefinitely <T>(WaitKey key, CancellationToken?cancellationToken = null)
 {
     return(Wait <T>(key, MaxTimeoutValue, cancellationToken));
 }
Example #4
0
 /// <summary>
 ///     Adds a new wait for the specified <paramref name="key"/> which does not time out.
 /// </summary>
 /// <param name="key">A unique WaitKey for the wait.</param>
 /// <param name="cancellationToken">The cancellation token for the wait.</param>
 /// <returns>A Task representing the wait.</returns>
 public Task WaitIndefinitely(WaitKey key, CancellationToken?cancellationToken = null)
 {
     return(WaitIndefinitely <object>(key, cancellationToken));
 }
Example #5
0
 /// <summary>
 ///     Adds a new wait for the specified <paramref name="key"/> and with the specified <paramref name="timeout"/>.
 /// </summary>
 /// <param name="key">A unique WaitKey for the wait.</param>
 /// <param name="timeout">The wait timeout.</param>
 /// <param name="cancellationToken">The cancellation token for the wait.</param>
 /// <returns>A Task representing the wait.</returns>
 public Task Wait(WaitKey key, int?timeout = null, CancellationToken?cancellationToken = null)
 {
     return(Wait <object>(key, timeout, cancellationToken));
 }
Example #6
0
 /// <summary>
 ///     Completes the oldest wait matching the specified <paramref name="key"/>.
 /// </summary>
 /// <param name="key">The unique WaitKey for the wait.</param>
 public void Complete(WaitKey key)
 {
     Complete <object>(key, null);
 }