EntLib TransientFaultHandling RetryPolicy.ExecuteAsync同步上下文

时间:2014-10-22 11:18:21

标签: c# asynchronous task-parallel-library async-await wcf-data-services

我在Windows 8商店应用程序中使用企业库瞬态故障处理应用程序块,其中包含用于ODATA的WCF数据服务客户端。我想使用重试逻辑来调用ODATA服务时发生的瞬态错误。我构建了一个自定义瞬态错误检测策略。

我还为DataServiceCollection构建了一个LoadTaskAsync扩展方法,因为LoadAsync方法不返回Task(而是DataServiceCollection引发LoadCompleted事件)。

所以我可以将数据加载到DataServiceCollection中,如下所示:

var query = this.DataContext.Products.Where(item => item.Modified >= anchor);
var products = new DataServiceCollection<Product>(this.DataContext);
await this.retryPolicy.ExecuteAsync(() => products.LoadTaskAsync(query));

现在,企业库瞬态故障处理应用程序块的文档指出了

  

传递给ExecuteAsync方法的taskFunc参数不一定在最初调用ExecuteAsync时使用的同一同步上下文中调用;因此,如果您需要从UI线程中启动任务,请务必在委托中明确地安排它。

我需要在UI线程上调用LoadTaskAsync方法,因为加载操作可能会更新已由数据上下文跟踪的产品以及绑定到UI的数据。

问题是怎么回事?最好不要修改LoadTaskAsync扩展方法(如果不是我要改变的代码怎么办)。我正在考虑为RetryPolicy创建一个扩展方法,该方法调用ExecuteAsync方法,同时确保在UI线程上调用taskFunc。

最简单的方法可能是修改LoadTaskAsync扩展方法以传入TaskCreationOptions.AttachedToParent,这样我就可以为RetryPolicy创建一个扩展方法,如下所示:

    public static Task<TResult> ExecuteCurrentSynchronizationContextAsync<TResult>(
        this RetryPolicy retryPolicy, 
        Func<TaskCreationOptions, Task<TResult>> taskFunc)
    {
        var scheduler = TaskScheduler.FromCurrentSynchronizationContext();

        return
            retryPolicy.ExecuteAsync(
                () =>
                Task.Factory.StartNew(
                    () => taskFunc(TaskCreationOptions.AttachedToParent), 
                    CancellationToken.None, 
                    TaskCreationOptions.None, 
                    scheduler).Unwrap());
    }

请注意,taskFunc现在必须是Func&lt; TaskCreationOptions,Task&lt; TResult&gt;&gt;。

我会这样称呼如下:

        await
            this.retryPolicy.ExecuteCurrentSynchronizationContextAsync(
                creationOptions => products.LoadTaskAsync(query, creationOptions));

由于我不想更改LoadTaskAsync扩展方法,我如何更改此RetryPolicy ExecuteCurrentSynchronizationContextAsync扩展方法,以便taskFunc可以是Func&lt; Task&lt; TResult&gt;&gt;再次确保在UI线程上调用taskFunc?

1 个答案:

答案 0 :(得分:1)

我不建议将AttachedToParent用于异步任务。事实上,大多数承诺式任务都会指定DenyChildAttach which prevents AttachedToParent from working

相反,您只需捕获同步上下文本身并使用它。有一个皱纹:Windows应用商店应用don't allow synchronous invocation on the synchronization context,因此您需要使用CoreDispatcher.RunAsync代替SynchronizationContext,或构建您自己的async感知扩展方法为SynchronizationContext。在这两个中,我更喜欢使用SynchronizationContext方法;在这种情况下,它需要更多代码,但这意味着您不必将代码(可能是服务层代码)绑定到此特定UI框架。

所以,首先我们在RunAsync上定义一个SynchronizationContext,它将(异步)在指定的同步上下文中执行异步代码:

public static Task<TResult> RunAsync<TResult>(this SynchronizationContext context, Func<Task<TResult>> func)
{
    var tcs = new TaskCompletionSource<TResult>();
    context.Post(async _ =>
    {
        try
        {
            tcs.TrySetResult(await func());
        }
        catch (OperationCanceledException)
        {
            tcs.TrySetCanceled();
        }
        catch (Exception ex)
        {
            tcs.TrySetException(ex);
        }
    }, null);
    return tcs.Task;
}

然后我们可以捕获并使用SynchronizationContext

public static Task<TResult> ExecuteOnCurrentSynchronizationContextAsync<TResult>(
    this RetryPolicy retryPolicy, 
    Func<Task<TResult>> taskFunc)
{
    var context = SynchronizationContext.Current ?? new SynchronizationContext();

    return retryPolicy.ExecuteAsync(() => context.RunAsync(taskFunc));
}