|
|
using DS.Module.UserModule;
|
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
|
using SqlSugar;
|
|
|
using NLog;
|
|
|
|
|
|
namespace DS.Module.SqlSugar;
|
|
|
|
|
|
/// <summary>
|
|
|
/// saas数据库服务
|
|
|
/// </summary>
|
|
|
public class SaasDbService : ISaasDbService
|
|
|
{
|
|
|
private readonly IServiceProvider _serviceProvider;
|
|
|
private readonly SqlSugarScope db;
|
|
|
private readonly IUser user;
|
|
|
private readonly ISaasDbService saasService;
|
|
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
|
|
|
|
|
/// <summary>
|
|
|
/// 构造函数
|
|
|
/// </summary>
|
|
|
/// <param name="serviceProvider"></param>
|
|
|
public SaasDbService(IServiceProvider serviceProvider)
|
|
|
{
|
|
|
_serviceProvider = serviceProvider;
|
|
|
db = (SqlSugarScope)_serviceProvider.GetRequiredService<ISqlSugarClient>();
|
|
|
user = _serviceProvider.GetRequiredService<IUser>();
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
|
/// 根据用户租户Id获取业务库
|
|
|
/// </summary>
|
|
|
/// <returns></returns>
|
|
|
public ISqlSugarClient GetBizDb()
|
|
|
{
|
|
|
var tenantId = user.GetTenantId().ToString();
|
|
|
|
|
|
if (!db.IsAnyConnection(tenantId))
|
|
|
{
|
|
|
var connInfo = GetMasterDb().Queryable<SysTenantLink>().First(x => x.TenantId == long.Parse(tenantId));
|
|
|
//用非默认ConfigId进行测试
|
|
|
//添加业务库只在当前上下文有效（原理：SqlSugarScope模式入门文档去看）
|
|
|
db.AddConnection(new ConnectionConfig()
|
|
|
{
|
|
|
ConfigId = tenantId,
|
|
|
ConnectionString = connInfo.Connection,
|
|
|
DbType = connInfo.DbType,
|
|
|
IsAutoCloseConnection = true
|
|
|
});
|
|
|
}
|
|
|
|
|
|
return db.GetConnection(tenantId);
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
|
/// 获取主库信息
|
|
|
/// </summary>
|
|
|
/// <returns></returns>
|
|
|
public ISqlSugarClient GetMasterDb()
|
|
|
{
|
|
|
return db.GetConnection("1288018625843826688");
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
|
/// 主库根据Id获取业务库
|
|
|
/// </summary>
|
|
|
/// <returns></returns>
|
|
|
public ISqlSugarClient GetBizDbById(object configId)
|
|
|
{
|
|
|
if(!db.IsAnyConnection(configId))
|
|
|
{
|
|
|
var connInfo = GetMasterDb().Queryable<SysTenantLink>().First(x => x.TenantId == Convert.ToInt64(configId));
|
|
|
//用非默认ConfigId进行测试
|
|
|
//添加业务库只在当前上下文有效（原理：SqlSugarScope模式入门文档去看）
|
|
|
db.AddConnection(new ConnectionConfig() {
|
|
|
ConfigId = configId,
|
|
|
ConnectionString = connInfo.Connection,
|
|
|
DbType = connInfo.DbType,
|
|
|
IsAutoCloseConnection = true });
|
|
|
}
|
|
|
|
|
|
return db.GetConnection(configId);
|
|
|
}
|
|
|
/// <summary>
|
|
|
/// 主库根据Id获取业务库Scope
|
|
|
/// </summary>
|
|
|
/// <param name="configId"></param>
|
|
|
/// <returns></returns>
|
|
|
public SqlSugarScopeProvider GetBizDbScopeById(object configId)
|
|
|
{
|
|
|
if(!db.IsAnyConnection(configId))
|
|
|
{
|
|
|
var connInfo = GetMasterDb().Queryable<SysTenantLink>().First(x => x.TenantId == Convert.ToInt64(configId));
|
|
|
//用非默认ConfigId进行测试
|
|
|
//添加业务库只在当前上下文有效（原理：SqlSugarScope模式入门文档去看）
|
|
|
db.AddConnection(new ConnectionConfig() {
|
|
|
ConfigId = configId,
|
|
|
ConnectionString = connInfo.Connection,
|
|
|
DbType = connInfo.DbType,
|
|
|
IsAutoCloseConnection = true });
|
|
|
}
|
|
|
|
|
|
return db.GetConnectionScope(configId);
|
|
|
}
|
|
|
public SqlSugarScopeProvider GetMasterDbScope()
|
|
|
{
|
|
|
return db.GetConnectionScope("1288018625843826688");
|
|
|
}
|
|
|
} |