java mybaits 读写分离_Spring Boot+MyBatis实现读写分离

申明：本教程收集于网络，请勿商用，仅供个人学习使用，如有侵权，请联系作者删除。

1.前言

读写分离要做的事情就是对于一条SQL该选择哪个数据库去执行，至于谁来做选择数据库这件事儿，无非两个，要么中间件帮我们做，要么程序自己做。因此，一般来讲，读写分离有两种实现方式。第一种是依靠中间件(比如：MyCat)，也就是说应用程序连接到中间件，中间件帮我们做SQL分离；第二种是应用程序自己去做分离。这里我们选择程序自己来做，主要是利用Spring提供的路由数据源，以及AOP。

然而，应用程序层面去做读写分离最大的弱点(不足之处)在于无法动态增加数据库节点，因为数据源配置都是写在配置中的，新增数据库意味着新加一个数据源，必然改配置，并重启应用。当然，好处就是相对简单。

​

2.AbstractRoutingDataSource

基于特定的查找key路由到特定的数据源。它内部维护了一组目标数据源，并且做了路由key与目标数据源之间的映射，提供基于key查找数据源的方法。

​

3.实践

3.1.maven依赖

4.0.0

com.cjs.example

cjs-datasource-demo

0.0.1-SNAPSHOT

jar

cjs-datasource-demo

org.springframework.boot

spring-boot-starter-parent

2.0.5.RELEASE

UTF-8

UTF-8

1.8

org.springframework.boot

spring-boot-starter-aop

org.springframework.boot

spring-boot-starter-jdbc

org.springframework.boot

spring-boot-starter-web

org.mybatis.spring.boot

mybatis-spring-boot-starter

1.3.2

org.apache.commons

commons-lang3

3.8

mysql

mysql-connector-java

runtime

org.springframework.boot

spring-boot-starter-test

test

org.springframework.boot

spring-boot-maven-plugin

3.2. 数据源配置

application.yml

spring:

datasource:

master:

jdbc-url: jdbc:mysql://192.168.1.16:3306/test

username: root

password:123456driver-class-name: com.mysql.jdbc.Driver

slave1:

jdbc-url: jdbc:mysql://192.168.1.26:3306/test

username: pig # 只读账户

password:123456driver-class-name: com.mysql.jdbc.Driver

slave2:

jdbc-url: jdbc:mysql://192.168.1.36:3306/test

username: pig # 只读账户

password:123456driver-class-name: com.mysql.jdbc.Driver

多数据源配置

/*** 关于数据源配置，参考SpringBoot官方文档第79章《Data Access》

* 79. Data Access

* 79.1 Configure a Custom DataSource

* 79.2 Configure Two DataSources*/@Configurationpublic classDataSourceConfig {

@Bean

@ConfigurationProperties("spring.datasource.master")publicDataSource masterDataSource() {returnDataSourceBuilder.create().build();

}

@Bean

@ConfigurationProperties("spring.datasource.slave1")publicDataSource slave1DataSource() {returnDataSourceBuilder.create().build();

}

@Bean

@ConfigurationProperties("spring.datasource.slave2")publicDataSource slave2DataSource() {returnDataSourceBuilder.create().build();

}

@Beanpublic DataSource myRoutingDataSource(@Qualifier("masterDataSource") DataSource masterDataSource,

@Qualifier("slave1DataSource") DataSource slave1DataSource,

@Qualifier("slave2DataSource") DataSource slave2DataSource) {

Map targetDataSources = new HashMap<>();

targetDataSources.put(DBTypeEnum.MASTER, masterDataSource);

targetDataSources.put(DBTypeEnum.SLAVE1, slave1DataSource);

targetDataSources.put(DBTypeEnum.SLAVE2, slave2DataSource);

MyRoutingDataSource myRoutingDataSource= newMyRoutingDataSource();

myRoutingDataSource.setDefaultTargetDataSource(masterDataSource);

myRoutingDataSource.setTargetDataSources(targetDataSources);returnmyRoutingDataSource;

}

}

这里，我们配置了4个数据源，1个master，2两个slave，1个路由数据源。前3个数据源都是为了生成第4个数据源，而且后续我们只用这最后一个路由数据源。

MyBatis配置

@EnableTransactionManagement

@Configurationpublic classMyBatisConfig {

​

@Resource(name= "myRoutingDataSource")privateDataSource myRoutingDataSource;

​

@Beanpublic SqlSessionFactory sqlSessionFactory() throwsException {

SqlSessionFactoryBean sqlSessionFactoryBean= newSqlSessionFactoryBean();

sqlSessionFactoryBean.setDataSource(myRoutingDataSource);

sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/*.xml"));returnsqlSessionFactoryBean.getObject();

}

​

@BeanpublicPlatformTransactionManager platformTransactionManager() {return newDataSourceTransactionManager(myRoutingDataSource);

}

}​

由于Spring容器中现在有4个数据源，所以我们需要为事务管理器和MyBatis手动指定一个明确的数据源。

3.3.设置路由key / 查找数据源

目标数据源就是那前3个这个我们是知道的，但是使用的时候是如果查找数据源的呢？

首先，我们定义一个枚举来代表这三个数据源

packagecom.muding.example.enums;

​public enumDBTypeEnum {

MASTER, SLAVE1, SLAVE2;

}

接下来，通过ThreadLocal将数据源设置到每个线程上下文中

public classDBContextHolder {

​private static final ThreadLocal contextHolder = new ThreadLocal<>();

​private static final AtomicInteger counter = new AtomicInteger(-1);

​public static voidset(DBTypeEnum dbType) {

contextHolder.set(dbType);

}

​public staticDBTypeEnum get() {returncontextHolder.get();

}

​public static voidmaster() {

set(DBTypeEnum.MASTER);

System.out.println("切换到master");

}

​public static voidslave() {//轮询

int index = counter.getAndIncrement() % 2;if (counter.get() > 9999) {

counter.set(-1);

}if (index == 0) {

set(DBTypeEnum.SLAVE1);

System.out.println("切换到slave1");

}else{

set(DBTypeEnum.SLAVE2);

System.out.println("切换到slave2");

}

}

}

获取路由key

public class MyRoutingDataSource extendsAbstractRoutingDataSource {

@Nullable

@OverrideprotectedObject determineCurrentLookupKey() {returnDBContextHolder.get();

}

}

设置路由key

默认情况下，所有的查询都走从库，插入/修改/删除走主库。我们通过方法名来区分操作类型(CRUD)

@Aspect

@Componentpublic classDataSourceAop {

​

@Pointcut("!@annotation(com.muding.example.annotation.Master) " +

"&& (execution(* com.muding​.example.service..*.select*(..)) " +

"|| execution(* com.muding​.example.service..*.get*(..)))")public voidreadPointcut() {

​

}

​

@Pointcut("@annotation(com.muding​.example.annotation.Master) " +

"|| execution(* com.muding​.example.service..*.insert*(..)) " +

"|| execution(* com.muding​.example.service..*.add*(..)) " +

"|| execution(* com.muding​.example.service..*.update*(..)) " +

"|| execution(* com.muding​.example.service..*.edit*(..)) " +

"|| execution(* com.muding​.example.service..*.delete*(..)) " +

"|| execution(* com.muding​.example.service..*.remove*(..))")public voidwritePointcut() {

​

}

​

@Before("readPointcut()")public voidread() {

DBContextHolder.slave();

}

​

@Before("writePointcut()")public voidwrite() {

DBContextHolder.master();

}

​

​/*** 另一种写法：if...else... 判断哪些需要读从数据库，其余的走主数据库*/

//@Before("execution(* com.muding​.example.service.impl.*.*(..))")//public void before(JoinPoint jp) {//String methodName = jp.getSignature().getName();//

//if (StringUtils.startsWithAny(methodName, "get", "select", "find")) {//DBContextHolder.slave();//}else {//DBContextHolder.master();//}//}

}

有一般情况就有特殊情况，特殊情况是某些情况下我们需要强制读主库，针对这种情况，我们定义一个主键，用该注解标注的就读主库

packagecom.muding.example.annotation;

​public @interfaceMaster {

}

例如，假设我们有一张表member

Servicepublic class MemberServiceImpl implementsMemberService {

​

@AutowiredprivateMemberMapper memberMapper;

​

@Transactional

@Overridepublic intinsert(Member member) {returnmemberMapper.insert(member);

}

​

@Master

@Overridepublic intsave(Member member) {returnmemberMapper.insert(member);

}

​

@OverridepublicList selectAll() {return memberMapper.selectByExample(newMemberExample());

}

​

@Master

@OverridepublicString getToken(String appId) {//有些读操作必须读主数据库//比如，获取微信access_token，因为高峰时期主从同步可能延迟//这种情况下就必须强制从主数据读

return null;

}

}

4.测试

@RunWith(SpringRunner.class)

@SpringBootTestpublic classCjsDatasourceDemoApplicationTests {

​

@AutowiredprivateMemberService memberService;

​

@Testpublic voidtestWrite() {

Member member= newMember();

member.setName("zhangsan");

memberService.insert(member);

}

@Testpublic voidtestRead() {for (int i = 0; i < 4; i++) {

memberService.selectAll();

}

}

@Testpublic voidtestSave() {

Member member= newMember();

member.setName("muding");

memberService.save(member);

}

@Testpublic voidtestReadFromMaster() {

memberService.getToken("1234");

}

}

查看控制台

​

​

5.工程结构

​

关注微信公众号"木不是丁"；获取更多学习资源

​