在 UI 线程上创建和启动任务

Question

当在工作线程上调用的方法需要在 UI 线程上运行代码并等待其完成后再执行其他操作时，可以这样完成：

    public int RunOnUi(Func<int> f)
    {
        int res = Application.Current.Dispatcher.Invoke(f);

        return res;
    }

但是如果我想用任务来做呢？ RunOnUi 方法有没有办法创建在 UI 上启动的任务并将其返回，以便调用者（在工作线程上运行）可以等待它？符合以下签名的内容：public Task<int> StartOnUi(Func<int> f)?

一种方法如下：

public Task<int> RunOnUi(Func<int> f)
{
    var task = new Task<int>(f);
    task.Start(_scheduler);

    return task;
}

在这里，假设 _schduler 拥有 ui TaskScheduler。但我不太喜欢创建“冷”任务并使用 start 方法运行它们。这是“推荐”的方式还是有更优雅的方式？

Answer 1

只需使用InvokeAsync 而不是Invoke，然后在函数返回的DispatcherOperation<int> 中返回Task<int>。

//Coding conventions say async functions should end with the word Async.
public Task<int> RunOnUiAsync(Func<int> f)
{
    var dispatcherOperation = Application.Current.Dispatcher.InvokeAsync(f);
    return dispatcherOperation.Task;
}

如果您无法访问 .NET 4.5，情况会稍微复杂一些。您需要使用BeginInvoke 和TaskCompletionSource 来包装BeginInvoke 返回的DispaterOperation

    public Task<int> RunOnUi(Func<int> f)
    {
        var operation = Application.Current.Dispatcher.BeginInvoke(f);
        var tcs = new TaskCompletionSource<int>();
        operation.Aborted += (sender, args) => tcs.TrySetException(new SomeExecptionHere());
        operation.Completed += (sender, args) => tcs.TrySetResult((int)operation.Result);

        //The operation may have already finished and this check accounts for 
        //the race condition where neither of the events will ever be called
        //because the events where raised before you subscribed.
        var status = operation.Status;
        if (status == DispatcherOperationStatus.Completed)
        {
            tcs.TrySetResult((int)operation.Result);
        }
        else if (status == DispatcherOperationStatus.Aborted)
        {
            tcs.TrySetException(new SomeExecptionHere());
        }

        return tcs.Task;
    }