public AmbientContextSuppressor()
{
_savedScope = DbContextScope.GetAmbientScope();
// We're hiding the ambient scope but not removing its instance
// altogether. This is to be tolerant to some programming errors.
//
// Suppose we removed the ambient scope instance here. If someone
// was to start a parallel task without suppressing
// the ambient context and then tried to suppress the ambient
// context within the parallel task while the original flow
// of execution was still ongoing (a strange thing to do, I know,
// but I'm sure this is going to happen), we would end up
// removing the ambient context instance of the original flow
// of execution from within the parallel flow of execution!
//
// As a result, any code in the original flow of execution
// that would attempt to access the ambient scope would end up
// with a null value since we removed the instance.
//
// It would be a fairly nasty bug to track down. So don't let
// that happen. Hiding the ambient scope (i.e. clearing the CallContext
// in our execution flow but leaving the ambient scope instance untouched)
// is safe.
DbContextScope.HideAmbientScope();
}
public DbContextScope(DbContextScopeOption joiningOption, bool readOnly, IsolationLevel? isolationLevel, IDbContextFactory dbContextFactory = null)
{
if (isolationLevel.HasValue && joiningOption == DbContextScopeOption.JoinExisting)
throw new ArgumentException("Cannot join an ambient DbContextScope when an explicit database transaction is required. When requiring explicit database transactions to be used (i.e. when the 'isolationLevel' parameter is set), you must not also ask to join the ambient context (i.e. the 'joinAmbient' parameter must be set to false).");
_disposed = false;
_completed = false;
_readOnly = readOnly;
_parentScope = GetAmbientScope();
if (_parentScope != null && joiningOption == DbContextScopeOption.JoinExisting)
{
if (_parentScope._readOnly && !this._readOnly)
{
throw new InvalidOperationException("Cannot nest a read/write DbContextScope within a read-only DbContextScope.");
}
_nested = true;
_dbContexts = _parentScope._dbContexts;
}
else
{
_nested = false;
_dbContexts = new DbContextCollection(readOnly, isolationLevel, dbContextFactory);
}
SetAmbientScope(this);
}
public DbContextScope(DbContextScopeOption joiningOption, bool readOnly, IsolationLevel?isolationLevel, IDbContextFactory dbContextFactory = null,
string nameOrConnectionString = null)
{
if (isolationLevel.HasValue && joiningOption == DbContextScopeOption.JoinExisting)
{
throw new ArgumentException("Cannot join an ambient DbContextScope when an explicit database transaction is required. When requiring explicit database transactions to be used (i.e. when the 'isolationLevel' parameter is set), you must not also ask to join the ambient context (i.e. the 'joinAmbient' parameter must be set to false).");
}
_disposed = false;
_completed = false;
_readOnly = readOnly;
_parentScope = GetAmbientScope();
if (_parentScope != null && joiningOption == DbContextScopeOption.JoinExisting)
{
if (_parentScope._readOnly && !this._readOnly)
{
throw new InvalidOperationException("Cannot nest a read/write DbContextScope within a read-only DbContextScope.");
}
_nested = true;
_dbContexts = _parentScope._dbContexts;
}
else
{
_nested = false;
_dbContexts = new DbContextCollection(readOnly, isolationLevel, dbContextFactory, nameOrConnectionString);
}
SetAmbientScope(this);
}
public void Dispose()
{
if (_disposed)
return;
if (_savedScope != null)
{
DbContextScope.SetAmbientScope(_savedScope);
_savedScope = null;
}
_disposed = true;
}
public void Dispose()
{
if (_disposed)
{
return;
}
if (_savedScope != null)
{
DbContextScope.SetAmbientScope(_savedScope);
_savedScope = null;
}
_disposed = true;
}
/// <summary>
/// Makes the provided 'dbContextScope' available as the the ambient scope via the CallContext.
/// </summary>
internal static void SetAmbientScope(DbContextScope newAmbientScope)
{
if (newAmbientScope == null)
{
throw new ArgumentNullException("newAmbientScope");
}
var current = CallContext.LogicalGetData(AmbientDbContextScopeKey) as InstanceIdentifier;
if (current == newAmbientScope._instanceIdentifier)
{
return;
}
// Store the new scope's instance identifier in the CallContext, making it the ambient scope
CallContext.LogicalSetData(AmbientDbContextScopeKey, newAmbientScope._instanceIdentifier);
// Keep track of this instance (or do nothing if we're already tracking it)
DbContextScopeInstances.GetValue(newAmbientScope._instanceIdentifier, key => newAmbientScope);
}
/// <summary>
/// Makes the provided 'dbContextScope' available as the the ambient scope via the CallContext.
/// </summary>
internal static void SetAmbientScope(DbContextScope newAmbientScope)
{
if (newAmbientScope == null)
{
throw new ArgumentNullException("newAmbientScope");
}
// Replace Call context with AsyncLocal
// https://stackoverflow.com/questions/41825375/why-asynclocal-is-different-from-callcontext
// var current = CallContext.LogicalGetData(AmbientDbContextScopeKey) as InstanceIdentifier;
var current = AsyncLocal.Value;
if (current == newAmbientScope._instanceIdentifier)
{
return;
}
// Store the new scope's instance identifier in the CallContext, making it the ambient scope
//CallContext.LogicalSetData(AmbientDbContextScopeKey, newAmbientScope._instanceIdentifier);
AsyncLocal.Value = newAmbientScope._instanceIdentifier;
// Keep track of this instance (or do nothing if we're already tracking it)
DbContextScopeInstances.GetValue(newAmbientScope._instanceIdentifier, key => newAmbientScope);
}
/// <summary>
/// 获取
/// </summary>
/// <typeparam name="TDbContext"></typeparam>
/// <returns></returns>
public TDbContext Get <TDbContext>() where TDbContext : DbContext
{
var ambientDbContextScope = DbContextScope.GetAmbientScope();
return(ambientDbContextScope == null ? null : ambientDbContextScope.DbContexts.Get <TDbContext>());
}
public DbContextReadOnlyScope(DbContextScopeOption joiningOption, IsolationLevel?isolationLevel, IDbContextFactory dbContextFactory = null)
{
_internalScope = new DbContextScope(joiningOption : joiningOption, readOnly : true, isolationLevel : isolationLevel, dbContextFactory : dbContextFactory);
}
public DbContextReadOnlyScope(DbContextScopeOption joiningOption, IsolationLevel? isolationLevel, IDbContextFactory dbContextFactory = null)
{
_internalScope = new DbContextScope(joiningOption: joiningOption, readOnly: true, isolationLevel: isolationLevel, dbContextFactory: dbContextFactory);
}
/// <summary>
/// Makes the provided 'dbContextScope' available as the the ambient scope via the CallContext.
/// </summary>
internal static void SetAmbientScope(DbContextScope newAmbientScope)
{
if (newAmbientScope == null)
throw new ArgumentNullException("newAmbientScope");
var current = CallContext.LogicalGetData(AmbientDbContextScopeKey) as InstanceIdentifier;
if (current == newAmbientScope._instanceIdentifier)
return;
// Store the new scope's instance identifier in the CallContext, making it the ambient scope
CallContext.LogicalSetData(AmbientDbContextScopeKey, newAmbientScope._instanceIdentifier);
// Keep track of this instance (or do nothing if we're already tracking it)
DbContextScopeInstances.GetValue(newAmbientScope._instanceIdentifier, key => newAmbientScope);
}
public TDbContext Get <TDbContext>(IDbTenant tenant) where TDbContext : TenantedDbContext
{
var ambientDbContextScope = DbContextScope.GetAmbientScope();
return(ambientDbContextScope == null ? null : ambientDbContextScope.DbContexts.Get <TDbContext>(tenant));
}
public TDbContext Get <TDbContext>(string connectionString) where TDbContext : DbContext
{
var ambientDbContextScope = DbContextScope.GetAmbientScope();
return(ambientDbContextScope == null ? null : ambientDbContextScope.DbContexts.Get <TDbContext>(connectionString));
}