Mybatis实现分包定义数据库

语言: CN / TW / HK

Mybatis实现分包定义数据库

背景

业务需求中需要连接两个数据库处理数据,需要用动态数据源。通过了解mybatis的框架,计划 使用分包的方式进行数据源的区分。

原理

前提:

我们使用mybatis都会有四个步骤

1:构建 SqlSessionFactory

2:通过 SqlSessionFactory 获取到 sqlSession 对象

3:通过 sqlSession 对象获取 Mapper 的动态代理对象

4:通过执行动态代理对象获取返回值

其实点开sqlSessionFactory就的 Configuration 对象中的 Environment 对象绑定了我们的 dataSource 对象

同样,我们通过debug发现,动态代理后的mapper对象是同样持有 Configuration 对象,绑定我们的连接信息。

猜想

需要分包实现不同数据源的话,需要对不同的mapper指定不同的 sqlSessionFactroy ,查看 mapperScan 的注解源码,

通过理解注释,我们是可以指定maper动态代理对象的 sqlSessionFactory 对象的。然后结果 basePackages 就可以对不同的包使用不同的 sqlSessionFactory 从而实现不同包使用不同的数据源。

实现

1:定义两个数据源

两个数据源分别命名位ds01,ds02,这里采用都是德鲁伊的数据库连接池。为了简介,详细配置没有展示了。

2:定义两个 SqlSessionFactory

定义两个 sqlSessionFactory ,分别绑定两个数据源。

3:不同包实现绑定不同的 sqlSessionFactory

通过 @MapperScan 指定扫描的包,通知指定 SqlSessionFactory

@MapperScan(basePackages = "com.lenven.demo.dao.ds02",sqlSessionFactoryRef = "salveSqlSessionFactory")
@MapperScan(basePackages = "com.lenven.demo.dao.ds01",sqlSessionFactoryRef = "masterSqlSessionFactory")

4:测试

对两个不同的包的mapper测试,都可以正确查询出数据

拓展:

通过实现

AbstractRoutingDataSource 也可以实现动态数据源。

该类实现 DataSource 的接口,可以配置对各数据源在

@Nullable
    private Map<Object, DataSource> resolvedDataSources;

这个属性里面。

让后通过暴露一个 determineCurrentLookupKey 获取需要使用的数据源的key。但是需要注意事务的问题。