Spring Boot(七)：Mybatis 多数据源最简解决方案

​

说起多数据源，一般都来解决那些问题呢，主从模式或者业务比较复杂需要连接不同的分库来支持业务。我们遇到的情况是后者，网上找了很多，大都是根据 Jpa 来做多数据源解决方案，要不就是老的 Spring 多数据源解决方案，还有的是利用 Aop 动态切换，感觉有点小复杂，其实我只是想找一个简单的多数据支持而已，折腾了两个小时整理出来，供大家参考。

废话不多说直接上代码吧

我们以 Mybatis Xml 版本为例，给大家展示如何如何配置多数据源。

配置文件

Pom 包就不贴了比较简单该依赖的就依赖，主要是数据库这边的配置：

1. mybatis.config-location=classpath:mybatis/mybatis-config.xml

2.  

3. spring.datasource.test1.jdbc-url=jdbc:mysql://localhost:3306/test1?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true

4. spring.datasource.test1.username=root

5. spring.datasource.test1.password=root

6. spring.datasource.test1.driver-class-name=com.mysql.cj.jdbc.Driver

7.  

8. spring.datasource.test2.jdbc-url=jdbc:mysql://localhost:3306/test2?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true

9. spring.datasource.test2.username=root

10. spring.datasource.test2.password=root

11. spring.datasource.test2.driver-class-name=com.mysql.cj.jdbc.Driver

一个 test1 库和一个 test2 库，其中 test1 位主库，在使用的过程中必须指定主库，不然会报错。

数据源配置

1. @Configuration

2. @MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef = "test1SqlSessionTemplate")

3. public class DataSource1Config {

4.  

5. @Bean(name = "test1DataSource")

6. @ConfigurationProperties(prefix = "spring.datasource.test1")

7. @Primary

8. public DataSource testDataSource() {

9. return DataSourceBuilder.create().build();

10. }

11.  

12. @Bean(name = "test1SqlSessionFactory")

13. @Primary

14. public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) throws Exception {

15. SqlSessionFactoryBean bean = new SqlSessionFactoryBean();

16. bean.setDataSource(dataSource);

17. bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/mapper/test1/*.xml"));

18. return bean.getObject();

19. }

20.  

21. @Bean(name = "test1TransactionManager")

22. @Primary

23. public DataSourceTransactionManager testTransactionManager(@Qualifier("test1DataSource") DataSource dataSource) {

24. return new DataSourceTransactionManager(dataSource);

25. }

26.  

27. @Bean(name = "test1SqlSessionTemplate")

28. @Primary

29. public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {

30. return new SqlSessionTemplate(sqlSessionFactory);

31. }

32. }

最关键的地方就是这块了，一层一层注入,首先创建 DataSource，然后创建 SqlSessionFactory 再创建事务，最后包装到 SqlSessionTemplate 中。其中需要指定分库的 mapper 文件地址，以及分库dao层代码

1. @MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef = "test1SqlSessionTemplate")

这块的注解就是指明了扫描 dao 层，并且给 dao 层注入指定的 SqlSessionTemplate。所有 @Bean都需要按照命名指定正确。

dao 层和 xml层

dao 层和 xml 需要按照库来分在不同的目录，比如：test1 库 dao 层在 com.neo.mapper.test1 包下，test2 库在 com.neo.mapper.test2

1. public interface User1Mapper {

2. List<UserEntity> getAll();

3. UserEntity getOne(Long id);

4. void insert(UserEntity user);

5. void update(UserEntity user);

6. void delete(Long id);

7. }

xml 层

1. <mapper namespace="com.neo.mapper.test1.User1Mapper" >

2. <resultMap id="BaseResultMap" type="com.neo.model.User" >

3. <id column="id" property="id" jdbcType="BIGINT" />

4. <result column="userName" property="userName" jdbcType="VARCHAR" />

5. <result column="passWord" property="passWord" jdbcType="VARCHAR" />

6. <result column="user_sex" property="userSex" javaType="com.neo.enums.UserSexEnum"/>

7. <result column="nick_name" property="nickName" jdbcType="VARCHAR" />

8. </resultMap>

9.  

10. <sql id="Base_Column_List" >

11. id, userName, passWord, user_sex, nick_name

12. </sql>

13.  

14. <select id="getAll" resultMap="BaseResultMap" >

15. SELECT

16. <include refid="Base_Column_List" />

17. FROM users

18. </select>

19.  

20. <select id="getOne" parameterType="java.lang.Long" resultMap="BaseResultMap" >

21. SELECT

22. <include refid="Base_Column_List" />

23. FROM users

24. WHERE id = #{id}

25. </select>

26.  

27. <insert id="insert" parameterType="com.neo.model.User" >

28. INSERT INTO

29. users

30. (userName,passWord,user_sex)

31. VALUES

32. (#{userName}, #{passWord}, #{userSex})

33. </insert>

34.  

35. <update id="update" parameterType="com.neo.model.User" >

36. UPDATE

37. users

38. SET

39. <if test="userName != null">userName = #{userName},</if>

40. <if test="passWord != null">passWord = #{passWord},</if>

41. nick_name = #{nickName}

42. WHERE

43. id = #{id}

44. </update>

45.  

46. <delete id="delete" parameterType="java.lang.Long" >

47. DELETE FROM

48. users

49. WHERE

50. id =#{id}

51. </delete>

52.  

53. </mapper>

测试

测试可以使用 SpringBootTest,也可以放到 Controller中，这里只贴 Controller 层的使用

1. @RestController

2. public class UserController {

3.  

4. @Autowired

5. private User1Mapper user1Mapper;

6.  

7. @Autowired

8. private User2Mapper user2Mapper;

9.  

10. @RequestMapping("/getUsers")

11. public List<UserEntity> getUsers() {

12. List<UserEntity> users=user1Mapper.getAll();

13. return users;

14. }

15.  

16. @RequestMapping("/getUser")

17. public UserEntity getUser(Long id) {

18. UserEntity user=user2Mapper.getOne(id);

19. return user;

20. }

21.  

22. @RequestMapping("/add")

23. public void save(UserEntity user) {

24. user2Mapper.insert(user);

25. }

26.  

27. @RequestMapping(value="update")

28. public void update(UserEntity user) {

29. user2Mapper.update(user);

30. }

31.  

32. @RequestMapping(value="/delete/{id}")

33. public void delete(@PathVariable("id") Long id) {

34. user1Mapper.delete(id);

35. }

36.  

37. }

Mybatis 注解版本配置多数据源和 Xml 版本基本一致，搭建直接参考文末的示例项目。

文章内容已经升级到 Spring Boot 2.x

示例代码-https://github.com/ityouknow/spring-boot-examples/tree/master/spring-boot-mybatis

 

精彩回顾：

面试点：Java 中 hashCode() 和 equals() 的关系

容器连接[Docker 系列-7]

现如今的技术浪潮中，我们到底该做些什么？

 

强烈推荐：

《Java 极客技术》知识星球限时优惠，现在加入只需 50 元，仅限前 1000 名，机不可失时不再来。趁早行动吧！

https://t.zsxq.com/J6Em2nU

 

隆重介绍：

Java 极客技术公众号，是由一群热爱 Java 开发的技术人组建成立，专注分享原创、高质量的 Java 文章。如果您觉得我们的文章还不错，请帮忙赞赏、在看、转发支持，鼓励我们分享出更好的文章。