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);
        }
    }
}