SpirngBoot整合Mybatis Plus多數據源

導讀

　　有一個這樣子的需求，線上正在跑的業務，由於業務發展需要，需重新開發一套新系統，等新系統開發完成后，需要無縫對接切換，當初具體設計見草圖。

 

添加依賴

        <!--lombok-->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.16</version>
            <!--scope=provided，說明它是在編譯階段生效，不需要打入包中，Lombok在編譯期將帶Lombok注解的Java文件正確編譯為完整的Class文件-->
            <scope>provided</scope>
        </dependency>
        <!--mysql-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        <!--mybatis plus和spring boot整合-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.4.0</version>
        </dependency>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.1.10</version>
        </dependency>

application.properties

server.port=9999
spring.datasource.master.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.master.url=jdbc:mysql://127.0.0.1/test?useUnicode=true&characterEncoding=utf-8&useSSL=false
spring.datasource.master.username=root
spring.datasource.master.password=root

spring.datasource.slave.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.slave.jdbc-url=jdbc:mysql://127.0.0.1/test2?useUnicode=true&characterEncoding=utf-8&useSSL=false
spring.datasource.slave.username=root
spring.datasource.slave.password=root

2種方式創建DataSource

Master配置，使用druid連接池

import com.alibaba.druid.pool.DruidDataSource;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.env.Environment;

import javax.sql.DataSource;

/**
 * @Author：chenyanbin
 */
@Configuration
@MapperScan(basePackages = "com.example.demo.mapper", sqlSessionFactoryRef = "masterSqlSessionFactory")
public class MasterDataSourceConfig {
    @Autowired
    private Environment env;

    @Primary
    @Bean(name = "masterDataSource")
//    @ConfigurationProperties("spring.datasource.master")
    public DataSource masterDataSource() {
//        return DataSourceBuilder.create().build();
        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setUrl(env.getProperty("spring.datasource.master.url"));
        dataSource.setUsername(env.getProperty("spring.datasource.master.username"));
        dataSource.setPassword(env.getProperty("spring.datasource.master.password"));
        dataSource.setDriverClassName(env.getProperty("spring.datasource.master.driver-class-name"));
        //配置初始化大小、最小、最大
        dataSource.setMinIdle(10);
        //配置初始化大小、最小、最大
        dataSource.setMaxActive(200);
        //配置初始化大小、最小、最大
        dataSource.setInitialSize(10);
        //配置獲取連接等待超時的時間
        dataSource.setMaxWait(60000);
        //配置一個連接在池中最小生存的時間,單位是毫秒
        dataSource.setMinEvictableIdleTimeMillis(300000);
        //配置間隔多久才進行一次檢測,檢測需要關閉的空閑連接,單位是毫秒
        dataSource.setTimeBetweenEvictionRunsMillis(60000);
        //默認的testWhileIdle=true,testOnBorrow=false,testOnReturn=false
        dataSource.setValidationQuery("SELECT 1");
        //申請連接時執行validationQuery檢測連接是否有效
        dataSource.setTestOnBorrow(false);
        //建議配置為true，不影響性能，並且保證安全性。
        dataSource.setTestWhileIdle(true);
        //是否緩存preparedStatement，也就是PSCache
        dataSource.setPoolPreparedStatements(false);
        return dataSource;
    }

    @Bean(name = "masterSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("masterDataSource") DataSource dataSource) throws Exception {
        MybatisSqlSessionFactoryBean sessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);

        return sessionFactoryBean.getObject();
    }
}

Slave配置

import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.annotation.MapperScan;
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;

/**
 * @Author：chenyanbin
 */
@Configuration
@MapperScan(basePackages = "com.example.demo.mapper2",sqlSessionFactoryRef = "slaveSqlSessionFactory")
public class SlaveDataSourceConfig {
    @Bean(name = "slaveDataSource")
    @ConfigurationProperties("spring.datasource.slave")
    public DataSource slaveDataSource(){
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "slaveSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("slaveDataSource") DataSource dataSource) throws Exception {
        MybatisSqlSessionFactoryBean sessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        return sessionFactoryBean.getObject();
    }
}

注意

　　master和slave掃描不同的mapper包路徑！！！！！！

　　如果需要指定.xml文件，需這樣配置！！！

    @Bean(name = "masterSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("masterDataSource") DataSource dataSource) throws Exception {
        MybatisSqlSessionFactoryBean sessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        sessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath:mapper/**/*.xml"));
        return sessionFactoryBean.getObject();
    }

MybatisPlus分頁插件設置

import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * MybatisPlus分頁配置
 *
 * @Author：chenyanbin
 */
@Configuration
public class MybatisPlusPageConfig {

    /**
     * 新版分頁插件
     *
     * @return
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor mybatisPlusInterceptor = new MybatisPlusInterceptor();
        mybatisPlusInterceptor.addInnerInterceptor(new PaginationInnerInterceptor());
        return mybatisPlusInterceptor;
    }
}

    @Bean(name = "masterSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(
            @Qualifier("masterDataSource") DataSource dataSource,
            @Qualifier("mybatisPlusInterceptor") MybatisPlusInterceptor mybatisPlusInterceptor
    ) throws Exception {
        MybatisSqlSessionFactoryBean sessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        sessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath:mapper/**/*.xml"));
        sessionFactoryBean.setPlugins(mybatisPlusInterceptor);
        return sessionFactoryBean.getObject();
    }

設置填充字段

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

/**
 * @Author：chenyanbin
 */
@Component
public class CustomMetaObjectHandler implements MetaObjectHandler {
    @Override
    public void insertFill(MetaObject metaObject) {
        //屬性名稱，不是字段名稱
        this.setFieldValByName("createTime", CommonUtil.getCurrentDate(), metaObject);
        this.setFieldValByName("createUserId", CommonUtil.getCurrentUserId(), metaObject);
        this.setFieldValByName("createUserName", CommonUtil.getCurrentUserName(), metaObject);
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        //屬性名稱，不是字段名稱
        this.setFieldValByName("updateTime", CommonUtil.getCurrentDate(), metaObject);
        this.setFieldValByName("updateUserId", CommonUtil.getCurrentUserId(), metaObject);
        this.setFieldValByName("updateUserName", CommonUtil.getCurrentUserName(), metaObject);
    }
}

/**
 * <p>
 * 角色信息表
 * </p>
 *
 * @author chenyanbin
 * @since 2021-09-28
 */
@Data
@TableName("sys_role")
@ApiModel(value = "SysRoleDO對象", description = "角色信息表")
public class RoleDO implements Serializable {

    private static final long serialVersionUID = 1L;

    @ApiModelProperty(value = "角色ID")
    @TableId(value = "id", type = IdType.AUTO)
    private Long id;

    @ApiModelProperty(value = "角色名稱")
    private String roleName;

    @ApiModelProperty(value = "角色狀態（0正常 1停用）")
    private byte status;

    @ApiModelProperty(value = "備注")
    private String remark;

    @ApiModelProperty(value = "創建時間")
    @TableField(value = "create_time", fill = FieldFill.INSERT)
    private Date createTime;

    @ApiModelProperty(value = "創建人id")
    @TableField(value = "create_user_id", fill = FieldFill.INSERT)
    private Long createUserId;

    @ApiModelProperty(value = "創建人姓名")
    @TableField(value = "create_user_name", fill = FieldFill.INSERT)
    private String createUserName;

    @ApiModelProperty(value = "修改時間")
    @TableField(value = "update_time", fill = FieldFill.UPDATE)
    private Date updateTime;

    @ApiModelProperty(value = "修改人id")
    @TableField(value = "update_user_id", fill = FieldFill.UPDATE)
    private Long updateUserId;

    @ApiModelProperty(value = "修改人姓名")
    @TableField(value = "update_user_name", fill = FieldFill.UPDATE)
    private String updateUserName;


}

    import com.baomidou.mybatisplus.core.config.GlobalConfig;



    @Bean(name = "masterSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(
            @Qualifier("masterDataSource") DataSource dataSource,
            @Qualifier("mybatisPlusInterceptor") MybatisPlusInterceptor mybatisPlusInterceptor,
            @Qualifier("customMetaObjectHandler") CustomMetaObjectHandler customMetaObjectHandler
    ) throws Exception {
        MybatisSqlSessionFactoryBean sessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        sessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath:mapper/**/*.xml"));
        //mybatis plus分頁插件
        sessionFactoryBean.setPlugins(mybatisPlusInterceptor);
        //自動填充字段
        GlobalConfig globalConfig=new GlobalConfig();
        globalConfig.setMetaObjectHandler(customMetaObjectHandler);
        sessionFactoryBean.setGlobalConfig(globalConfig);
        return sessionFactoryBean.getObject();
    }

多數據源事務問題

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import javax.sql.DataSource;

/**
 * 事務
 *
 * @Author：chenyanbin
 */
@Configuration
public class CustomTransactionHandler {
    @Bean(name = "masterTransaction")
    public DataSourceTransactionManager masterTransaction(
            @Qualifier("masterDataSource") DataSource dataSource
    ) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "historyTransaction")
    public DataSourceTransactionManager historyTransaction(
            @Qualifier("historyDataSource") DataSource dataSource
    ) {
        return new DataSourceTransactionManager(dataSource);
    }
}

指定使用那個數據源的事務

@Transactional(transactionManager = "masterTransaction")
public void remove(Long id) {}

 

 

啟動類

@SpringBootApplication(
        exclude = {DataSourceAutoConfiguration.class, MybatisPlusAutoConfiguration.class}
)

啟動類上排查，自動裝配，使用我們自定義的多數據源！！！

演示

　　多個數據源，同時也是支持事務的