TaskWaitingExtension.cs 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Threading;
  6. using System.Threading.Tasks;
  7. namespace CSharpUtil.Extensions.Tasks
  8. {
  9. public static class TaskWaitingExtensions
  10. {
  11. public static async Task<TResult> WaitAsync<TResult>(this Task<TResult> task, TimeSpan timeout)
  12. {
  13. using (var timeoutCancellationTokenSource = new CancellationTokenSource())
  14. {
  15. var delayTask = Task.Delay(timeout, timeoutCancellationTokenSource.Token);
  16. if (await Task.WhenAny(task, delayTask) == task)
  17. {
  18. timeoutCancellationTokenSource.Cancel();
  19. return await task;
  20. }
  21. throw new TimeoutException("The operation has timed out.");
  22. }
  23. }
  24. public static async Task WaitAsync(this Task task, TimeSpan timeout)
  25. {
  26. using (var timeoutCancellationTokenSource = new CancellationTokenSource())
  27. {
  28. var delayTask = Task.Delay(timeout, timeoutCancellationTokenSource.Token);
  29. if (await Task.WhenAny(task, delayTask) == task)
  30. {
  31. timeoutCancellationTokenSource.Cancel();
  32. return;
  33. }
  34. throw new TimeoutException("The operation has timed out.");
  35. }
  36. }
  37. }
  38. }