/// <summary>
/// 应用模块服务
/// </summary>
/// <param name="provider">服务提供者</param>
public override void UsePack(IServiceProvider provider)
{
OSharpOptions options = provider.GetOSharpOptions();
using (IServiceScope scope = provider.CreateScope())
{
ILogger logger = provider.GetService <ILoggerFactory>().CreateLogger(GetType());
TDbContext context = CreateDbContext(scope.ServiceProvider);
if (context != null)
{
OSharpDbContextOptions contextOptions = options.GetDbContextOptions(context.GetType());
if (contextOptions == null)
{
logger.LogWarning($"上下文类型“{context.GetType()}”的数据库上下文配置不存在");
return;
}
if (contextOptions.DatabaseType != DatabaseType.SqlServer)
{
logger.LogWarning($"上下文类型“{contextOptions.DatabaseType}”不是 {nameof(DatabaseType.SqlServer)} 类型");
return;
}
if (contextOptions.AutoMigrationEnabled)
{
context.CheckAndMigration();
DbContextModelCache modelCache = scope.ServiceProvider.GetService <DbContextModelCache>();
if (modelCache != null)
{
modelCache.Set(context.GetType(), context.Model);
}
IsEnabled = true;
}
}
}
}
/// <summary>
/// 将基于Osharp数据上下文基类<see cref="DbContextBase"/>上下文类型添加到服务集合中
/// </summary>
/// <typeparam name="TDbContext">基于Osharp数据上下文基类<see cref="DbContextBase"/>上下文类型</typeparam>
/// <param name="services">依赖注入服务集合</param>
/// <param name="optionsAction">数据库选项创建配置，将在内置配置后运行</param>
/// <returns>依赖注入服务集合</returns>
public static IServiceCollection AddOsharpDbContext <TDbContext>(this IServiceCollection services, Action <IServiceProvider, DbContextOptionsBuilder> optionsAction = null) where TDbContext : DbContextBase
{
services.AddDbContext <TDbContext>((provider, builder) =>
{
Type dbContextType = typeof(TDbContext);
OsharpOptions osharpOptions = provider.GetOSharpOptions();
OsharpDbContextOptions osharpDbContextOptions = osharpOptions?.GetDbContextOptions(dbContextType);
if (osharpDbContextOptions == null)
{
throw new OsharpException($"无法找到数据上下文“{dbContextType.DisplayName()}”的配置信息");
}
//启用延迟加载
if (osharpDbContextOptions.LazyLoadingProxiesEnabled)
{
builder = builder.UseLazyLoadingProxies();
}
DatabaseType databaseType = osharpDbContextOptions.DatabaseType;
//处理数据库驱动差异处理
IDbContextOptionsBuilderDriveHandler driveHandler = provider.GetServices <IDbContextOptionsBuilderDriveHandler>()
.FirstOrDefault(m => m.Type == databaseType);
if (driveHandler == null)
{
throw new OsharpException($"无法解析类型为“{databaseType}”的 {typeof(IDbContextOptionsBuilderDriveHandler).DisplayName()} 实例");
}
ScopedDictionary scopedDictionary = provider.GetService <ScopedDictionary>();
string key = $"DnConnection_{osharpDbContextOptions.ConnectionString}";
DbConnection existingDbConnection = scopedDictionary.GetValue <DbConnection>(key);
builder = driveHandler.Handle(builder, osharpDbContextOptions.ConnectionString, existingDbConnection);
//使用模型缓存
DbContextModelCache modelCache = provider.GetService <DbContextModelCache>();
IModel model = modelCache?.Get(dbContextType);
if (model != null)
{
builder = builder.UseModel(model);
}
//额外的选项
optionsAction?.Invoke(provider, builder);
});
return(services);
}
/// <summary>
/// 应用模块服务
/// </summary>
/// <param name="provider">服务提供者</param>
public override void UsePack(IServiceProvider provider)
{
OsharpOptions options = provider.GetOSharpOptions();
OsharpDbContextOptions contextOptions = options.GetDbContextOptions(typeof(TDbContext));
if (contextOptions?.DatabaseType != DatabaseType)
{
return;
}
using (IServiceScope scope = provider.CreateScope())
{
TDbContext context = CreateDbContext(scope.ServiceProvider);
if (context != null && contextOptions.AutoMigrationEnabled)
{
context.CheckAndMigration();
DbContextModelCache modelCache = scope.ServiceProvider.GetService <DbContextModelCache>();
modelCache?.Set(context.GetType(), context.Model);
}
}
IsEnabled = true;
}
