springboot整合mybatis-plus多数据源配置

青灯
649 阅读2分钟

一、修改数据源配置

连接池的配置可以依旧沿用单数据源的情况,此处只简单写出数据源配置

spring:
  datasource:
    # MySql
    first:
      jdbc-url: jdbc:mysql://localhost:3306/project_first?useSSL=false&useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&transformedBitIsBoolean=true&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true&allowPublicKeyRetrieval=true
      username: root
      password: root
    second:
      jdbc-url: jdbc:mysql://localhost:3306/project_second?useSSL=false&useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&transformedBitIsBoolean=true&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true&allowPublicKeyRetrieval=true
      username: root
      password: root

二、添加数据源配置类

注意:

1 在扫描xml文件时,要保证两个数据源扫描的路径不一致,否则在使用mybatis-plus进行批量插入时可能会出现问题.

2 使用多数据源配置后,mybatis-plus自动配置的分页和自动填充都会失效, 需要自己去添加相应的配置

1. 主数据源配置

import javax.sql.DataSource;

import com.baomidou.mybatisplus.core.config.GlobalConfig;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springeos.common.handler.EosMetaObjectHandler;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.jdbc.DataSourceBuilder;
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.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;

@Configuration
@MapperScan(basePackages = "com.almond.more-datasource.*.mapper", sqlSessionTemplateRef  = "firstSqlSessionTemplate")
public class DataSourceFirst {

	@Bean(name = "firstDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.first")
    @Primary
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "firstSqlSessionFactory")
    @Primary
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("firstDataSource") DataSource dataSource, @Qualifier("firstPaginationInterceptor") PaginationInterceptor paginationInterceptor) throws Exception {
        // 此处是mybatis-plus的配置
        // mybatis的配置则应是SqlSessionFactoryBean
    	MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:firstMapper/**/*Mapper.xml"));
		// 分页支持
		Interceptor[] plugins = new Interceptor[]{paginationInterceptor};
		bean.setPlugins(plugins);
		//全局配置
		GlobalConfig globalConfig  = new GlobalConfig();
		//配置填充器
		globalConfig.setMetaObjectHandler(new EosMetaObjectHandler());
		bean.setGlobalConfig(globalConfig);
		return bean.getObject();
    }

	@Bean("firstPaginationInterceptor")
	@Primary
	public PaginationInterceptor paginationInterceptor() {
		PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
		paginationInterceptor.setLimit(-1);
		return paginationInterceptor;
	}

    @Bean(name = "firstTransactionManager")
    @Primary
    public DataSourceTransactionManager testTransactionManager(@Qualifier("firstDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "firstSqlSessionTemplate")
    @Primary
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("firstSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

2. 其他数据源和主数据源配置同理

import javax.sql.DataSource;

import com.baomidou.mybatisplus.core.config.GlobalConfig;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springeos.common.handler.EosMetaObjectHandler;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;

@Configuration
@MapperScan(basePackages = "org.springeos.modules.eos.mapper", sqlSessionTemplateRef  = "secondSqlSessionTemplate")
public class DataSourceSecond {

	@Bean(name = "secondDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.second")
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondSqlSessionFactory")
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("secondDataSource") DataSource dataSource, @Qualifier("secondPaginationInterceptor") PaginationInterceptor paginationInterceptor) throws Exception {
    	MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:org/springeos/**/eos/mapper/*Mapper.xml"));
		// 分页支持
		Interceptor[] plugins = new Interceptor[]{paginationInterceptor};
		bean.setPlugins(plugins);
		//全局配置
		GlobalConfig globalConfig  = new GlobalConfig();
		//配置填充器
        // EosMetaObjectHandler是自定义的配置类
		globalConfig.setMetaObjectHandler(new EosMetaObjectHandler());
		bean.setGlobalConfig(globalConfig);
        return bean.getObject();
    }

	@Bean("secondPaginationInterceptor")
	public PaginationInterceptor paginationInterceptor() {
		PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
		paginationInterceptor.setLimit(-1);
		return paginationInterceptor;
	}

    @Bean(name = "secondTransactionManager")
    public DataSourceTransactionManager testTransactionManager(@Qualifier("secondDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "secondSqlSessionTemplate")
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("secondSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

三、其他相关类

下方类中使用的相关实体类请自行替换

/**
 * @Classname EosMetaObjectHandler
 * @Description mybatis-plus自定填充处理
 * @Date 2020/9/10 8:58
 * @Created by hongxing
 */
@Component
public class EosMetaObjectHandler implements MetaObjectHandler {
	/**
	 * 添加时填充
	 * @param metaObject
	 */
	@Override
	public void insertFill(MetaObject metaObject) {
		EosUser user = AuthUtil.getUser();
		Date now = new Date();
		this.strictInsertFill(metaObject, "createTime", Date.class, now);
		this.strictInsertFill(metaObject, "createUser", Long.class, user.getUserId());
		this.strictUpdateFill(metaObject, "updateTime", Date.class, now);
		this.strictUpdateFill(metaObject, "updateUser", Long.class, user.getUserId());
		this.strictInsertFill(metaObject, "isDeleted", Integer.class, 0);
		this.strictInsertFill(metaObject, "isDeleted", Integer.class, EosConstant.DB_NOT_DELETED);
	}

	/**
	 * 修改时填充
	 * @param metaObject
	 */
	@Override
	public void updateFill(MetaObject metaObject) {
		EosUser user = AuthUtil.getUser();
		Date now = new Date();
		this.strictUpdateFill(metaObject, "updateTime", Date.class, now);
		this.strictUpdateFill(metaObject, "updateUser", Long.class, user.getUserId());
	}
}

拜拜~~ 越努力越幸运~

avatar
文章
阅读
粉丝