如何编写能够处理任务和ValueTask的方法？

Question

假设您想要编写一个类似于下面的方法。它用简单的性能监视代码封装一个返回ValueTask<T>的函数：

static async Task Measure<T>(Func<ValueTask<T>> body)
{
    Console.WriteLine($"Starting perf test");
    var sw = Stopwatch.StartNew();
    await body();
    sw.Stop();
    Console.WriteLine(sw.Elapsed);
}

我的问题是:是否有一种方法只编写一次此函数，以便它能够接收Func<ValueTask<T>> 和 Func<Task<T>>

当然，您可以简单地复制代码并只更改参数的类型。

static async Task Measure<T>(Func<Task<T>> body) { ... }

执行将是完全相同的。我在问自己，在必须处理ValueTask和Task时，是否有可能避免这种代码重复。到目前为止，我还没有想出一个好的解决办法。有什么想法吗？

Answer 1

根据官方文件：广义异步返回类型

ValueTask结构有一个带有Task参数的构造函数，因此您可以从任何现有异步方法的返回值构造ValueTask：

这意味着您可以编写一个重载，它将包装body，并且只调用一个将完成工作的方法。

static Task Measure<T>(Func<Task<T>> body)
{
    var wrapped = () => new ValueTask<T>( body() );
    return Measure( wrapped );
}

static async Task Measure<T>(Func<ValueTask<T>> body)
{
    Console.WriteLine($"Starting perf test");
    var sw = Stopwatch.StartNew();
    await body();
    sw.Stop();
    Console.WriteLine(sw.Elapsed);
}