ds8-solution-pro/ds-wms-service/DS.Module.SqlSugar/SaasDbService.cs

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 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(string configId)
    {
        if(!db.IsAnyConnection(configId))
        {
            var connInfo = GetMasterDb().Queryable<SysTenantLink>().First(x => x.TenantId == long.Parse(configId));
            //用非默认ConfigId进行测试
            //添加业务库只在当前上下文有效（原理：SqlSugarScope模式入门文档去看）
            db.AddConnection(new ConnectionConfig() { 
                ConfigId = configId, 
                ConnectionString = connInfo.Connection, 
                DbType = connInfo.DbType,
                IsAutoCloseConnection = true });
        }

        return db.GetConnection(configId);
    }
}