/// <summary>An IQueryable extension method that gets database context from the query.</summary>
/// <param name="query">The query to act on.</param>
/// <returns>The database context from the query.</returns>
public static DbContext GetDbContext(this IQueryable query)
{
var compilerField = typeof(EntityQueryProvider).GetField("_queryCompiler", BindingFlags.NonPublic | BindingFlags.Instance);
var compiler = (QueryCompiler)compilerField.GetValue(query.Provider);
var queryContextFactoryField = compiler.GetType().GetField("_queryContextFactory", BindingFlags.NonPublic | BindingFlags.Instance);
var queryContextFactory = (RelationalQueryContextFactory)queryContextFactoryField.GetValue(compiler);
#if EFCORE
var stateManagerField = typeof(QueryContextFactory).GetProperty("StateManager", BindingFlags.NonPublic | BindingFlags.Instance);
var stateManagerDynamic = stateManagerField.GetValue(queryContextFactory);
IStateManager stateManager = stateManagerDynamic as IStateManager;
if (stateManager == null)
{
Microsoft.EntityFrameworkCore.Internal.LazyRef <IStateManager> lazyStateManager = stateManagerDynamic as Microsoft.EntityFrameworkCore.Internal.LazyRef <IStateManager>;
if (lazyStateManager != null)
{
stateManager = lazyStateManager.Value;
}
}
if (stateManager == null)
{
stateManager = ((dynamic)stateManagerDynamic).Value;
}
#else
var stateManagerField = typeof(QueryContextFactory).GetField("_stateManager", BindingFlags.NonPublic | BindingFlags.Instance);
var stateManager = (IStateManager)stateManagerField.GetValue(queryContextFactory);
#endif
return(stateManager.Context);
}
private static IStateManager GetStateManagerInternal <T>(IQueryable <T> source, object queryContextFactory)
{
#if EFCORE
object stateManagerDynamic;
List <PropertyInfo> list = queryContextFactory.GetType().GetRuntimeProperties().ToList();
PropertyInfo dependenciesProperty = null;
foreach (PropertyInfo property in list)
{
if (property.Name.Equals("Dependencies"))
{
dependenciesProperty = property;
break;
}
}
if (dependenciesProperty != null)
{
// EFCore 2.x
var dependencies = dependenciesProperty.GetValue(queryContextFactory);
var stateManagerField = typeof(DbContext).GetTypeFromAssembly_Core("Microsoft.EntityFrameworkCore.Query.QueryContextDependencies").GetProperty("StateManager", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance);
stateManagerDynamic = stateManagerField.GetValue(dependencies);
}
else
{
// EFCore 1.x
var stateManagerField = typeof(QueryContextFactory).GetProperty("StateManager", BindingFlags.NonPublic | BindingFlags.Instance);
stateManagerDynamic = stateManagerField.GetValue(queryContextFactory);
}
IStateManager stateManager = stateManagerDynamic as IStateManager;
if (stateManager == null)
{
Microsoft.EntityFrameworkCore.Internal.LazyRef <IStateManager> lazyStateManager = stateManagerDynamic as Microsoft.EntityFrameworkCore.Internal.LazyRef <IStateManager>;
if (lazyStateManager != null)
{
stateManager = lazyStateManager.Value;
}
}
if (stateManager == null)
{
stateManager = ((dynamic)stateManagerDynamic).Value;
}
#else
var stateManagerField = typeof(QueryContextFactory).GetField("_stateManager", BindingFlags.NonPublic | BindingFlags.Instance);
var stateManager = (IStateManager)stateManagerField.GetValue(queryContextFactory);
#endif
return(stateManager);
}
/// <summary>An IQueryable extension method that gets database context from the query.</summary>
/// <param name="query">The query to act on.</param>
/// <returns>The database context from the query.</returns>
public static DbContext GetDbContext(this IQueryable query)
{
var compilerField = typeof(EntityQueryProvider).GetField("_queryCompiler", BindingFlags.NonPublic | BindingFlags.Instance);
var compiler = (QueryCompiler)compilerField.GetValue(query.Provider);
var queryContextFactoryField = compiler.GetType().GetField("_queryContextFactory", BindingFlags.NonPublic | BindingFlags.Instance);
var queryContextFactory = (RelationalQueryContextFactory)queryContextFactoryField.GetValue(compiler);
#if EFCORE
object stateManagerDynamic;
var dependenciesProperty = typeof(RelationalQueryContextFactory).GetProperty("Dependencies", BindingFlags.NonPublic | BindingFlags.Instance);
if (dependenciesProperty != null)
{
// EFCore 2.x
var dependencies = dependenciesProperty.GetValue(queryContextFactory);
var stateManagerField = typeof(DbContext).GetTypeFromAssembly_Core("Microsoft.EntityFrameworkCore.Query.QueryContextDependencies").GetProperty("StateManager", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance);
stateManagerDynamic = stateManagerField.GetValue(dependencies);
}
else
{
// EFCore 1.x
var stateManagerField = typeof(QueryContextFactory).GetProperty("StateManager", BindingFlags.NonPublic | BindingFlags.Instance);
stateManagerDynamic = stateManagerField.GetValue(queryContextFactory);
}
IStateManager stateManager = stateManagerDynamic as IStateManager;
if (stateManager == null)
{
Microsoft.EntityFrameworkCore.Internal.LazyRef <IStateManager> lazyStateManager = stateManagerDynamic as Microsoft.EntityFrameworkCore.Internal.LazyRef <IStateManager>;
if (lazyStateManager != null)
{
stateManager = lazyStateManager.Value;
}
}
if (stateManager == null)
{
stateManager = ((dynamic)stateManagerDynamic).Value;
}
#else
var stateManagerField = typeof(QueryContextFactory).GetField("_stateManager", BindingFlags.NonPublic | BindingFlags.Instance);
var stateManager = (IStateManager)stateManagerField.GetValue(queryContextFactory);
#endif
return(stateManager.Context);
}
