MyBatis 多数据源读写分离（注解实现）

àì夳堔傛蜴生んèń 2021-07-31 20:48 462阅读 0赞

优质文章，及时送达

![640?wx\_fmt=jpeg][640_wx_fmt_jpeg]

作者 | hy\_xiaobin

来源 | juejin.im/post/5d8705e65188253f4b629f47

--------------------

首 先需 要建立两个库进行测试，我这里使用的是master\_test和slave\_test两个库，两张库都有一张同样的表（偷懒，喜喜），表结构

表名  t\_user

![640?wx\_fmt=jpeg][640_wx_fmt_jpeg 1]

表中分别添加两条不同数据，方便测试 主数据库记录name为xiaobin，从库为xiaoliu 开始使用Springboot 整合mybatis，首先引入pom文件

<?xml version="1.0" encoding="UTF-8"?>
    <project xmlns="http://maven.apache.org/POM/4.0.0"
            xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
            xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
       <modelVersion>4.0.0</modelVersion>
       <parent>
           <groupId>org.springframework.boot</groupId>
           <artifactId>spring-boot-starter-parent</artifactId>
           <version>2.1.4.RELEASE</version>
       </parent>
       <groupId>com.xiaobin</groupId>
       <artifactId>mysql_master_slave</artifactId>
       <version>1.0-SNAPSHOT</version>
    
       <properties>
           <java.version>1.8</java.version>
           <lombok.version>1.18.6</lombok.version>
           <mybatis.version>1.3.2</mybatis.version>
           <lombox.version>1.18.6</lombox.version>
       </properties>
    
       <dependencies>
           
           <dependency>
               <groupId>org.springframework.boot</groupId>
               <artifactId>spring-boot-starter-web</artifactId>
           </dependency>
           
           <dependency>
               <groupId>org.projectlombok</groupId>
               <artifactId>lombok</artifactId>
               <version>${lombok.version}</version>
           </dependency>
           
           <dependency>
               <groupId>org.springframework.boot</groupId>
               <artifactId>spring-boot-starter-test</artifactId>
           </dependency>
           
           <dependency>
               <groupId>org.projectlombok</groupId>
               <artifactId>lombok</artifactId>
               <version>${lombox.version}</version>
           </dependency>
           
           <dependency>
               <groupId>org.mybatis.spring.boot</groupId>
               <artifactId>mybatis-spring-boot-starter</artifactId>
               <version>${mybatis.version}</version>
           </dependency>
           
           <dependency>
               <groupId>mysql</groupId>
               <artifactId>mysql-connector-java</artifactId>
           </dependency>
           <dependency>
               <groupId>org.springframework.boot</groupId>
               <artifactId>spring-boot-starter-jdbc</artifactId>
           </dependency>
           
           <dependency>
               <groupId>com.alibaba</groupId>
               <artifactId>druid-spring-boot-starter</artifactId>
               <version>1.1.10</version>
           </dependency>
           
           <dependency>
               <groupId>org.springframework.boot</groupId>
               <artifactId>spring-boot-starter-aop</artifactId>
           </dependency>
    
       </dependencies>
    </project>

动态数据源配置

这里使用的数据源为druid，实现数据源之间的切换用@DataSource自定义注解，配置Aop进行切换 application.yml 配置文件

spring:
        datasource:
            type: com.alibaba.druid.pool.DruidDataSource
            druid:
              xiaobin-master:
                driverClassName: com.mysql.jdbc.Driver
                username: root
                password: root
                url: jdbc:mysql://localhost:3306/master_test?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8
              xiaobin-slave:
                driverClassName: com.mysql.jdbc.Driver
                username: root
                password: root
                url: jdbc:mysql://localhost:3306/slave_test?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8
    mybatis:
      mapper-locations: classpath:mapper/*.xml

#### 多数据源配置类 ####

@Configuration
    @Component
    public class DynamicDataSourceConfig {
    
    
        @Bean
        @ConfigurationProperties("spring.datasource.druid.xiaobin-master")
        public DataSource xiaobinMasterDataSource(){
            return DruidDataSourceBuilder.create().build();
        }
    
    
        @Bean
        @ConfigurationProperties("spring.datasource.druid.xiaobin-slave")
        public DataSource xiaobinSlaveDataSource(){
            return DruidDataSourceBuilder.create().build();
        }
    
        @Bean
        @Primary
        public DynamicDataSource dataSource(DataSource xiaobinMasterDataSource, DataSource xiaobinSlaveDataSource) {
            Map<Object, Object> targetDataSources = new HashMap<>();
            targetDataSources.put("xiaobin-master",xiaobinMasterDataSource);
            targetDataSources.put("xiaobin-slave", xiaobinSlaveDataSource);
            return new DynamicDataSource(xiaobinMasterDataSource, targetDataSources);
        }
    
    }

#### 动态数据源切换类 ####

public class DynamicDataSource  extends AbstractRoutingDataSource {
    
        private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
    
        public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
            super.setDefaultTargetDataSource(defaultTargetDataSource);
            super.setTargetDataSources(targetDataSources);
            super.afterPropertiesSet();
        }
    
        @Override
        protected Object determineCurrentLookupKey() {
            return getDataSource();
        }
    
        public static void setDataSource(String dataSource) {
            contextHolder.set(dataSource);
        }
    
        public static String getDataSource() {
            return contextHolder.get();
        }
    
        public static void clearDataSource() {
            contextHolder.remove();
        }
    }

#### 自定义@DataSource注解 ####

在需要切换数据的Dao添加此注解

package com.xiaobin.annotation;
    
    import java.lang.annotation.*;
    
    @Target(ElementType.METHOD)
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface DataSource {
        String name() default "";
    }

#### Aop切面类配置 ####

@Aspect
    @Component
    public class DataSourceAspect {
    
        @Pointcut("@annotation(com.xiaobin.annotation.DataSource)")
        public void dataSourcePointCut() {
    
        }
    
    
        @Around("dataSourcePointCut()")
        public Object around(ProceedingJoinPoint point) throws Throwable {
            MethodSignature signature = (MethodSignature) point.getSignature();
            Method method = signature.getMethod();
    
    
            DataSource dataSource = method.getAnnotation(DataSource.class);
            if(dataSource == null){
                DynamicDataSource.setDataSource("xiaobin-master");
            }else {
                DynamicDataSource.setDataSource(dataSource.name());
            }
    
            try {
                return point.proceed();
            } finally {
                DynamicDataSource.clearDataSource();
            }
        }
    }

#### 启动配置注解信息，重要（不然运行会报错） ####

@SpringBootApplication(exclude= {DataSourceAutoConfiguration.class})
    @MapperScan(basePackages = "com.xiaobin.mapper")
    @Import({DynamicDataSourceConfig.class})
    public class StartApp {
        public static void main(String[] args) {
            SpringApplication.run(StartApp.class);
        }
    }

#### 测试controller ####

@RestController
    @RequestMapping
    public class UserController {
    
        @Autowired
        private UserMapper userMapper;
    
        @GetMapping("/{name}/list")
        public List<TUser> list(@PathVariable("name")String name){
            if(name.equals("master")){
                return userMapper.queryAllWithMaster();
            }else{
                return userMapper.queryAllWithSlave();
            }
        }
    }

#### **效果图** ####

更具路径传值，进行主从数据源切换

![640?wx\_fmt=png][640_wx_fmt_png]

![640?wx\_fmt=png][640_wx_fmt_png 1]

#### 目录结构 ####

![640?wx\_fmt=png][640_wx_fmt_png 2]

源码地址（数据库需要自己创建）

https://gitee.com/MyXiaoXiaoBin/learning-to-share/tree/master/mysql\_master\_slave

--------------------

\-END-

![640?wx\_fmt=jpeg][640_wx_fmt_jpeg 2]

学Java，请关注公众号：Java后端

**喜** **欢** **文** **章** **，** **点** **个** **在看 ** ![640?wx\_fmt=jpeg][640_wx_fmt_jpeg 3]

[640_wx_fmt_jpeg]: /images/20210731/054e19d4844d4d63afabb1bc6df78eb4.png?wx_fmt=jpeg
[640_wx_fmt_jpeg 1]: /images/20210731/992adae298a944f2b4c316ae895aebc0.png?wx_fmt=jpeg
[640_wx_fmt_png]: /images/20210731/14754404c46c4145b994cbacc230d4e2.png?wx_fmt=png
[640_wx_fmt_png 1]: /images/20210731/cefe9f80a915470a9e4571517fd3f216.png?wx_fmt=png
[640_wx_fmt_png 2]: /images/20210731/979663bc7d2a406db852b68cd1e9935c.png?wx_fmt=png
[640_wx_fmt_jpeg 2]: /images/20210731/ff3b6614774448f5a2e021d3098565ae.png?wx_fmt=jpeg
[640_wx_fmt_jpeg 3]: /images/20210731/56a0c267d80944688670440ad03fb378.png?wx_fmt=jpeg