在WCF服務中使用異步/等待時第一次等待時，OperationContext.Current爲空

Question

我使用.NET 4.5中的異步/等待模式來實現WCF中的某些服務方法。 示例服務：

合同：

[ServiceContract(Namespace = "http://async.test/")] 
public interface IAsyncTest 
{ 
    Task DoSomethingAsync(); 
} 

實現：

MyAsyncService : IAsyncTest 
{ 
    public async Task DoSomethingAsync() 
    { 
     var context = OperationContext.Current; // context is present 

     await Task.Delay(10); 

     context = OperationContext.Current; // context is null 
    } 
} 

我遇到的問題是，第一awaitOperationContext.Current回報null後，我無法訪問OperationContext.Current.IncomingMessageHeaders。

在這個簡單的例子中，這不是一個問題，因爲我可以在await之前捕獲上下文。但是在現實世界中，OperationContext.Current正從調用堆棧的內部進行訪問，我真的不想爲了進一步傳遞上下文而更改大量代碼。

有沒有辦法在await點之後獲得操作上下文，而無需手動將其傳遞到堆棧上？

Answer 1

我認爲你最好的選擇是實際捕獲它並手動傳遞它。您可能會發現這會提高代碼的可測試性。

這就是說，有幾個其他選項：

1. 將它添加到LogicalCallContext。
2. 安裝您自己的SynchronizationContext，它會在設置OperationContext.Current時設置Post;這是ASP.NET如何保留其HttpContext.Current。
3. 安裝您自己的TaskScheduler其中OperationContext.Current。

您可能還想在Microsoft Connect上提出此問題。

Answer 2

不幸的是，這不起作用，我們將看到在未來的版本中修復。

與此同時，有重新應用上下文當前線程，這樣你就不必繞過物體的方式：

public async Task<double> Add(double n1, double n2) 
    { 

     OperationContext ctx = OperationContext.Current; 

     await Task.Delay(100); 

     using (new OperationContextScope(ctx)) 
     { 
      DoSomethingElse(); 
     } 
     return n1 + n2; 
    } 

在上面的例子中，DoSomethingElse（ ）方法將按預期方式訪問OperationContext.Current。

Answer 3

幸運的是，我們的實際服務實現通過Unity IoC容器實例化。這使我們能夠創建一個IWcfOperationContext，它被配置爲具有PerResolveLifetimeManager，這僅僅意味着對於RealService的每個實例，將只有一個WcfOperationContext實例。
在WcfOperationContext的構造函數中，我們捕獲OperationContext.Current，然後所有需要它的地方從IWcfOperationContext得到它。這實際上是斯蒂芬克利裏在他的回答中提出的。

Answer 4

這裏是一個樣本SynchronizationContext實現：

public class OperationContextSynchronizationContext : SynchronizationContext 
{ 
    private readonly OperationContext context; 

    public OperationContextSynchronizationContext(IClientChannel channel) : this(new OperationContext(channel)) { } 

    public OperationContextSynchronizationContext(OperationContext context) 
    { 
     OperationContext.Current = context; 
     this.context = context; 
    } 

    public override void Post(SendOrPostCallback d, object state) 
    { 
     OperationContext.Current = context; 
     d(state); 
    } 
} 

與用法：

var currentSynchronizationContext = SynchronizationContext.Current; 
try 
{ 
    SynchronizationContext.SetSynchronizationContext(new OperationContextSynchronizationContext(client.InnerChannel)); 
    var response = await client.RequestAsync(); 
    // safe to use OperationContext.Current here 
} 
finally 
{ 
    SynchronizationContext.SetSynchronizationContext(currentSynchronizationContext); 
} 

Answer 5

擴展在克利先生的＃1選項，下面的代碼可以被放置在WCF服務的構造函數在邏輯調用上下文中存儲和檢索OperationContext：

if (CallContext.LogicalGetData("WcfOperationContext") == null) 
{ 
    CallContext.LogicalSetData("WcfOperationContext", OperationContext.Current); 
} 
else if (OperationContext.Current == null) 
{ 
    OperationContext.Current = (OperationContext)CallContext.LogicalGetData("WcfOperationContext"); 
} 

就這樣，在任何地方您有一個空的上下文的問題，你可以寫類似以下內容：

var cachedOperationContext = CallContext.LogicalGetData("WcfOperationContext") as OperationContext; 
var user = cachedOperationContext != null ? cachedOperationContext.ServiceSecurityContext.WindowsIdentity.Name : "No User Info Available"; 

聲明：本歲的代碼，我不記得我所需要的else if的原因構造函數，但它與async有關，我知道它是我需要的。

Answer 6

更新：正如在下面的評論中指出的，這個解決方案不是線程安全的，所以我想上面討論的解決方案仍然是最好的方法。

我通過將HttpContext註冊到我的DI容器（Application_BeginRequest）中解決問題，並在需要時解決它。

註冊：

this.UnityContainer.RegisterInstance<HttpContextBase>(new HttpContextWrapper(HttpContext.Current)); 

解析：

var context = Dependencies.ResolveInstance<HttpContextBase>(); 

Answer 7

這似乎是固定在.net 4.6.2。請參閱the announcement