如何编写一个可以处理Task和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从任何现有异步方法的返回值构造 a：

这意味着您可以编写一个重载来包装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);
}