You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.8 KiB
49 lines
1.8 KiB
5 days ago
|
namespace Admin.NET.Core;
|
||
|
|
||
|
/// <summary>
|
||
|
/// SqlSugar仓储类
|
||
|
/// </summary>
|
||
|
/// <typeparam name="T"></typeparam>
|
||
|
public class SqlSugarRepository<T> : SimpleClient<T> where T : class, new()
|
||
|
{
|
||
|
protected ITenant iTenant = null; // 多租户事务
|
||
|
|
||
|
public SqlSugarRepository(ISqlSugarClient context = null) : base(context)
|
||
|
{
|
||
|
iTenant = App.GetRequiredService<ISqlSugarClient>().AsTenant();
|
||
|
|
||
|
// 若实体贴有多库特性,则返回指定的连接
|
||
|
if (typeof(T).IsDefined(typeof(TenantAttribute), false))
|
||
|
{
|
||
|
base.Context = iTenant.GetConnectionScopeWithAttr<T>();
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// 若实体贴有系统表特性,则返回默认的连接
|
||
|
if (typeof(T).IsDefined(typeof(SystemTableAttribute), false))
|
||
|
{
|
||
|
base.Context = iTenant.GetConnectionScope(SqlSugarConst.ConfigId);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// 若当前未登录或是默认租户Id,则返回默认的连接
|
||
|
var tenantId = App.GetRequiredService<UserManager>().TenantId;
|
||
|
if (tenantId < 1 || tenantId.ToString() == SqlSugarConst.ConfigId) return;
|
||
|
|
||
|
// 根据租户Id切库
|
||
|
if (!iTenant.IsAnyConnection(tenantId.ToString()))
|
||
|
{
|
||
|
var tenant = App.GetRequiredService<SysCacheService>().Get<List<SysTenant>>(CacheConst.KeyTenant)
|
||
|
.FirstOrDefault(u => u.Id == tenantId);
|
||
|
iTenant.AddConnection(new ConnectionConfig()
|
||
|
{
|
||
|
ConfigId = tenant.Id,
|
||
|
DbType = tenant.DbType,
|
||
|
ConnectionString = tenant.Connection,
|
||
|
IsAutoCloseConnection = true
|
||
|
});
|
||
|
SqlSugarSetup.SetDbAop(iTenant.GetConnectionScope(tenantId.ToString()));
|
||
|
}
|
||
|
base.Context = iTenant.GetConnectionScope(tenantId.ToString());
|
||
|
}
|
||
|
}
|