带有额外异常处理的异步 Task<T> 超时

Question

在我的项目中，我从动态链接库中引用类型和接口。 使用这个特定库时，我要做的第一件事是创建EA.Repository 的实例，该实例在库中定义，用作进一步使用的入口点。

实例化EA.Repository repository = new EA.Repository() 在后台执行一些复杂的操作，我发现自己面临三种可能的结果：

1. 实例化需要一些时间，但最终成功完成
2. 抛出异常（立即或一段时间后）
3. 实例化永远阻塞（在这种情况下，我想取消并通知用户）

我能够使用Task 提出一种异步方法：

public static void Connect()
{
    // Do the lengthy instantiation asynchronously
    Task<EA.Repository> task = Task.Run(() => { return new EA.Repository(); });

    bool isCompletedInTime;

    try
    {
        // Timeout after 5.0 seconds
        isCompletedInTime = task.Wait(5000);
    }
    catch (Exception)
    {
        // If the instantiation fails (in time), throw a custom exception
        throw new ConnectionException();
    }

    if (isCompletedInTime)
    {
        // If the instantiation finishes in time, store the object for later
        EapManager.Repository = task.Result;
    }
    else
    {       
        // If the instantiation did not finish in time, throw a custom exception
        throw new TimeoutException();
    }
}

（我知道，您可能已经在这里发现了很多问题。请耐心等待我......我们将不胜感激！）

到目前为止，这种方法有效 - 我可以模拟“异常”和“超时”两种情况，并获得所需的行为。

但是，我已经确定了另一种极端情况：假设实例化任务花费的时间足够长，以至于超时到期，然后引发异常。在这种情况下，我有时会收到一个AggregateException 说该任务尚未被观察到。

我正在努力寻找可行的解决方案。当超时到期时我无法真正取消任务，因为阻塞实例化显然阻止了我使用CancellationToken 方法。

我唯一能想到的就是在抛出我的自定义 TimeoutException 之前开始异步观察任务（即启动另一个任务）：

Task observerTask = Task.Run(() => {
    try { task.Wait(); }
    catch (Exception) { }
});

throw new TimeoutException();

当然，如果实例化真的永远阻塞，我已经有第一个任务永远无法完成。有了观察者任务，现在我什至有两个！

我对整个方法很不安全，所以欢迎任何建议！

非常感谢您！

Answer 1

我不确定我是否完全理解你想要达到的目标，但是如果你做这样的事情会怎样 -

public static void Connect()
{
    Task<EA.Repository> _realWork = Task.Run(() => { return new EA.Repository(); });
    Task _timeoutTask = Task.Delay(5000);
    Task.WaitAny(new Task[]{_realWork, timeoutTask});
    if (_timeoutTask.Completed)
    {
        // timed out
    }
    else
    {
        // all good, access _realWork.Result
    }
}

或者你甚至可以缩短一点 -

public static void Connect()
{
    Task<EA.Repository> _realWork = Task.Run(() => { return new EA.Repository(); });
    var completedTaskIndex = Task.WaitAny(new Task[]{_realWork}, 5000);
    if (completedTaskIndex == -1)
    {
        // timed out
    }
    else
    {
        // all good, access _realWork.Result
    }
}

您也可以随时 call Task.Run with a CancellationToken 超时，但这会引发异常 - 上述解决方案可让您控制行为而不会引发异常（即使您始终可以 try/catch）

Answer 2

这是一个扩展方法，您可以使用它来显式观察在未观察到时可能失败的任务：

public static Task<T> AsObserved<T>(this Task<T> task)
{
    task.ContinueWith(t => t.Exception);
    return task;
}

使用示例：

var task = Task.Run(() => new EA.Repository()).AsObserved();