Mybatis 读写分离简单实现

package com.box.batisplus.db.source;

import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

@Slf4j
@Aspect
@Component
public class DataSourceAspect {

    @Pointcut("!@annotation(com.box.batisplus.annotation.Master) " +
            "&& (execution(* com.box.*.service..*.select*(..)) " +
            "|| execution(* com.box.*.service..*.get*(..)))"+
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.get*(..)))"+
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.list*(..)))"+
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.count*(..)))"+
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.page*(..)))")
    public void readPointcut() {

    }

    @Pointcut("@annotation(com.box.batisplus.annotation.Master) " +
            "|| execution(* com.box.*.service..*.insert*(..)) " +
            "|| execution(* com.box.*.service..*.add*(..)) " +
            "|| execution(* com.box.*.service..*.save*(..)) " +
            "|| execution(* com.box.*.service..*.update*(..)) " +
            "|| execution(* com.box.*.service..*.edit*(..)) " +
            "|| execution(* com.box.*.service..*.delete*(..)) " +
            "|| execution(* com.box.*.service..*.remove*(..))" +
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.remove*(..))" +
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.update*(..))" +
            "|| execution(* com.baomidou.mybatisplus.extension.service..*.save*(..))")

    public void writePointcut() {

    }

    @Before("readPointcut()")
    public void read() {
        DBContextHolder.slave();
    }

    @Before("writePointcut()")
    public void write() {
        DBContextHolder.master();
    }



    @After("readPointcut()")
    public void destroy() {
        log.info("DataSourceAspect.destroy is db:{}",DBContextHolder.get());
        if(DBContextHolder.get() != null){
            DBContextHolder.remove();
            log.info("DataSourceAspect.destroy is db remove:{}",DBContextHolder.get());
        }

    }

}

package com.box.batisplus.config;

/**
 * 数据源配置
 *
 * @author yuan.dingwang
 * @date 2021年03月30日 15:36
 */

import com.box.batisplus.db.source.MyRoutingDataSource;
import com.box.batisplus.enums.DBTypeEnum;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * 关于数据源配置，参考SpringBoot官方文档第79章《Data Access》
 * 79. Data Access
 * 79.1 Configure a Custom DataSource
 * 79.2 Configure Two DataSources
 */
@Slf4j
@Configuration
public class DataSourceConfig {

    @Bean
    @ConfigurationProperties("spring.datasource.master")
    public DataSource masterDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.slave1")
    public DataSource slave1DataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.slave2")
    public DataSource slave2DataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public DataSource myRoutingDataSource(@Qualifier("masterDataSource") DataSource masterDataSource,
                                          @Qualifier("slave1DataSource") DataSource slave1DataSource,
                                          @Qualifier("slave2DataSource") DataSource slave2DataSource) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DBTypeEnum.MASTER, masterDataSource);
        targetDataSources.put(DBTypeEnum.SLAVE1, slave1DataSource);
        targetDataSources.put(DBTypeEnum.SLAVE2, slave2DataSource);
        MyRoutingDataSource myRoutingDataSource = new MyRoutingDataSource();
        myRoutingDataSource.setDefaultTargetDataSource(masterDataSource);
        myRoutingDataSource.setTargetDataSources(targetDataSources);
        log.info("==============DataSourceConfig =====================");
        return myRoutingDataSource;
    }
}

################### 数据配置 start ##########################
datasource:
master:
jdbc-url: jdbc:mysql://47.103.114.48:3306/box_mall?useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&serverTimezone=GMT%2B8&useSSL=false&autoReconnect=true&failOverReadOnly=false
username: boxlife
password: boxlife2019
driver-class-name: com.mysql.cj.jdbc.Driver
druid:
initial-size: 5
max-active: 25
min-idle: 5
max-wait: 120000
min-evictable-idle-time-millis: 30000
max-evictable-idle-time-millis: 30000
time-between-eviction-runs-millis: 0
validation-query: select 1
validation-query-timeout: -1
test-on-borrow: false
test-on-return: false
test-while-idle: true
pool-prepared-statements: true
max-open-prepared-statements: 100
filters: stat
share-prepared-statements: true
# 通过connectProperties属性来打开mergeSql功能；慢SQL记录
connectProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500
slave1:
jdbc-url: jdbc:mysql://47.103.114.48:3306/box_mall?useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&serverTimezone=GMT%2B8&useSSL=false&autoReconnect=true&failOverReadOnly=false
username: boxlife
password: boxlife2019
driver-class-name: com.mysql.cj.jdbc.Driver
druid:
initial-size: 5
max-active: 25
min-idle: 5
max-wait: 120000
min-evictable-idle-time-millis: 30000
max-evictable-idle-time-millis: 30000
time-between-eviction-runs-millis: 0
validation-query: select 1
validation-query-timeout: -1
test-on-borrow: false
test-on-return: false
test-while-idle: true
pool-prepared-statements: true
max-open-prepared-statements: 100
filters: stat
share-prepared-statements: true
# 通过connectProperties属性来打开mergeSql功能；慢SQL记录
connectProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500
slave2:
jdbc-url: jdbc:mysql://47.103.114.48:3306/box_mall?useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&serverTimezone=GMT%2B8&useSSL=false&autoReconnect=true&failOverReadOnly=false
username: boxlife
password: boxlife2019
driver-class-name: com.mysql.cj.jdbc.Driver
druid:
initial-size: 5
max-active: 25
min-idle: 5
max-wait: 120000
min-evictable-idle-time-millis: 30000
max-evictable-idle-time-millis: 30000
time-between-eviction-runs-millis: 0
validation-query: select 1
validation-query-timeout: -1
test-on-borrow: false
test-on-return: false
test-while-idle: true
pool-prepared-statements: true
max-open-prepared-statements: 100
filters: stat
share-prepared-statements: true
# 通过connectProperties属性来打开mergeSql功能；慢SQL记录
connectProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500
################### mybatis plus start ##########################
mybatis-plus:
mapper-locations: classpath:/mapper/*Mapper.xml
#实体扫描，多个package用逗号或者分号分隔
typeAliasesPackage: com.box.member.model.entity
#typeEnumsPackage: com.baomidou.springboot.entity.enums
global-config:
#刷新mapper 调试神器
db-config:
#主键类型 0:"数据库ID自增", 1:"用户输入ID",2:"全局唯一ID (数字类型唯一ID)", 3:"全局唯一ID UUID";
id-type: INPUT
#字段策略 0:"忽略判断",1:"非 NULL 判断"),2:"非空判断"
field-strategy: not_empty
#驼峰下划线转换
column-underline: true
#数据库大写下划线转换
#capital-mode: true
#逻辑删除配置 注解已配置
#logic-delete-value: 1
#logic-not-delete-value: 0
db-type: mysql
refresh: true
#自定义填充策略接口实现
#meta-object-handler: com.baomidou.springboot.xxx
#自定义SQL注入器
#sql-injector: com.baomidou.mybatisplus.extension.injector.LogicSqlInjector
configuration:
map-underscore-to-camel-case: true
cache-enabled: false

package com.box.batisplus.config;

/**
 * 数据源配置
 *
 * @author yuan.dingwang
 * @date 2021年03月30日 15:36
 */

import com.box.batisplus.db.source.MyRoutingDataSource;
import com.box.batisplus.enums.DBTypeEnum;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * 关于数据源配置，参考SpringBoot官方文档第79章《Data Access》
 * 79. Data Access
 * 79.1 Configure a Custom DataSource
 * 79.2 Configure Two DataSources
 */
@Slf4j
@Configuration
public class DataSourceConfig {

    @Bean
    @ConfigurationProperties("spring.datasource.master")
    public DataSource masterDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.slave1")
    public DataSource slave1DataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.slave2")
    public DataSource slave2DataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public DataSource myRoutingDataSource(@Qualifier("masterDataSource") DataSource masterDataSource,
                                          @Qualifier("slave1DataSource") DataSource slave1DataSource,
                                          @Qualifier("slave2DataSource") DataSource slave2DataSource) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DBTypeEnum.MASTER, masterDataSource);
        targetDataSources.put(DBTypeEnum.SLAVE1, slave1DataSource);
        targetDataSources.put(DBTypeEnum.SLAVE2, slave2DataSource);
        MyRoutingDataSource myRoutingDataSource = new MyRoutingDataSource();
        myRoutingDataSource.setDefaultTargetDataSource(masterDataSource);
        myRoutingDataSource.setTargetDataSources(targetDataSources);
        log.info("==============DataSourceConfig =====================");
        return myRoutingDataSource;
    }
}