如何编写可以处理Task和ValueTask的方法?

时间:2017-05-10 10:52:15

标签: c# c#-7.0

想象一下,您想要编写类似于以下方法的方法。它包含一个返回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) { ... }

实施完全相同。我问自己,在处理ValueTaskTask时是否可以避免这种代码重复。到目前为止,我无法想出一个好的解决方案。有什么想法吗?

1 个答案:

答案 0 :(得分:4)

根据官方文件: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);
}