优雅的实现超时连接Task

TimeoutHelper帮助类

以下是一个封装方法的实现,它接受一个 Action 或 Func 委托和一个超时时间(以毫秒为单位)。如果委托在指定的时间内执行完成,返回 true;否则返回 false:

using System;
using System.Threading;
using System.Threading.Tasks;

public static class TimeoutHelper
{
    /// <summary>
    /// 执行指定的委托,并在给定的超时时间内返回结果。
    /// </summary>
    /// <param name="action">需要执行的委托。</param>
    /// <param name="timeoutMilliseconds">超时时间(毫秒)。</param>
    /// <returns>在超时内完成返回 true,否则返回 false。</returns>
    public static bool ExecuteWithTimeout(Action action, int timeoutMilliseconds)
    {
        if (action == null) throw new ArgumentNullException(nameof(action));
        if (timeoutMilliseconds <= 0) throw new ArgumentOutOfRangeException(nameof(timeoutMilliseconds));

        try
        {
            // 创建一个任务来执行 Action
            var task = Task.Run(action);
            // 等待任务完成,或超时
            return task.Wait(timeoutMilliseconds);
        }
        catch
        {
            // 捕获异常并返回 false
            return false;
        }
    }

    /// <summary>
    /// 执行指定的 Func,并在给定的超时时间内返回结果。
    /// </summary>
    /// <typeparam name="T">返回值类型。</typeparam>
    /// <param name="func">需要执行的 Func。</param>
    /// <param name="timeoutMilliseconds">超时时间(毫秒)。</param>
    /// <param name="result">输出结果,如果超时则为默认值。</param>
    /// <returns>在超时内完成返回 true,否则返回 false。</returns>
    public static bool ExecuteWithTimeout<T>(Func<T> func, int timeoutMilliseconds, out T result)
    {
        if (func == null) throw new ArgumentNullException(nameof(func));
        if (timeoutMilliseconds <= 0) throw new ArgumentOutOfRangeException(nameof(timeoutMilliseconds));

        try
        {
            // 创建一个任务来执行 Func
            var task = Task.Run(func);
            // 等待任务完成,或超时
            if (task.Wait(timeoutMilliseconds))
            {
                result = task.Result;
                return true;
            }
        }
        catch
        {
            // 捕获异常并设置结果为默认值
        }

        result = default!;
        return false;
    }
}

调用示例1:

bool success = TimeoutHelper.ExecuteWithTimeout(() =>
{
    // 模拟耗时操作
    Thread.Sleep(1000);
}, 500); // 超时时间为 500 毫秒

Console.WriteLine(success ? "操作成功完成" : "操作超时");

调用示例2:

bool success = TimeoutHelper.ExecuteWithTimeout(() =>
{
    // 模拟耗时操作
    Thread.Sleep(1000);
    return "结果数据";
}, 500, out string result);

Console.WriteLine(success ? $"操作成功,结果: {result}" : "操作超时");

特点:

  • 线程安全:使用 Task.Run 执行任务,独立于主线程。
  • 支持返回值:通过泛型 Func 方法支持返回结果。
  • 超时控制:指定时间内未完成会直接返回,确保程序不会因为某些操作卡住。
  • 错误捕获:安全处理委托中的异常,避免程序崩溃。
posted @   Smileher  阅读(19)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 【设计模式】告别冗长if-else语句:使用策略模式优化代码结构
· 提示词工程——AI应用必不可少的技术
· 字符编码:从基础到乱码解决
· 地球OL攻略 —— 某应届生求职总结
点击右上角即可分享
微信分享提示