springboot+druid+mybatis plus读写分离[多数据源配置]

读写分离可以通过多个数据源来实现,读数据从从库中读,写数据写入到主库中;思路:
1、yml中配置多个数据源;
2、通过AOP匹配key切换到不同的数据源;

一、AbstractRoutingDataSource

AbstractRoutingDataSource基于特定的key值到特定的数据源,内部维护了一组目标数据源,做了路由key和目标数据源之间的映射,我们可以基于指定的key来切换数据源;

image.png
public abstract class AbstractRoutingDataSource{
     /**determineCurrentLookupKey的返回值指定了key*/
     protected abstract Object determineCurrentLookupKey(){
        return key;
    }
}

二、pom.xml

maven的pom.xml文件,只写了必要的依赖,其它依赖可以自己添加;

<?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.0.0.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
  <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
        <java.version>1.8</java.version>
        <mybatisplus-spring-boot-starter.version>1.0.5</mybatisplus-spring-boot-starter.version>
        <mybatisplus.version>2.1.4</mybatisplus.version>       
        <druid.version>1.1.10</druid.version>    
        <skip>true</skip>
    </properties>
<dependencies>
  <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-context-support</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-cache</artifactId>
        </dependency>

        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.1.1</version>
            <exclusions>
                <exclusion>
                    <artifactId>mybatis</artifactId>
                    <groupId>org.mybatis</groupId>
                </exclusion>
                <exclusion>
                    <artifactId>mybatis-spring</artifactId>
                    <groupId>org.mybatis</groupId>
                </exclusion>
            </exclusions>
        </dependency>

        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid-spring-boot-starter</artifactId>
            <version>${druid.version}</version>
        </dependency>

        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatisplus-spring-boot-starter</artifactId>
            <version>${mybatisplus-spring-boot-starter.version}</version>
        </dependency> 

        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus</artifactId>
            <version>${mybatisplus.version}</version>
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
</dependencies>
    <build>
        <resources>
            <resource>
                <directory>src/main/java</directory>
                <includes>
                    <include>**/**</include>
                </includes>
            </resource>
            <resource>
                <directory>src/main/resources</directory>
                <includes>
                    <include>**/**</include>
                </includes>
            </resource>
        </resources>
  </build>      
</project>

三、application.yml数据源配置

#生产环境配置
server:
  #端口
  port: 8070
spring: 
  #redis配置
  redis:
    database: 0
    host: 127.0.0.1
    port: 6379
    password:    # 密码(默认为空)
    timeout: 6000ms  # 连接超时时长(毫秒)
    jedis:
      pool:
        max-active: 200
        max-idle: 1000  # 连接池中的最大空闲连接
        max-wait: -1s  # 连接池最大阻塞等待时间(使用负值表示没有限制)
        min-idle: 5 # 连接池中的最小空闲连接
  # 数据源配置
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    druid:
      stat-view-servlet:
        loginUsername: admin
        loginPassword: 123456
      validationQuery: SELECT 1
      initialSize: 10
      minIdle: 10
      maxActive: 200
      minEvictableIdleTimeMillis: 180000
      testOnBorrow: false
      testWhileIdle: true
      removeAbandoned: true
      removeAbandonedTimeout: 1800
      logAbandoned: true
      poolPreparedStatements: true
      maxOpenPreparedStatements: 100
    dynamic:
      datasource:
        master:
          username: root
          password: xxxxx
          driver-class-name: com.mysql.jdbc.Driver
          url: jdbc:mysql://xxx.xx.x.xx:3306/oneclick?useUnicode=true&characterEncoding=utf8&useSSL=false&tinyInt1isBit=true
        slave1:
          username: root
          password: xxxxx
          driver-class-name: com.mysql.jdbc.Driver
          url: jdbc:mysql://xxx.xx.x.xx:3306/oneclick?useUnicode=true&characterEncoding=utf8&useSSL=false&tinyInt1isBit=true

#日志
logging:
  config: classpath:log/logback.xml
  path: /log/oneclick/yjs-client-gateway.log/prod/

上面配置了两个数据数据源,主库master,从库slave1;主库负责写数据,从库负责读数据;

四、mybatis-plus和druid配置

4.1、mybatis-plus配置和数据源配置

package com.unnet.yjs.config.db;

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
import com.baomidou.mybatisplus.MybatisConfiguration;
import com.baomidou.mybatisplus.entity.GlobalConfiguration;
import com.baomidou.mybatisplus.plugins.PaginationInterceptor;
import com.baomidou.mybatisplus.plugins.PerformanceInterceptor;
import com.baomidou.mybatisplus.spring.MybatisSqlSessionFactoryBean;
import com.baomidou.mybatisplus.toolkit.GlobalConfigUtils;
import com.unnet.yjs.config.SysMetaObjectHandler;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.type.JdbcType;
import org.mybatis.spring.boot.autoconfigure.MybatisProperties;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StringUtils;

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


/**
 * Email: love1208tt@foxmail.com
 * Copyright (c)  2019. missbe
 * @author lyg   19-7-8 下午2:04
 *
 * MybatisPlus的数据源和插件配置
 **/

@Configuration
public class MybatisPlusConfig {
    private static final Logger LOGGER = LoggerFactory.getLogger(MybatisPlusConfig.class);
    @Resource
    private SysMetaObjectHandler sysMetaObjectHandler;
    @Resource
    private MybatisProperties mybatisProperties;


    /***
     * mybatis-plus的性能优化-设置最大执行时长和格式化sql输出
     */
    @Bean
    @Profile({"dev","test"})
    public PerformanceInterceptor performanceInterceptor() {
        LOGGER.info("dev和test环境下加载PerformanceInterceptor性能分析插件.");
        PerformanceInterceptor performanceInterceptor=new PerformanceInterceptor();
        /*<!-- SQL 执行性能分析,开发环境使用,线上不推荐。 maxTime 指的是 sql 最大执行时长 -->*/
        performanceInterceptor.setMaxTime(1000);
        /*<!--SQL是否格式化 默认false-->*/
        performanceInterceptor.setFormat(true);
        return performanceInterceptor;
    }

    /**
     * mybatis-plus分页插件-设置分页类型为mysql和本地分页
     */
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        LOGGER.info("加载PaginationInterceptor本地分页插件.");
        PaginationInterceptor page = new PaginationInterceptor();
        page.setLocalPage(true);
        page.setDialectType("mysql");
        return page;
    }

    /**
     * 数据库主库
     */
    @Bean
    @ConfigurationProperties("spring.datasource.dynamic.datasource.master")
    public DataSource masterDataSource(){
        LOGGER.info("加载主数据源masterDataSource.");
        return DruidDataSourceBuilder.create().build();
    }
    /**
     * 数据库从库
     */
    @Bean
    @ConfigurationProperties("spring.datasource.dynamic.datasource.slave1")
    public DataSource slave1DataSource(){
        LOGGER.info("加载从数据源slave1DataSource.");
        return DruidDataSourceBuilder.create().build();
    }

    /**
     * 动态数据源
     */
    @Bean
    public DataSource myRoutingDataSource(@Qualifier("masterDataSource") DataSource masterDataSource,
                                          @Qualifier("slave1DataSource") DataSource slave1DataSource) {
        LOGGER.info("加载[masterDataSource-slave1DataSource]设置为动态数据源DynamicDataSource.");
        Map<Object, Object> targetDataSources = new HashMap<>(2);
        targetDataSources.put(DBTypeEnum.MASTER, masterDataSource);
        targetDataSources.put(DBTypeEnum.SLAVE1, slave1DataSource);
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setDefaultTargetDataSource(masterDataSource);
        dynamicDataSource.setTargetDataSources(targetDataSources);
        return dynamicDataSource;
    }

    /**
     * 配置mybatis-plus的SqlSessionFactory
     */
    @Bean
    public MybatisSqlSessionFactoryBean sqlSessionFactory(@Qualifier("masterDataSource") DataSource master,
                                                          @Qualifier("slave1DataSource") DataSource slave) throws Exception {
        LOGGER.info("自定义配置mybatis-plus的SqlSessionFactory.");

        MybatisSqlSessionFactoryBean mybatisPlus = new MybatisSqlSessionFactoryBean();
        mybatisPlus.setDataSource(myRoutingDataSource(master, slave));

        MybatisConfiguration configuration = new MybatisConfiguration();
        configuration.setJdbcTypeForNull(JdbcType.NULL);
        configuration.setMapUnderscoreToCamelCase(true);
        configuration.setCacheEnabled(false);
        ///自定义配置
        mybatisPlus.setConfiguration(configuration);

        GlobalConfiguration globalConfiguration = GlobalConfigUtils.defaults();
        /////自定义填充策略接口实现
        globalConfiguration.setMetaObjectHandler(sysMetaObjectHandler);
        ////主键类型  0:"数据库ID自增", 1:"用户输入ID",2:"全局唯一ID (数字类型唯一ID)", 3:"全局唯一ID UUID";
        globalConfiguration.setIdType(0);
        /////字段策略 0:"忽略判断",1:"非 NULL 判断"),2:"非空判断"
        globalConfiguration.setFieldStrategy(2);
        ////驼峰下划线转换
        globalConfiguration.setDbColumnUnderline(true);
        ///自定义全局配置
        mybatisPlus.setGlobalConfig(globalConfiguration);


        if (StringUtils.hasLength(this.mybatisProperties.getTypeAliasesPackage())) {
            mybatisPlus.setTypeAliasesPackage(this.mybatisProperties.getTypeAliasesPackage());
        }
        if (StringUtils.hasLength(this.mybatisProperties.getTypeHandlersPackage())) {
            mybatisPlus.setTypeHandlersPackage(this.mybatisProperties.getTypeHandlersPackage());
        }
        if (!ObjectUtils.isEmpty(this.mybatisProperties.resolveMapperLocations())) {
            mybatisPlus.setMapperLocations(this.mybatisProperties.resolveMapperLocations());
        }
        //// 设置mapper.xml文件的路径
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        org.springframework.core.io.Resource[] resource = resolver.getResources("classpath:mapper/*.xml");
        mybatisPlus.setMapperLocations(resource);
        //添加插件到SqlSessionFactory才能生效
        mybatisPlus.setPlugins(new Interceptor[]{paginationInterceptor(),performanceInterceptor()});

        return mybatisPlus;
    }
}

4.2、数据源枚举

package com.unnet.yjs.config.db;

/**
 * Email: love1208tt@foxmail.com
 * Copyright (c)  2019. missbe
 * @author lyg   19-7-8 下午2:05
 *
 * 数据源类型
 **/

public enum DBTypeEnum {
    /**主库*/
    MASTER,
    /**从库1*/
    SLAVE1,
    /**从库2*/
    SLAVE2;
}

4.3、数据源线程安全设置key

package com.unnet.yjs.config.db;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * Email: love1208tt@foxmail.com
 * Copyright (c)  2019. missbe
 * @author lyg   19-7-8 下午1:00
 *
 * 线程安全的数据源切换
 **/

public class DbContextHolder {
    private static final Logger LOGGER = LoggerFactory.getLogger(DataSourceApp.class);

    private static final ThreadLocal<DBTypeEnum> CONTEXT_HOLDER = new ThreadLocal<>();

    private static final AtomicInteger COUNTER = new AtomicInteger(-1);

    public static void set(DBTypeEnum dbType) {
        CONTEXT_HOLDER.set(dbType);
    }

    public static DBTypeEnum get() {
        return CONTEXT_HOLDER.get();
    }

    static void master() {
        set(DBTypeEnum.MASTER);
        System.out.println("切换到master");
    }

    static void slave() {
        //  轮询
        int index = COUNTER.getAndIncrement() % 2;
        if (COUNTER.get() > 9999) {
            COUNTER.set(-1);
        }
        set(DBTypeEnum.SLAVE1);
        LOGGER.info(index +"->切换到slave1");
    }

    /**
     * 清除上下文数据
     */
    static void clearDbType() {
        CONTEXT_HOLDER.remove();
    }

}

4.4、AOP切面配置key切换

package com.unnet.yjs.config.db;

import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

/**
 * Email: love1208tt@foxmail.com
 * Copyright (c)  2019. missbe
 * @author lyg   19-7-8 下午2:04
 *
 * 配置数据源切换的AOP切面
 * com.baomidou.mybatisplus.service的select开头的方法走从库,insert、update、delete开头的方法走主库
 * com.unnet.yjs.service的select开头的方法走从库,insert、update、delete开头的方法走主库
 **/
@Aspect
@Component
public class DataSourceAop {
    private static final Logger LOGGER = LoggerFactory.getLogger(DataSourceAop.class);

    @Pointcut("!@annotation(com.unnet.yjs.annotation.Master) && " +
            "(execution(* com.unnet.yjs.service..*.select*(..)) || " +
            "execution(* com.unnet.yjs.service..*.get*(..)) || " +
            "execution(* com.baomidou.mybatisplus.service..*.select*(..)) || " +
            "execution(* com.unnet.yjs.service..*.find*(..)))")
    public void readPointCut(){  }

    @Pointcut("@annotation(com.unnet.yjs.annotation.Master) || execution(* com.unnet.yjs.service..*.insert*(..)) " +
            "|| execution(* com.unnet.yjs.service..*.update*(..)) || execution(* com.baomidou.mybatisplus.service..*.update*(..)) " +
            "|| execution(* com.unnet.yjs.service..*.delete*(..)) || execution(* com.baomidou.mybatisplus.service..*.delete*(..)) " +
            "|| execution(* com.baomidou.mybatisplus.service..*.insert*(..))")
    public void writePointcut() {    }

    @Before("readPointCut()")
    public void read(){
        LOGGER.info("------switch read-------");
        DbContextHolder.slave();
    }

    @Before("writePointcut()")
    public void write(){
        LOGGER.info("------switch write-------");
        DbContextHolder.master();
    }
}

4.5、数据key获取策略设置

package com.unnet.yjs.config.db;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

import javax.annotation.Nullable;

/**
 * Email: love1208tt@foxmail.com
 * Copyright (c)  2019. missbe
 * @author lyg   19-7-8 下午2:11
 *
 *
 **/
public class DynamicDataSource extends AbstractRoutingDataSource {
    @Nullable
    @Override
    protected Object determineCurrentLookupKey() {
        return DbContextHolder.get();
    }
}

4.6、Master注解

package com.unnet.yjs.annotation;
/**
 * Email: love1208tt@foxmail.com
 * Copyright (c)  2019. missbe
 * @author lyg   7/7/19 9:06 PM
 * 
 * 强制读主库切换
 **/

public @interface Master {
}

主要用于需要自己强制切换到主库的时候;

五、测试

  1. swagger测试这个手机号是否存在;


    image.png
  2. 日志如下:

切换到master
 Time:11 ms - ID:com.unnet.yjs.dao.AgentDao.selectList
 Execute SQL:
    SELECT
        id,
        create_by AS createId,
        create_date AS createDate,
        update_by AS updateId,
        update_date AS updateDate,
        del_flag AS delFlag,
        remarks  
    FROM
        t_agent   
    WHERE
        (
            phone_number = 'xxxxxxxxxxx'
        )

2019-07-08 14:51:27.557  INFO  : ------switch read-------
2019-07-08 14:51:27.557  INFO  : -1->切换到slave1
2019-07-08 14:51:27.561  INFO  : {dataSource-3} inited
 Time:39 ms - ID:com.unnet.yjs.dao.TerminatorDao.selectList
 Execute SQL:
    SELECT
         id,
        create_by AS createId,
        create_date AS createDate,
        update_by AS updateId,
        update_date AS updateDate,
        del_flag AS delFlag,
        remarks  
    FROM
        t_terminator   
    WHERE
        (
            terminator_machine_code = 'xx-xx-xx-xxx-xx'
        )

2019-07-08 14:51:28.005  INFO  : 15310663435:14102d47-8d22-42c0-ac79-20e047037a0c-j5s7yk->该帐号对应机器码为首次登录,返回营业厅数据信息.
2019-07-08 14:51:28.044  INFO  : ------switch read-------
2019-07-08 14:51:28.044  INFO  : 0->切换到slave1
 Time:35 ms - ID:com.unnet.yjs.dao.HallDao.selectList
 Execute SQL:
    SELECT
        id,
        create_by AS createId,
        create_date AS createDate,
        update_by AS updateId,
        update_date AS updateDate,
        del_flag AS delFlag,
        remarks  
    FROM
        t_hall   
    WHERE
        (
            agent_id = 'xx'
        )

六、参考

欢迎关注南阁公众号

南阁子也
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,098评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,213评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,960评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,519评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,512评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,533评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,914评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,574评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,804评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,563评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,644评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,350评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,933评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,908评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,146评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,847评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,361评论 2 342