设置方法的最大运行时间

Question

如果我有一个方法想要执行一些（可能）长时间运行的函数，并且我想限制它的执行时间，我一直在使用这种模式（请原谅代码中的任何错误，键入手动，而不是在 IDE 中，这是对较大代码段的简化）。

public string GetHello()
{
    var task = Task.Run(() => 
    {
        // do something long running
        return "Hello";
    });

    bool success = task.Wait(TimeSpan.FromMilliseconds(1000));

    if (success)
    {
        return task.Result;
    }
    else
    {
        throw new TimeoutException("Timed out.");
    }
}

如果我想在异步容量中使用GetHello 方法，即public async Task<string> GetHello()，我将如何在希望保留类似模式的同时做到这一点？我有以下内容，但我收到了关于 This async method lacks 'await' operators and will run synchronously 的编译器警告。

public async Task<string> GetHello()
{
    var task = Task.Run(async () => 
    {
        // await something long running
        return "Hello";
    });

    bool success = task.Wait(TimeSpan.FromMilliseconds(1000));

    if (success)
    {
        return task.Result;
    }
    else
    {
        throw new TimeoutException("Timed out.");
    }
}

我只是不知道如何更改此设置或将 await 放在何处以使其按预期工作。

Answer 1

您可以结合使用 CancellationToken 和 awaitable Task.WhenAny 来实现所需的行为：

public async Task<string> GetHello()
{
    var cts = new CancellationTokenSource();
    var task = Task.Run(async () =>
    {
        // await something long running and pass/use cts.Token here too
        return "Hello";
    }, cts.Token);

    var delay = Task.Delay(1000, cts.Token);

    var finishedFirst = await Task.WhenAny(task, delay);
    cts.Cancel();
    if (finishedFirst == task)
    {
        return task.Result;
    }
    else
    {
        throw new TimeoutException("Timed out.");
    }
}