56 lines
2.2 KiB
C#
56 lines
2.2 KiB
C#
|
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
|||
|
|
//
|
|||
|
|
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
|||
|
|
//
|
|||
|
|
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
|||
|
|
|
|||
|
|
namespace Admin.NET.Core.Utils;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 切换数据库
|
|||
|
|
/// </summary>
|
|||
|
|
public static class SqlSugarExtensions
|
|||
|
|
{
|
|||
|
|
public static ISqlSugarClient ForTenant<TEntity>(this ISqlSugarClient db)
|
|||
|
|
{
|
|||
|
|
var attr = typeof(TEntity).GetCustomAttribute<TenantAttribute>();
|
|||
|
|
var tenantId = attr != null ? GetConfigIdFromAttribute(attr) : SqlSugarConst.DefaultTenantId;
|
|||
|
|
return db.AsTenant().GetConnection(tenantId ?? SqlSugarConst.DefaultTenantId);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static object GetConfigIdFromAttribute(TenantAttribute attr)
|
|||
|
|
{
|
|||
|
|
const string targetKey = "configId";
|
|||
|
|
var type = attr.GetType();
|
|||
|
|
|
|||
|
|
// 方式1：尝试通过属性获取
|
|||
|
|
var prop = type.GetProperty(targetKey, BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic);
|
|||
|
|
if (prop != null) return prop.GetValue(attr);
|
|||
|
|
|
|||
|
|
// 方式2：尝试通过私有字段获取
|
|||
|
|
var fieldNames = new[]
|
|||
|
|
{
|
|||
|
|
$"<{targetKey}>k__BackingField", // 自动属性字段
|
|||
|
|
$"_{targetKey}", // 常规私有字段 (如_configId)
|
|||
|
|
$"m_{targetKey}", // 匈牙利命名法
|
|||
|
|
"configId" // 直接字段访问
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
foreach (var name in fieldNames)
|
|||
|
|
{
|
|||
|
|
var field = type.GetField(name, BindingFlags.NonPublic | BindingFlags.Instance);
|
|||
|
|
if (field != null) return field.GetValue(attr);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 方式3：通过Dynamic访问（备用方案）
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
dynamic d = attr;
|
|||
|
|
return d.configId;
|
|||
|
|
}
|
|||
|
|
catch
|
|||
|
|
{
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|