多库操作2：终于实现多个数据库操作

▼

更多精彩推荐，上午10点到达

▼

圣诞节快乐

在上周的文章【多库操作：多个数据库的动态切换（一）】中，我们简单说了说，如何切换数据库，虽然实现了大部分的功能，但是最后也遗留了小问题，后来我和别的小伙伴讨论了下，那个小问题其实不是Bug，而是设计思路的偏差，所以我又重新思考了一下，做了一定的优化，从而实现了多个数据库共存的情况，当然目前这个也能满足事务提交。

那下面，我们就重新再来说说，【多库】操作到底是怎样的？具体的操作，可以点击【阅读原文】我写的wiki（https://github.com/anjoy8/Blog.Core/wiki/support-muti-db）。

借鉴大佬思路：@銀翼の奇術師

常见的两种多库操作方式

之前咱们简单说过，不过这里再详细的说一说，多库操作到底是如何操作的。

在平时开发中，我们习惯了面向数据库开发，就是一个项目下来以后，我们就会赶紧的去构思逻辑，然后领导一声令下，五点办公室开会，大家一顿操作，设计了一个堪称完美的数据库表结构，无论是备用字段，还是以后的业务逻辑的扩展，都应有尽有。后来为了满足可能遇到的各种情况，表是能详细就详细，不怕多，就怕改，这也是为何现在ORM很火的原因，当年我也是改了很久的DBHelper，修改一下，整个人都崩溃????

后来随着业务的发展，和数据库的瓶颈，就出现了分库的口号，大家开始拆分数据库了，常见的有两种模式：

①、读写分离，多个数据库的表结构是一样的，但是Query和Command不是在一起的，这样能突破瓶颈，使得业务能进一步提高。

②、模块分离，还是多个数据库，只不过每个数据库负责不同的模块，比如密码库，就只有密码表相关的，用户库仅仅是用户相关的，商品库就是商品相关的。

当然如果使用了微服务，是完全没有这个问题的，微服务就是多个api服务，每个api一个数据库，一个模块，一个业务逻辑的，比如这样：

（图片来源网络，侵删）

第一种方案呢，我在我的第二个DDD系列已经说到了，这里应该就不会在Blog.Core里再添加这个功能了，那今天咱们就做一下第②个方案，多个数据库负责不同的模块，可以进行不同的切换，当然如果以后想要新增其他模块功能，只需要自己新建个数据库就行了，然后配置连接字符串，轻松搞定，具体的好处很多，用到的时候肯定都知道了。

下面，我们就详细来说说具体的操作过程吧，随便交代一下，我用的ORM是Sqlsugar5.0.10+版本，其他ORM我没有尝试，自行修改。

修改Sqlsugar服务注入方式

在SqlsugarSetup.cs中，修改SqlSugarClient注入方式，要把多个数据库连接同时注入进去：

 /// <summary>/// SqlSugar 启动服务/// </summary>public static class SqlsugarSetup{public static void AddSqlsugarSetup(this IServiceCollection services){if (services == null) throw new ArgumentNullException(nameof(services));// 默认添加主数据库连接MainDb.CurrentDbConnId = Appsettings.app(new string[] { "MainDB" });// 把多个连接对象注入服务，这里必须采用Scope，因为有事务操作services.AddScoped<ISqlSugarClient>(o =>{var listConfig = new List<ConnectionConfig>()BaseDBConfig.MutiConnectionString.ForEach(m =>{listConfig.Add(new ConnectionConfig(){ConfigId = m.ConnId.ObjToString().ToLower(),ConnectionString = m.Conn,DbType = (DbType)m.DbType,IsAutoCloseConnection = true,IsShardSameThread = false,MoreSettings = new ConnMoreSettings(){IsAutoRemoveDataCache = true}//InitKeyType = InitKeyType.SystemTable});});return new SqlSugarClient(listConfig);});}}

修改工作单元，获取DB服务

在UnitOfWork.cs文件中，把刚刚第二步中注册的服务，通过构造函数注入获取，并且做相应的事务操作，注意，这里必须要保证同一个scope，如果想要实现事务，就必须保证DB的唯一性：

 public class UnitOfWork : IUnitOfWork{private readonly ISqlSugarClient _sqlSugarClient;public UnitOfWork(ISqlSugarClient sqlSugarClient){_sqlSugarClient = sqlSugarClient;}/// <summary>/// 获取DB，保证唯一性/// </summary>/// <returns></returns>public SqlSugarClient GetDbClient(){// 必须要as，后边会用到切换数据库操作return _sqlSugarClient as SqlSugarClient;}public void BeginTran(){GetDbClient().BeginTran();}public void CommitTran(){try{GetDbClient().CommitTran(); //}catch (Exception ex){GetDbClient().RollbackTran();throw ex;}}public void RollbackTran(){GetDbClient().RollbackTran();}}

如果修改好了，工作单元，记得也要修改要工作单元接口。

动态获取 _db 实例

在上边，我们在工作单元uow（unitOfWork）中，注入了数据库实例，那现在就要获取这个实例了，很简单，直接基类仓储BaseRepository.cs构造函数中，依赖注入我们的IUnitOfWork接口：

 private SqlSugarClient _dbBase;public BaseRepository(IUnitOfWork unitOfWork){_unitOfWork = unitOfWork;_dbBase = unitOfWork.GetDbClient();}

获取到这个 _dbBase 以后，其实这个时候已经可以了，我们就可以任意的使用这个db实例，但是我们今天的目的是要动态切换，重头戏来了，既然我们每次请求都需要这个db，那简单，我们就修改它的属性：

 private ISqlSugarClient _db{get{/* 如果要开启多库支持，* 1、在appsettings.json 中开启MutiDBEnabled节点为true，必填* 2、设置一个主连接的数据库ID，MainDB，必填*/if (Appsettings.app(new string[] { "MutiDBEnabled" }).ObjToBool()){// 默认会获取当前Model的特性，看看是否配置了连接db的ConnIDif (typeof(TEntity).GetTypeInfo().GetCustomAttributes(typeof(SugarTable), true).FirstOrDefault((x => x.GetType() == typeof(SugarTable))) is SugarTable sugarTable){_dbBase.ChangeDatabase(sugarTable.TableDescription.ToLower());}else{// 如果不存在，则表明当前Model是主数据库操作// 这个配置的地点，看文章上边第二节，注入服务的时候_dbBase.ChangeDatabase(MainDb.CurrentDbConnId.ToLower());}}return _dbBase;}}

实体类和连接字符串的配置

首先呢，我们需要在appsettings.json中，配置多个库的连接字符串，注意，如果想要打开多个，就要把那几个的Enabled全部设置为true：

  "MainDB": "WMBLOG_SQLITE",// 当前主库的连接字符串，不填写的话，默认是下边true的第一个"MutiDBEnabled": false,// 是否开启多库，默认是false"DBS": [/*MySql = 0,SqlServer = 1,Sqlite = 2,Oracle = 3,PostgreSQL = 4*/{"ConnId": "WMBLOG_SQLITE","DBType": 2,"Enabled": true,"Connection": "WMBlog.db" //只写数据库名就行，我会拼接字符串},{"ConnId": "WMBLOG_MSSQL","DBType": 1,"Enabled": true,"Connection": "Server=.;Database=WMBlogDB;User ID=sa;Password=123;","ProviderName": "System.Data.SqlClient"},{"ConnId": "WMBLOG_MYSQL","DBType": 0,"Enabled": false,"Connection": "Server=localhost; Port=3306;Stmt=; Database=wmblogdb; Uid=root; Pwd=456;"},{"ConnId": "WMBLOG_ORACLE","DBType": 3,"Enabled": false,"Connection": "Provider=OraOLEDB.Oracle; Data Source=WMBlogDB; User Id=sss; Password=789;","OracleConnection_other1": "User ID=sss;Password=789;Data Source=(DESCRIPTION=(ADDRESS_LIST=(ADDRESS=(PROTOCOL=TCP)(HOST=192.168.8.65)(PORT=1521)))(CONNECT_DATA=(SERVICE_NAME = orcl)))"}],

然后我们修改一下Model，配置SugarTable特性，第一个参数是表明，第二个是对应的db连接字符串的ConnID，这里我们用密码表做测试：

/// <summary>
/// 密码库表
/// </summary>
[SugarTable("PasswordLib", "WMBLOG_MSSQL")]
public class PasswordLib
{[SugarColumn(IsNullable = false, IsPrimaryKey = true, IsIdentity = true)]public int PLID { get; set; }}

到了这里，我们就可以随意的做多库操作了。

我们看看效果吧。

实现操作两个数据库效果

首先，开启多库配置：

我们测试两个数据库，一个是Sqlite主库，一个是Sqlserver从库，

从主库中，获取博客信息，从库中获取密码表信息，就是刚刚我们在上边配置的实体。

 /// <summary>/// 测试多库连接/// </summary>/// <returns></returns>[HttpGet("TestMutiDBAPI")][AllowAnonymous]public async Task<object> TestMutiDBAPI(){// 从主库（Sqlite）中，操作blogsvar blogs = await _blogArticleServices.Query(d => d.bID == 1);// 从从库（Sqlserver）中，获取pwdsvar pwds = await _passwordLibServices.Query(d => d.PLID > 0;return new{blogs,pwds};}

为了做对比效果，我把这两个表，从数据库中删掉，也就是blog在从库中删掉，pwd在主库中删掉：

我们做一个动图，来看看效果：

尽管两个表在对方的数据库不存在，但是我们还是获取到了数据：

那是不是这样就高枕无忧了呢，别慌！咱们是事务不会被破坏吧！来再试试。

检验事务操作是否正常

项目中有一个测试接口，大概意思就是先读取一个表数据，然后添加一条数据，中间制造一个异常，最后做回滚操作，看看是否添加的数据被删掉。

 [HttpGet]public async Task<IEnumerable<string>> Get(){List<string> returnMsg = new List<string>() { };try{returnMsg.Add($"Begin Transaction");_unitOfWork.BeginTran();var passwords = await _passwordLibServices.Query(d=>d.IsDeleted==false);returnMsg.Add($"first time : the count of passwords is :{passwords.Count}");returnMsg.Add($"insert a data into the table PasswordLib now.");var insertPassword = await _passwordLibServices.Add(new PasswordLib(){IsDeleted = false,plAccountName = "aaa",plCreateTime = DateTime.Now});passwords = await _passwordLibServices.Query(d => d.IsDeleted == false);returnMsg.Add($"second time : the count of passwords is :{passwords.Count}");returnMsg.Add($" ");//......var guestbooks = await _guestbookServices.Query();returnMsg.Add($"first time : the count of guestbooks is :{guestbooks.Count}");int ex = 0;returnMsg.Add($"There's an exception!!");returnMsg.Add($" ");int throwEx = 1 / ex;var insertGuestbook = await _guestbookServices.Add(new Guestbook(){username = "bbb",blogId = 1,createdate = DateTime.Now,isshow = true});guestbooks = await _guestbookServices.Query();returnMsg.Add($"first time : the count of guestbooks is :{guestbooks.Count}");returnMsg.Add($" ");_unitOfWork.CommitTran();}catch (Exception){_unitOfWork.RollbackTran();var passwords = await _passwordLibServices.Query();returnMsg.Add($"third time : the count of passwords is :{passwords.Count}");var guestbooks = await _guestbookServices.Query();returnMsg.Add($"third time : the count of guestbooks is :{guestbooks.Count}");}return returnMsg;}

动图太大，这么不放了，肯定是正确的，直接看结果吧：