c# - 在 QueueBackgroundWorkItem 上实现 WhenAll

Question

我需要在 ASP.NET WebAPI 中生成一些长时间运行的任务，我选择使用 QueueBackgroundWorkItem 来实现。

我的问题是 -

我需要为这些任务实现一个 WhenAll 机制，这意味着 - 我需要知道一些任务何时完成，然后执行其他一些任务。

如何使用 QueueBackgroundWorkItem 来完成？我找不到这样做的方法，因为它返回 void，并接受 Action。

谢谢!

Answer 1

您可以使用 TaskCompletionSource创建这样的辅助方法:

public static Task RunViaQueueBackgroundWorkItem(Action<CancellationToken> action)
{
    TaskCompletionSource<int> tcs = new TaskCompletionSource<int>();
    HostingEnvironment.QueueBackgroundWorkItem((ct) =>
    {
        try
        {
            action(ct);

            if (ct.IsCancellationRequested)
                tcs.TrySetCanceled();
            else
                tcs.TrySetResult(0);
        }
        catch (Exception ex)
        {
            tcs.TrySetException(ex);
        }
    });

    return tcs.Task;
}

此方法将返回一个任务，该任务将在操作完成后完成。请注意，它处理在调用操作期间发生异常的情况。如果请求取消，它也会处理这种情况。

这是一个如何使用它的例子:

public async Task<string> Get(int id)
{
    var t1 = RunViaQueueBackgroundWorkItem(ct => DoSomeTask1(ct));
    var t2 = RunViaQueueBackgroundWorkItem(ct => DoSomeTask2(ct));

    await Task.WhenAll(t1, t2);

    //...
}

为了完整起见，这里是 QueueBackgroundWorkItem 的另一个重载的辅助方法:

public static Task RunViaQueueBackgroundWorkItem(Func<CancellationToken,Task> asyncAction)
{
    TaskCompletionSource<int> tcs = new TaskCompletionSource<int>();
    HostingEnvironment.QueueBackgroundWorkItem(async (ct) => 
    {
        try
        {
            await asyncAction(ct);

            if (ct.IsCancellationRequested)
                tcs.TrySetCanceled();
            else
                tcs.TrySetResult(0);
        }
        catch (Exception ex)
        {
            tcs.TrySetException(ex);
        }
    });

    return tcs.Task;
}