98 lines
3.2 KiB
C#
98 lines
3.2 KiB
C#
//-----------------------------------------------------------------------------
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//-----------------------------------------------------------------------------
|
|
|
|
namespace System.ServiceModel
|
|
{
|
|
using System;
|
|
using System.ServiceModel.Channels;
|
|
using System.Threading;
|
|
|
|
public sealed class OperationContextScope : IDisposable
|
|
{
|
|
[ThreadStatic]
|
|
static OperationContextScope legacyCurrentScope;
|
|
|
|
static AsyncLocal<OperationContextScope> currentScope = new AsyncLocal<OperationContextScope>();
|
|
|
|
OperationContext currentContext;
|
|
bool disposed;
|
|
readonly OperationContext originalContext = OperationContext.Current;
|
|
readonly OperationContextScope originalScope = OperationContextScope.CurrentScope;
|
|
readonly Thread thread = Thread.CurrentThread;
|
|
|
|
public OperationContextScope(IContextChannel channel)
|
|
{
|
|
this.PushContext(new OperationContext(channel));
|
|
}
|
|
|
|
public OperationContextScope(OperationContext context)
|
|
{
|
|
this.PushContext(context);
|
|
}
|
|
|
|
private static OperationContextScope CurrentScope
|
|
{
|
|
get
|
|
{
|
|
return ServiceModelAppSettings.DisableOperationContextAsyncFlow ? legacyCurrentScope : currentScope.Value;
|
|
}
|
|
|
|
set
|
|
{
|
|
if (ServiceModelAppSettings.DisableOperationContextAsyncFlow)
|
|
{
|
|
legacyCurrentScope = value;
|
|
}
|
|
else
|
|
{
|
|
currentScope.Value = value;
|
|
}
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
if (!this.disposed)
|
|
{
|
|
this.disposed = true;
|
|
this.PopContext();
|
|
}
|
|
}
|
|
|
|
void PushContext(OperationContext context)
|
|
{
|
|
bool isAsyncFlowEnabled = OperationContext.ShouldUseAsyncLocalContext;
|
|
|
|
this.currentContext = context;
|
|
|
|
if (isAsyncFlowEnabled)
|
|
{
|
|
OperationContext.EnableAsyncFlow(this.currentContext);
|
|
}
|
|
|
|
CurrentScope = this;
|
|
OperationContext.Current = this.currentContext;
|
|
}
|
|
|
|
void PopContext()
|
|
{
|
|
if (ServiceModelAppSettings.DisableOperationContextAsyncFlow && this.thread != Thread.CurrentThread)
|
|
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(new InvalidOperationException(SR.GetString(SR.SFxInvalidContextScopeThread0)));
|
|
|
|
if (CurrentScope != this)
|
|
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(new InvalidOperationException(SR.GetString(SR.SFxInterleavedContextScopes0)));
|
|
|
|
if (OperationContext.Current != this.currentContext)
|
|
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(new InvalidOperationException(SR.GetString(SR.SFxContextModifiedInsideScope0)));
|
|
|
|
CurrentScope = this.originalScope;
|
|
OperationContext.Current = this.originalContext;
|
|
|
|
if (this.currentContext != null)
|
|
this.currentContext.SetClientReply(null, false);
|
|
}
|
|
}
|
|
}
|
|
|