UNIVPLMDataIntegration/Admin.NET/Admin.NET.Core/Utils/SqlSugarExtensions.cs


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;
        }
    }
}