调用外部方法时的任务调度

Question

我有以下扩展，它允许我强制任务超时。

public static async Task<T> TimeoutAfter<T>(this Task<T> task, TimeSpan timeSpan, string message = null)
{
    if (task == await Task.WhenAny(task, Task.Delay(timeSpan)))
        return await task;
    else
    {
        throw new TimeoutException(message);
    }
}

我有以下任务：

private async Task<string> TestTask(PageData page)
{
    await Task.Delay(TimeSpan.FromSeconds(10));
    return "teststring";
}

当我使用以下命令调用任务时，我得到了预期的结果。

var teststring = await TestTask()
                .TimeoutAfter(TimeSpan.FromSeconds(5), "timeout message");

我遇到的问题是，当我调用外部库（在本例中为 c++ CLR 类库）中的方法时，永远不会到达 TimeoutAfter() 中的代码，并且任务永远不会超时。我可以通过使用 Task.Factory.StartNew()（如下）来解决这个问题，将回调传递给长时间运行的任务并在主任务开始执行之前创建我的 Task.Delay 任务，但这不能作为扩展.

private async Task<T> Timeout<T>(Func<Task<T>> taskFunc)
{
    var taskWait = Task.Delay(TimeSpan.FromSeconds(15));
    var task = Task.Factory.StartNew(() => taskFunc());

    if (task == await Task.WhenAny(task, taskWait))
        return await (await task);
    else
    {
        throw new TimeoutException("timeout");
    }
}

为什么 TimeoutAfter() 中的代码似乎只在某些内容位于它正在扩展的任务中时才执行？我可以向扩展程序添加什么功能来防止这种情况发生？

Answer 1

我认为你的主要问题是当你尝试使用扩展方法时你甚至没有开始你的任务。

Task 是完全异步的，must be started 可以访问其代码。开始任务的一种方法是使用 StartNew，就像您在工厂中尝试的那样。

但您也可以只调用Start 方法。唯一需要注意的是，如果 start 已经开始或已经完成，调用 Start 时会出现异常。

所以，我认为您的解决方案可以是：

public static async Task<T> TimeoutAfter<T>(this Task<T> task, TimeSpan timeSpan, string message = null)
{
    if(task.Status != TaskStatus.Running && task.Status != TaskStatus.RanToCompletion){
       task.Start();
    }
    if (task == await Task.WhenAny(task, Task.Delay(timeSpan)))
       return await task;
    else
    {
        throw new TimeoutException(message);
    }
}