feat: 完成支持多db模式

This commit is contained in:
chenchun
2024-11-19 16:36:33 +08:00
parent 353a6b9d0c
commit eb2c05e9df
8 changed files with 284 additions and 285 deletions

View File

@@ -1,4 +1,5 @@
using SqlSugar; using SqlSugar;
using ArgumentException = System.ArgumentException;
namespace Yi.Framework.SqlSugarCore.Abstractions namespace Yi.Framework.SqlSugarCore.Abstractions
{ {
@@ -53,6 +54,5 @@ namespace Yi.Framework.SqlSugarCore.Abstractions
/// 开启Saas多租户 /// 开启Saas多租户
/// </summary> /// </summary>
public bool EnabledSaasMultiTenancy { get; set; } = false; public bool EnabledSaasMultiTenancy { get; set; } = false;
} }
} }

View File

@@ -1,24 +0,0 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
using SqlSugar;
namespace Yi.Framework.SqlSugarCore.Abstractions
{
public interface ISqlSugarDbConnectionCreator
{
DbConnOptions Options { get; }
Action<ISqlSugarClient> OnSqlSugarClientConfig { get; set; }
Action<object, DataAfterModel> DataExecuted { get; set; }
Action<object, DataFilterModel> DataExecuting { get; set; }
Action<string, SugarParameter[]> OnLogExecuting { get; set; }
Action<string, SugarParameter[]> OnLogExecuted { get; set; }
Action<PropertyInfo, EntityColumnInfo> EntityService { get; set; }
ConnectionConfig Build(Action<ConnectionConfig>? action = null);
void SetDbAop(ISqlSugarClient currentDb);
}
}

View File

@@ -1,7 +1,7 @@
using System.Reflection; using System.Reflection;
using SqlSugar; using SqlSugar;
namespace Yi.Framework.SqlSugarCore; namespace Yi.Framework.SqlSugarCore.Abstractions;
public interface ISqlSugarDbContextDependencies public interface ISqlSugarDbContextDependencies
{ {

View File

@@ -9,20 +9,31 @@ using Yi.Framework.SqlSugarCore.Abstractions;
namespace Yi.Framework.SqlSugarCore namespace Yi.Framework.SqlSugarCore
{ {
public static class SqlsugarCoreExtensions public static class SqlSugarCoreExtensions
{ {
/// <summary>
/// 新增db对象可支持多个
/// </summary>
/// <param name="service"></param>
/// <param name="serviceLifetime"></param>
/// <typeparam name="TDbContext"></typeparam>
/// <returns></returns>
public static IServiceCollection AddYiDbContext<TDbContext>(this IServiceCollection service, ServiceLifetime serviceLifetime = ServiceLifetime.Transient) where TDbContext : class, ISqlSugarDbContextDependencies public static IServiceCollection AddYiDbContext<TDbContext>(this IServiceCollection service, ServiceLifetime serviceLifetime = ServiceLifetime.Transient) where TDbContext : class, ISqlSugarDbContextDependencies
{ {
service.Add(new ServiceDescriptor(typeof(ISqlSugarDbContextDependencies), typeof(TDbContext), serviceLifetime)); service.AddTransient<ISqlSugarDbContextDependencies, TDbContext>();
return service; return service;
} }
/// <summary>
/// 新增db对象可支持多个
/// </summary>
/// <param name="service"></param>
/// <param name="options"></param>
/// <typeparam name="TDbContext"></typeparam>
/// <returns></returns>
public static IServiceCollection AddYiDbContext<TDbContext>(this IServiceCollection service, Action<DbConnOptions> options) where TDbContext : class, ISqlSugarDbContextDependencies public static IServiceCollection AddYiDbContext<TDbContext>(this IServiceCollection service, Action<DbConnOptions> options) where TDbContext : class, ISqlSugarDbContextDependencies
{ {
service.Configure<DbConnOptions>(ops => service.Configure<DbConnOptions>(options.Invoke);
{
options.Invoke(ops);
});
service.AddYiDbContext<TDbContext>(); service.AddYiDbContext<TDbContext>();
return service; return service;
} }

View File

@@ -1,6 +1,7 @@
using System.Reflection; using System.Reflection;
using SqlSugar; using SqlSugar;
using Volo.Abp.DependencyInjection; using Volo.Abp.DependencyInjection;
using Yi.Framework.SqlSugarCore.Abstractions;
namespace Yi.Framework.SqlSugarCore; namespace Yi.Framework.SqlSugarCore;

View File

@@ -1,10 +1,12 @@
using System.Reflection; using System.Collections.Concurrent;
using System.Reflection;
using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options; using Microsoft.Extensions.Options;
using SqlSugar; using SqlSugar;
using Volo.Abp.Data; using Volo.Abp.Data;
using Volo.Abp.DependencyInjection; using Volo.Abp.DependencyInjection;
using Volo.Abp.MultiTenancy; using Volo.Abp.MultiTenancy;
using Volo.Abp.Threading;
using Volo.Abp.Users; using Volo.Abp.Users;
using Yi.Framework.SqlSugarCore.Abstractions; using Yi.Framework.SqlSugarCore.Abstractions;
using Check = Volo.Abp.Check; using Check = Volo.Abp.Check;
@@ -17,6 +19,7 @@ namespace Yi.Framework.SqlSugarCore
/// SqlSugar 客户端 /// SqlSugar 客户端
/// </summary> /// </summary>
public ISqlSugarClient SqlSugarClient { get; private set; } public ISqlSugarClient SqlSugarClient { get; private set; }
private IAbpLazyServiceProvider LazyServiceProvider { get; } private IAbpLazyServiceProvider LazyServiceProvider { get; }
private ICurrentTenant CurrentTenant => LazyServiceProvider.LazyGetRequiredService<ICurrentTenant>(); private ICurrentTenant CurrentTenant => LazyServiceProvider.LazyGetRequiredService<ICurrentTenant>();
@@ -24,23 +27,29 @@ namespace Yi.Framework.SqlSugarCore
private ISerializeService SerializeService => LazyServiceProvider.LazyGetRequiredService<ISerializeService>(); private ISerializeService SerializeService => LazyServiceProvider.LazyGetRequiredService<ISerializeService>();
private IEnumerable<ISqlSugarDbContextDependencies> SqlSugarDbContextDependencies=>LazyServiceProvider.LazyGetRequiredService<IEnumerable<ISqlSugarDbContextDependencies>>(); private IEnumerable<ISqlSugarDbContextDependencies> SqlSugarDbContextDependencies =>
LazyServiceProvider.LazyGetRequiredService<IEnumerable<ISqlSugarDbContextDependencies>>();
private static readonly ConcurrentDictionary<string, ConnectionConfig> ConnectionConfigCache = new();
public SqlSugarDbContextFactory(IAbpLazyServiceProvider lazyServiceProvider) public SqlSugarDbContextFactory(IAbpLazyServiceProvider lazyServiceProvider)
{ {
LazyServiceProvider = lazyServiceProvider; LazyServiceProvider = lazyServiceProvider;
var connectionString = GetCurrentConnectionString();
//获取连接配置操作,需要进行缓存 //获取连接配置操作,需要进行缓存
var connectionConfig = BuildConnectionConfig(action: options => var connectionConfig = ConnectionConfigCache.GetOrAdd(connectionString, (_) =>
BuildConnectionConfig(action: options =>
{ {
options.ConnectionString = GetCurrentConnectionString(); options.ConnectionString = connectionString;
options.DbType = GetCurrentDbType(); options.DbType = GetCurrentDbType();
}); }));
SqlSugarClient = new SqlSugarClient(connectionConfig); SqlSugarClient = new SqlSugarClient(connectionConfig);
//生命周期以下都可以直接使用sqlsugardb了 //生命周期以下都可以直接使用sqlsugardb了
// Aop及多租户连接字符串和类型需要单独设置 // Aop及多租户连接字符串和类型需要单独设置
// Aop操作需要进行缓存 // Aop操作不能进行缓存
SetDbAop(SqlSugarClient); SetDbAop(SqlSugarClient);
} }
@@ -54,31 +63,30 @@ namespace Yi.Framework.SqlSugarCore
sqlSugarClient.CurrentConnectionConfig.ConfigureExternalServices.SerializeService = SerializeService; sqlSugarClient.CurrentConnectionConfig.ConfigureExternalServices.SerializeService = SerializeService;
//将所有ISqlSugarDbContextDependencies进行累加 //将所有ISqlSugarDbContextDependencies进行累加
Action<string, SugarParameter[]> onLogExecuting=null; Action<string, SugarParameter[]> onLogExecuting = null;
Action<string, SugarParameter[]> onLogExecuted=null; Action<string, SugarParameter[]> onLogExecuted = null;
Action<object, DataFilterModel> dataExecuting = null; Action<object, DataFilterModel> dataExecuting = null;
Action<object, DataAfterModel> dataExecuted = null; Action<object, DataAfterModel> dataExecuted = null;
Action<ISqlSugarClient> onSqlSugarClientConfig = null; Action<ISqlSugarClient> onSqlSugarClientConfig = null;
foreach (var dependency in SqlSugarDbContextDependencies.OrderBy(x=>x.ExecutionOrder)) foreach (var dependency in SqlSugarDbContextDependencies.OrderBy(x => x.ExecutionOrder))
{ {
onLogExecuting+= dependency.OnLogExecuting; onLogExecuting += dependency.OnLogExecuting;
onLogExecuted += dependency.OnLogExecuted; onLogExecuted += dependency.OnLogExecuted;
dataExecuting += dependency.DataExecuting; dataExecuting += dependency.DataExecuting;
dataExecuted += dependency.DataExecuted; dataExecuted += dependency.DataExecuted;
onSqlSugarClientConfig += dependency.OnSqlSugarClientConfig; onSqlSugarClientConfig += dependency.OnSqlSugarClientConfig;
} }
//最先存放db操作 //最先存放db操作
onSqlSugarClientConfig(sqlSugarClient); onSqlSugarClientConfig(sqlSugarClient);
sqlSugarClient.Aop.OnLogExecuting = onLogExecuting; sqlSugarClient.Aop.OnLogExecuting =onLogExecuting;
sqlSugarClient.Aop.OnLogExecuted =onLogExecuted; sqlSugarClient.Aop.OnLogExecuted = onLogExecuted;
sqlSugarClient.Aop.DataExecuting = dataExecuting;
sqlSugarClient.Aop.DataExecuted = dataExecuted;
sqlSugarClient.Aop.DataExecuting =dataExecuting;
sqlSugarClient.Aop.DataExecuted =dataExecuted;
} }
/// <summary> /// <summary>
@@ -86,14 +94,17 @@ namespace Yi.Framework.SqlSugarCore
/// </summary> /// </summary>
/// <returns></returns> /// <returns></returns>
/// <exception cref="ArgumentException"></exception> /// <exception cref="ArgumentException"></exception>
protected virtual ConnectionConfig BuildConnectionConfig(Action<ConnectionConfig>? action=null) protected virtual ConnectionConfig BuildConnectionConfig(Action<ConnectionConfig>? action = null)
{ {
var dbConnOptions = Options; var dbConnOptions = Options;
#region options #region options
if (dbConnOptions.DbType is null) if (dbConnOptions.DbType is null)
{ {
throw new ArgumentException("DbType配置为空"); throw new ArgumentException("DbType配置为空");
} }
var slavaConFig = new List<SlaveConnectionConfig>(); var slavaConFig = new List<SlaveConnectionConfig>();
if (dbConnOptions.EnabledReadWrite) if (dbConnOptions.EnabledReadWrite)
{ {
@@ -110,12 +121,14 @@ namespace Yi.Framework.SqlSugarCore
slavaConFig.Add(new SlaveConnectionConfig() { ConnectionString = s }); slavaConFig.Add(new SlaveConnectionConfig() { ConnectionString = s });
}); });
} }
#endregion #endregion
#region config #region config
var connectionConfig = new ConnectionConfig() var connectionConfig = new ConnectionConfig()
{ {
ConfigId= ConnectionStrings.DefaultConnectionStringName, ConfigId = ConnectionStrings.DefaultConnectionStringName,
DbType = dbConnOptions.DbType ?? DbType.Sqlite, DbType = dbConnOptions.DbType ?? DbType.Sqlite,
ConnectionString = dbConnOptions.Url, ConnectionString = dbConnOptions.Url,
IsAutoCloseConnection = true, IsAutoCloseConnection = true,
@@ -127,7 +140,7 @@ namespace Yi.Framework.SqlSugarCore
EntityNameService = (type, entity) => EntityNameService = (type, entity) =>
{ {
if (dbConnOptions.EnableUnderLine && !entity.DbTableName.Contains('_')) if (dbConnOptions.EnableUnderLine && !entity.DbTableName.Contains('_'))
entity.DbTableName = UtilMethods.ToUnderLine(entity.DbTableName);// 驼峰转下划线 entity.DbTableName = UtilMethods.ToUnderLine(entity.DbTableName); // 驼峰转下划线
}, },
EntityService = (c, p) => EntityService = (c, p) =>
{ {
@@ -138,16 +151,17 @@ namespace Yi.Framework.SqlSugarCore
} }
if (dbConnOptions.EnableUnderLine && !p.IsIgnore && !p.DbColumnName.Contains('_')) if (dbConnOptions.EnableUnderLine && !p.IsIgnore && !p.DbColumnName.Contains('_'))
p.DbColumnName = UtilMethods.ToUnderLine(p.DbColumnName);// 驼峰转下划线 p.DbColumnName = UtilMethods.ToUnderLine(p.DbColumnName); // 驼峰转下划线
//将所有ISqlSugarDbContextDependencies的EntityService进行累加 //将所有ISqlSugarDbContextDependencies的EntityService进行累加
//额外的实体服务需要这里配置, //额外的实体服务需要这里配置,
Action<PropertyInfo, EntityColumnInfo> entityService = null; Action<PropertyInfo, EntityColumnInfo> entityService = null;
foreach (var dependency in SqlSugarDbContextDependencies.OrderBy(x=>x.ExecutionOrder)) foreach (var dependency in SqlSugarDbContextDependencies.OrderBy(x => x.ExecutionOrder))
{ {
entityService += dependency.EntityService; entityService += dependency.EntityService;
} }
entityService(c, p); entityService(c, p);
} }
}, },
@@ -165,7 +179,9 @@ namespace Yi.Framework.SqlSugarCore
{ {
action.Invoke(connectionConfig); action.Invoke(connectionConfig);
} }
#endregion #endregion
return connectionConfig; return connectionConfig;
} }
@@ -177,7 +193,7 @@ namespace Yi.Framework.SqlSugarCore
{ {
var connectionStringResolver = LazyServiceProvider.LazyGetRequiredService<IConnectionStringResolver>(); var connectionStringResolver = LazyServiceProvider.LazyGetRequiredService<IConnectionStringResolver>();
var connectionString = var connectionString =
connectionStringResolver.ResolveAsync().ConfigureAwait(false).GetAwaiter().GetResult(); AsyncHelper.RunSync(() => connectionStringResolver.ResolveAsync());
if (string.IsNullOrWhiteSpace(connectionString)) if (string.IsNullOrWhiteSpace(connectionString))
{ {

View File

@@ -6,12 +6,9 @@ using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Logging; using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options; using Microsoft.Extensions.Options;
using SqlSugar; using SqlSugar;
using Volo.Abp;
using Volo.Abp.Auditing;
using Volo.Abp.Data; using Volo.Abp.Data;
using Volo.Abp.Domain; using Volo.Abp.Domain;
using Volo.Abp.Domain.Repositories; using Volo.Abp.Domain.Repositories;
using Volo.Abp.Modularity;
using Yi.Framework.SqlSugarCore.Abstractions; using Yi.Framework.SqlSugarCore.Abstractions;
using Yi.Framework.SqlSugarCore.Repositories; using Yi.Framework.SqlSugarCore.Repositories;
using Yi.Framework.SqlSugarCore.Uow; using Yi.Framework.SqlSugarCore.Uow;
@@ -47,6 +44,7 @@ namespace Yi.Framework.SqlSugarCore
//将默认db传递给abp连接字符串模块 //将默认db传递给abp连接字符串模块
Configure<AbpDbConnectionOptions>(x => { x.ConnectionStrings.Default = dbConfig.Url; }); Configure<AbpDbConnectionOptions>(x => { x.ConnectionStrings.Default = dbConfig.Url; });
context.Services.AddYiDbContext<DefaultSqlSugarDbContext>();
return Task.CompletedTask; return Task.CompletedTask;
} }
@@ -72,7 +70,6 @@ namespace Yi.Framework.SqlSugarCore
logger.LogInformation(sb.ToString()); logger.LogInformation(sb.ToString());
//Todo准备支持多租户种子数据及CodeFirst
if (options.EnabledCodeFirst) if (options.EnabledCodeFirst)
{ {

View File

@@ -22,10 +22,7 @@ using Volo.Abp.AspNetCore.Serilog;
using Volo.Abp.Auditing; using Volo.Abp.Auditing;
using Volo.Abp.Autofac; using Volo.Abp.Autofac;
using Volo.Abp.BackgroundJobs.Hangfire; using Volo.Abp.BackgroundJobs.Hangfire;
using Volo.Abp.BackgroundWorkers;
using Volo.Abp.BackgroundWorkers.Hangfire;
using Volo.Abp.Caching; using Volo.Abp.Caching;
using Volo.Abp.Hangfire;
using Volo.Abp.MultiTenancy; using Volo.Abp.MultiTenancy;
using Volo.Abp.Swashbuckle; using Volo.Abp.Swashbuckle;
using Yi.Abp.Application; using Yi.Abp.Application;
@@ -266,11 +263,12 @@ namespace Yi.Abp.Web
} }
else else
{ {
if (context.Request.Cookies.TryGetValue("Token",out var cookiesToken)) if (context.Request.Cookies.TryGetValue("Token", out var cookiesToken))
{ {
context.Token = cookiesToken; context.Token = cookiesToken;
} }
} }
return Task.CompletedTask; return Task.CompletedTask;
} }
}; };
@@ -314,7 +312,6 @@ namespace Yi.Abp.Web
context.Services.AddAuthorization(); context.Services.AddAuthorization();
return Task.CompletedTask; return Task.CompletedTask;
} }
@@ -375,7 +372,8 @@ namespace Yi.Abp.Web
app.UseAbpSerilogEnrichers(); app.UseAbpSerilogEnrichers();
//Hangfire定时任务面板可配置授权意框架支持jwt //Hangfire定时任务面板可配置授权意框架支持jwt
app.UseAbpHangfireDashboard("/hangfire", options => app.UseAbpHangfireDashboard("/hangfire",
options =>
{ {
options.AsyncAuthorization = new[] { new YiTokenAuthorizationFilter(app.ApplicationServices) }; options.AsyncAuthorization = new[] { new YiTokenAuthorizationFilter(app.ApplicationServices) };
}); });