TimeHelperClass.cs 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Threading.Tasks;
  6. using System.Threading;
  7. namespace TimeOutHelper
  8. {
  9. public static class TimeHelperClass
  10. {
  11. /// <summary>
  12. /// 异步方法超时处理
  13. /// </summary>
  14. /// <typeparam name="T"></typeparam>
  15. /// <param name="function"></param>
  16. /// <param name="timeout"></param>
  17. /// <returns></returns>
  18. /// <exception cref="TimeoutException"></exception>
  19. public static async Task<T> ExecuteWithTimeoutAsync<T>(Func<CancellationToken, Task<T>> function, TimeSpan timeout)
  20. {
  21. using (var cts = new CancellationTokenSource())
  22. {
  23. var timeoutTask = Task.Delay(timeout, cts.Token);
  24. var functionTask = function(cts.Token);
  25. if (await Task.WhenAny(timeoutTask, functionTask) == timeoutTask)
  26. {
  27. cts.Cancel();
  28. throw new TimeoutException($"操作在 {timeout.TotalSeconds} 秒后超时");
  29. }
  30. return await functionTask;
  31. }
  32. }
  33. /// <summary>
  34. /// 同步方法超时处理
  35. /// </summary>
  36. /// <typeparam name="T"></typeparam>
  37. /// <param name="function"></param>
  38. /// <param name="timeout"></param>
  39. /// <returns></returns>
  40. public static async Task<T> ExecuteWithTimeout<T>(Func<T> function, TimeSpan timeout)
  41. {
  42. return await ExecuteWithTimeoutAsync<T>(ct => Task.Run(() => function(), ct), timeout);
  43. }
  44. }
  45. }