EntityFramework實現多數據源動態切換

時間 2020-10-25

標籤 web sql 數據庫 app ide spa code orm server blog 欄目 HTML 简体版

原文原文鏈接

作項目的時候通常是一個連接一個庫，可是有時候碰上分庫的需求場景，好比saas模式，a租戶數據存在a數據庫，b租戶數據存在b數據庫。那怎麼來實現數據源的動態、靈活的切換呢？下面咱們就來看看怎麼實現。web

實現思路

一、 web.config配置數據庫鏈接串sql

 <add name="Bqool_ServiceEntities" connectionString="metadata=res://*/BqoolService.csdl|res://*/BqoolService.ssdl|res://*/BqoolService.msl;provider=System.Data.SqlClient;provider connection string=&quot;data source={0};initial catalog={1};persist security info=True;user id=XXXXXX;password=XXXXX;MultipleActiveResultSets=True;App=EntityFramework&quot;" providerName="System.Data.EntityClient" />

二、注入ConnectionService數據庫

conn.Bqool_Service).Keyed<IConnectionService>(ConnectionType.Bqool_Service));

三、獲取動態生成的數據庫連接app

public class ConnectionSchDbService : IConnectionService
    {
        private readonly string _sourceConnectionString;
        private readonly IAccountServiceRelService _accountServiceRel;
        private readonly ILoginStatusServices _loginStatusService;
        private bool? _isConnectUI;
        private readonly Logger _logger;

        public ConnectionSchDbService(
            string sourceConnectionString,
            IAccountServiceRelService accountServiceRel,
            ILoginStatusServices loginStatusService
)
        {
            _sourceConnectionString = sourceConnectionString;
            _accountServiceRel = accountServiceRel;
            _loginStatusService = loginStatusService;

            _logger = LogManager.GetLogger(GetType().FullName);
        }

        public string GetDbConnectionString(string account = null)
        {
            var isConnectUI = false;
            if (_isConnectUI == true)
            {
                isConnectUI = true;
            }
            return SetConnection(account, isConnectUI);
        }

        public string GetServerName(string account = null)
        {
            var accountServiceRel = GetAccountServiceRelHistory(account);
            if (accountServiceRel != null)
            {
                return accountServiceRel.Server_Name;
            }
            return null;
        }

        public string GetServiceName(string account = null)
        {
            var accountServiceRel = GetAccountServiceRelHistory(account);
            if (accountServiceRel != null)
            {
                return accountServiceRel.Schedule_Name;
            }
            return null;
        }

        public void SetConnectionUI(bool isConnectUI)
        {
            _isConnectUI = isConnectUI;
        }

        private string SetConnection(string account, bool isConnectUI = false)
        {
            var environmentMode = CommonSetting.GetEnvironmentMode();
            var returnConnection = string.Empty;

            //Default Setting (Account Service Rel 還沒創建的預設值)
            switch (environmentMode)
            {
                case EnvironmentMode.Dev: //開發
                    returnConnection = string.Format(_sourceConnectionString, "192.168.xxxx.xxxx", "Servicexxxx"); //開發環境使用 service1 ~ service3
                  xxxxx
                    break;

                xxxxxx

                default:
                    break;
            }

            //連線到 Account_Service_Rel_History
            var accountServiceRel = GetAccountServiceRelHistory(account);
            if (accountServiceRel != null)
            {
                returnConnection = string.Format(_sourceConnectionString, accountServiceRel.Server_IP, accountServiceRel.Schedule_Name);
            }

            if (!System.Environment.UserInteractive && System.Web.HttpContext.Current == null) //Windows Service
            {
                _logger.Debug($"account:{account ?? _loginStatusService.GetAccount()}, isConnectUI:{isConnectUI}, conn:{StringTools.ReplaceRegex(returnConnection, @"user.*", "")}");
            }
            return returnConnection;
        }

public interface IConnectionService
    {
        /// <summary>
        /// 取得資料庫連線
        /// </summary>
        /// <param name="account">不填使用登入的 Account</param>
        /// <returns></returns>
        string GetDbConnectionString(string account = null);

4 、定義訪問的上下文DbContextide

  public class DbContextFactory : IDbContextFactory
    {
        private readonly IConnectionService _connectionService;
        private readonly ILoginStatusServices _loginStatusServices;
        private readonly string _user;

        public DbContextFactory(IConnectionService connectionService, ILoginStatusServices loginStatusService)
        {
            _connectionService = connectionService;
            _loginStatusServices = loginStatusService;

            _user = loginStatusService.GetUserEmail();
        }

        private string _defaultConnectionString;
        private DbContext _dbContext;
        private SqlConnection _sqlConnection;
        private string _account;
        private string _serverName;
        private string _serviceName;

        public DbContext GetDbContext(string account = null)
        {
            var isResetConnection = IsResetConnection(account);
            if (_dbContext != null && isResetConnection)
            {
                DisposeDbContext();
            }
            if (_dbContext == null || isResetConnection)
            {                
                if (string.IsNullOrEmpty(_defaultConnectionString) || isResetConnection)
                {
                    _defaultConnectionString = _connectionService.GetDbConnectionString(account);
                }

                Type t = typeof(DbContext);
                _dbContext =
                    (DbContext)Activator.CreateInstance(t, _defaultConnectionString);
            }

            return _dbContext;
        }

5 、根據獲得的上下文連接SqlConnection ，訪問數據庫spa

 public class GenericDapperRep : IGenericDapperRep
    {
        private IDbContextFactory _factory;
        protected string _serverName { get; set; }
        protected string _serviceName { get; set; }
        protected string _user { get; set; }
        protected SqlConnection _conn
        {
            get
            {
                return _factory.GetSqlConnection();
            }
        }

        public GenericDapperRep(IDbContextFactory factory)
        {
            _factory = factory;
            //factory.SetConnectionUI(false);
            SetDbContextFactory(factory);
        }

        //public GenericDapperRep(IDbContextFactory factory, bool isConnectUI)
        //{
        //    _factory = factory;
        //    factory.SetConnectionUI(isConnectUI);
        //    SetDbContextFactory(factory);
        //}

        private void SetDbContextFactory(IDbContextFactory factory)
        {
            if (factory == null)
            {
                throw new ArgumentNullException("factory");
            }
            //_conn = factory.GetSqlConnection();
            _serverName = factory.GetServerName();
            _serviceName = factory.GetServiceName();

            _user = factory.GetUser();
        }

        protected T DbQueryWrapper<T>(Func<T> doWork)
        {
            return _factory.DbQueryWrapper(doWork);
        }

        protected void Create(string sql, object data)
        {
            SetCreate(data);
            _conn.Execute(sql, data);
        }