| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Text;
- using System.Threading.Tasks;
- using System.Threading;
- namespace TimeOutHelper
- {
- public static class TimeHelperClass
- {
- /// <summary>
- /// 异步方法超时处理
- /// </summary>
- /// <typeparam name="T"></typeparam>
- /// <param name="function"></param>
- /// <param name="timeout"></param>
- /// <returns></returns>
- /// <exception cref="TimeoutException"></exception>
- public static async Task<T> ExecuteWithTimeoutAsync<T>(Func<CancellationToken, Task<T>> function, TimeSpan timeout)
- {
- using (var cts = new CancellationTokenSource())
- {
- var timeoutTask = Task.Delay(timeout, cts.Token);
- var functionTask = function(cts.Token);
- if (await Task.WhenAny(timeoutTask, functionTask) == timeoutTask)
- {
- cts.Cancel();
- throw new TimeoutException($"操作在 {timeout.TotalSeconds} 秒后超时");
- }
- return await functionTask;
- }
- }
- /// <summary>
- /// 同步方法超时处理
- /// </summary>
- /// <typeparam name="T"></typeparam>
- /// <param name="function"></param>
- /// <param name="timeout"></param>
- /// <returns></returns>
- public static async Task<T> ExecuteWithTimeout<T>(Func<T> function, TimeSpan timeout)
- {
- return await ExecuteWithTimeoutAsync<T>(ct => Task.Run(() => function(), ct), timeout);
- }
- }
- }
|