59 lines
1.7 KiB
C#
59 lines
1.7 KiB
C#
|
||
namespace Admin.NET.Core.Utils;
|
||
public static class SqlSugarExtensions
|
||
{
|
||
private const string DefaultTenantId = "1300000000001";
|
||
|
||
public static ISqlSugarClient ForTenant<TEntity>(this ISqlSugarClient db)
|
||
{
|
||
var attr = typeof(TEntity).GetCustomAttribute<TenantAttribute>();
|
||
var tenantId = attr != null ? GetConfigIdFromAttribute(attr) : DefaultTenantId;
|
||
return db.AsTenant().GetConnection(tenantId ?? DefaultTenantId);
|
||
}
|
||
|
||
private static string 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) as string;
|
||
}
|
||
|
||
// 方式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) as string;
|
||
}
|
||
}
|
||
|
||
// 方式3：通过Dynamic访问（备用方案）
|
||
try
|
||
{
|
||
dynamic d = attr;
|
||
return d.configId;
|
||
}
|
||
catch
|
||
{
|
||
return null;
|
||
}
|
||
}
|
||
}
|
||
|