c# - 如何编写可以处理Task和ValueTask的方法？

假设你想写一个类似于下面的方法。它用一些简单的性能监视代码包装返回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时，是否可以避免这种代码重复。到目前为止，我还没有想出一个好的解决办法。有什么想法吗？

最佳答案

根据官方文件：Generalized async return types
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);
}