网站建设资讯

NEWS

网站建设资讯

Mybatis中怎么利用注解实现多数据源读写分离

今天就跟大家聊聊有关Mybatis中怎么利用注解实现多数据源读写分离,可能很多人都不太了解,为了让大家更加了解,小编给大家总结了以下内容,希望大家根据这篇文章可以有所收获。

成都创新互联公司专注于拉萨网站建设服务及定制,我们拥有丰富的企业做网站经验。 热诚为您提供拉萨营销型网站建设,拉萨网站制作、拉萨网页设计、拉萨网站官网定制、成都小程序开发服务,打造拉萨网络公司原创品牌,更为您提供拉萨网站排名全网营销落地服务。

首先需要建立两个库进行测试,我这里使用的是master_test和slave_test两个库,两张库都有一张同样的表(偷懒,喜喜),表结构

表名 t_user

| 字段名 | 类型 | 备注 || :------: | :------: | :------: || id | int | 主键自增ID || name | varchar | 名称 |

表中分别添加两条不同数据,方便测试

主数据库记录name为xiaobin,从库为xiaoliu

开始使用Springboot 整合mybatis,首先引入pom文件

 4.0.0      org.springframework.boot    spring-boot-starter-parent    2.1.4.RELEASE    com.xiaobin  MySQL_master_slave  1.0-SNAPSHOT      1.8    1.18.6    1.3.2    1.18.6                  org.springframework.boot      spring-boot-starter-web                  org.projectlombok      lombok      ${lombok.version}                  org.springframework.boot      spring-boot-starter-test                  org.projectlombok      lombok      ${lombox.version}                  org.mybatis.spring.boot      mybatis-spring-boot-starter      ${mybatis.version}                  mysql      mysql-connector-java              org.springframework.boot      spring-boot-starter-jdbc                  com.alibaba      druid-spring-boot-starter      1.1.10                  org.springframework.boot      spring-boot-starter-aop      

动态数据源配置

这里使用的数据源为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=utf8mybatis: mapper-locations: classpath:mapper/*.xml

多数据源配置类

package com.xiaobin.config;import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;import org.springframework.boot.context.properties.ConfigurationProperties;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.context.annotation.Primary;import org.springframework.stereotype.Component;import javax.sql.DataSource;import java.util.HashMap;import java.util.Map;/** * 创建时间: 2019/9/22 11:42 * 备注:多数据源配置信息 * 码农自学交流小群:260532022,欢迎大家的加入,分享学习是一件开心事 **/@Configuration@Componentpublic 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 targetDataSources = new HashMap<>();    targetDataSources.put("xiaobin-master",xiaobinMasterDataSource);    targetDataSources.put("xiaobin-slave", xiaobinSlaveDataSource);    return new DynamicDataSource(xiaobinMasterDataSource, targetDataSources);  }}

动态数据源切换类

package com.xiaobin.config;import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;import org.springframework.lang.Nullable;import javax.sql.DataSource;import java.util.Map;/** * 创建时间: 2019/9/22 11:51 * 备注:动态数据源 * 码农自学交流小群:260532022,欢迎大家的加入,分享学习是一件开心事 **/public class DynamicDataSource extends AbstractRoutingDataSource {  private static final ThreadLocal contextHolder = new ThreadLocal<>();  public DynamicDataSource(DataSource defaultTargetDataSource, Map 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.*;/** * 创建时间: 2019/9/22 11:53 * 备注:自定义数据源选择注解 * 码农自学交流小群:260532022,欢迎大家的加入,分享学习是一件开心事 **/@Target(ElementType.METHOD)@Retention(RetentionPolicy.RUNTIME)@Documentedpublic @interface DataSource {  String name() default "";}

Aop切面类配置

package com.xiaobin.aspect;import com.xiaobin.annotation.DataSource;import com.xiaobin.config.DynamicDataSource;import org.aspectj.lang.ProceedingJoinPoint;import org.aspectj.lang.annotation.Around;import org.aspectj.lang.annotation.Aspect;import org.aspectj.lang.annotation.Pointcut;import org.aspectj.lang.reflect.MethodSignature;import org.springframework.stereotype.Component;import java.lang.reflect.Method;/** * 创建时间: 2019/9/22 11:54 * 备注: * 码农自学交流小群:260532022,欢迎大家的加入,分享学习是一件开心事 **/@Aspect@Componentpublic 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();    }  }}

启动配置注解信息,重要(不然运行会报错)

package com.xiaobin;import com.xiaobin.config.DynamicDataSourceConfig;import org.mybatis.spring.annotation.MapperScan;import org.springframework.boot.SpringApplication;import org.springframework.boot.autoconfigure.SpringBootApplication;import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;import org.springframework.context.annotation.Import;/** * 创建时间: 2019/9/22 11:17 * 备注: * 码农自学交流小群:260532022,欢迎大家的加入,分享学习是一件开心事 **/@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

package com.xiaobin.api;import com.xiaobin.Entity.TUser;import com.xiaobin.mapper.UserMapper;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.web.bind.annotation.GetMapping;import org.springframework.web.bind.annotation.PathVariable;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RestController;import java.util.List; @RestController@RequestMappingpublic class UserController {  @Autowired  private UserMapper userMapper;  @GetMapping("/{name}/list")  public List list(@PathVariable("name")String name){    if(name.equals("master")){      return userMapper.queryAllWithMaster();    }else{      return userMapper.queryAllWithSlave();    }  }}

看完上述内容,你们对Mybatis中怎么利用注解实现多数据源读写分离有进一步的了解吗?如果还想了解更多知识或者相关内容,请关注创新互联行业资讯频道,感谢大家的支持。


文章名称:Mybatis中怎么利用注解实现多数据源读写分离
文章起源:http://njwzjz.com/article/jichoc.html