springboot+mybatisplus+druid如何实现多数据源+分布式事务

这篇文章主要介绍springboot+mybatisplus+druid如何实现多数据源+分布式事务，文中介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们一定要看完！

成都创新互联服务项目包括浉河网站建设、浉河网站制作、浉河网页制作以及浉河网络营销策划等。多年来，我们专注于互联网行业，利用自身积累的技术优势、行业经验、深度合作伙伴关系等，向广大中小型企业、政府机构等提供互联网行业的解决方案，浉河网站推广取得了明显的社会效益与经济效益。目前，我们服务的客户以成都为中心已经辐射到浉河省份的部分城市，未来相信会继续扩大服务区域并继续获得客户的支持与信任！

jdk环境：1.8

springboot：2.1.3.RELEASE

mybatisplus：3.2.0

本文主要用atomikos的AtomikosDataSourceBean+druid配置连接池，另一个是AtomikosNonXADataSourceBean

1、maven 需要注意MySQL-connector-java版本为6.x，超过这个版本可能会报异常

com.baomidou

mybatis-plus-boot-starter

3.2.0

org.springframework.boot

spring-boot-starter-jta-atomikos

mysql

mysql-connector-java

6.0.6

org.springframework.boot

spring-boot-starter-aop

org.projectlombok

lombok

true

org.springframework.boot

spring-boot-configuration-processor

true

com.alibaba

druid-spring-boot-starter

1.1.10

2、新建DataSourceContextHolder.java 提供设置、获取、清除当前数据源的方法;

public class DataSourceContextHolder {

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

/**

* 设置数据源

*

* @param db

*/

public static void setDataSource(String db) {

contextHolder.set(db);

}

/**

* 取得当前数据源

*

* @return

*/

public static String getDataSource() {

return contextHolder.get();

}

/**

* 清除上下文数据

*/

public static void clear() {

contextHolder.remove();

}

}

3、DataSource.java 数据源注解 及DataSourceKeyEnum.java 数据源枚举类;并提供获取枚举的方法

import java.lang.annotation.*;

@Target({ElementType.METHOD, ElementType.TYPE})

@Retention(RetentionPolicy.RUNTIME)

@Documented

public @interface DataSource {

DataSourceKeyEnum value();

}

import lombok.Getter;

import org.apache.commons.lang3.StringUtils;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

public enum DataSourceKeyEnum {

MASTER("master"),

/**

* 表示 所有的SLAVE, 随机选择一个SLAVE0， 或者SLAVE1

*/

SLAVE("slave"),

SLAVE0("slave0"),

SLAVE1("slave1"),;

@Getter

private String value;

DataSourceKeyEnum(String value) {

this.value = value;

}

public static List getSlaveList() {

return Arrays.asList(SLAVE0, SLAVE1);

}

/**

* 根据方法名称选择

*

* @param name 方法名称

*/

public static DataSourceKeyEnum getDSKeyByMethodName(String name) {

if (StringUtils.isEmpty(name)) {

return null;

}

if (name.contains("update") || name.contains("delete") || name.contains("remove") || name.contains("insert")) {

return MASTER;

}

if (name.contains("select") || name.contains("query") || name.contains("find") || name.contains("get")) {

List list = getSlaveList();

Collections.shuffle(list);

return list.get(0);

}

return MASTER;

}

/**

* 根据注解获取数据源

*

* @param dataSource

* @return

*/

public static DataSourceKeyEnum getDataSourceKey(DataSource dataSource) {

if (dataSource == null) {

return MASTER;

}

if (dataSource.value() == DataSourceKeyEnum.SLAVE) {

List dataSourceKeyList = DataSourceKeyEnum.getSlaveList();

// FIXME 目前乱序

Collections.shuffle(dataSourceKeyList);

return dataSourceKeyList.get(0);

} else {

return dataSource.value();

}

}

/**

* 根据className和Method 获取数据源枚举

*

* @param className

* @return

*/

public static String getByClassName(String className, Method method) {

//方法上的注解

DataSource dataSource = AnnotationUtils.findAnnotation(method, DataSource.class);

DataSourceKeyEnum keyEnum;

if (dataSource != null) {//注解存在则以注解为主

keyEnum = DataSourceKeyEnum.getDataSourceKey(dataSource);

} else {

keyEnum = DataSourceKeyEnum.getDSKeyByMethodName(method.getName());

}

return keyEnum.getValue();

}

}

4、DataSourceAspect.java 数据源切面类，这里是只做了对mapper层拦截，包括拦截了mybatisplus的公共BaseMapper

import lombok.extern.slf4j.Slf4j;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.Around;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Pointcut;

import org.aspectj.lang.reflect.MethodSignature;

import org.springframework.aop.support.AopUtils;

import org.springframework.context.annotation.Lazy;

import org.springframework.core.annotation.Order;

import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

import java.lang.reflect.Type;

@Component

@Slf4j

@Aspect

@Order(-1)

public class DataSourceAspect {

@Pointcut("execution(* com.admin.*.dao.*Mapper.*(..))||execution(* com.baomidou.mybatisplus.core.mapper.*Mapper.*(..)))")

public void pointCut() {

}

@Around("pointCut()")

public Object doBefore(ProceedingJoinPoint pjp) throws Throwable {

MethodSignature signature = (MethodSignature) pjp.getSignature();

Method method = signature.getMethod();

//拦截到的公共BaseMapper里的接口可通过该方式获得具体实现类的信息

Type[] types = AopUtils.getTargetClass(pjp.getTarget()).getGenericInterfaces(); // getGenericInterfaces方法能够获取类/接口实现的所有接口

String name = types[0].getTypeName();

String dataSource = DataSourceKeyEnum.getByClassName(name, method);

log.info("选择的数据源："+dataSource);

DataSourceContextHolder.setDataSource(dataSource);

Object o=pjp.proceed();

DataSourceContextHolder.clear();

return o;

}

}

5、application.yml 相关配置

spring:

datasource:

druid:

master:

xaDataSourceClassName: com.alibaba.druid.pool.xa.DruidXADataSource

uniqueResourceName: master

xaDataSource:

url: jdbc:mysql://127.0.0.1:3306/test?charset=utf8mb4&useUnicode=true&characterEncoding=utf8&useSSL=false&autoReconnect=true&serverTimezone=Asia/Shanghai

username: root

password: 123456

driver-class-name: com.mysql.cj.jdbc.Driver

initial-size: 5

min-idle: 5

max-active: 20

#获取连接等待超时时间

max-wait: 60000

#间隔多久进行一次检测，检测需要关闭的空闲连接

time-between-eviction-runs-millis: 60000

#一个连接在池中最小生存的时间

min-evictable-idle-time-millis: 300000

#指定获取连接时连接校验的sql查询语句

validation-query: SELECT 'x'

#验证连接的有效性

test-while-idle: true

#获取连接时候验证，会影响性能(不建议true)

test-on-borrow: false

#打开PSCache，并指定每个连接上PSCache的大小。oracle设为true，mysql设为false。分库分表较多推荐设置为false

pool-prepared-statements: false

max-pool-prepared-statement-per-connection-size: 20

# 配置监控统计拦截的filters，去掉后监控界面sql无法统计，'wall'用于防火墙

filters: config,wall,stat

slave0:

xaDataSourceClassName: com.alibaba.druid.pool.xa.DruidXADataSource

uniqueResourceName: slave0

xaDataSource:

url: jdbc:mysql://127.0.0.1:3306/test?charset=utf8mb4&useUnicode=true&characterEncoding=utf8&useSSL=false&autoReconnect=true&serverTimezone=Asia/Shanghai

username: root

password: 123456

driver-class-name: com.mysql.cj.jdbc.Driver

initial-size: 5

min-idle: 5

max-active: 20

#获取连接等待超时时间

max-wait: 60000

#间隔多久进行一次检测，检测需要关闭的空闲连接

time-between-eviction-runs-millis: 60000

#一个连接在池中最小生存的时间

min-evictable-idle-time-millis: 300000

#指定获取连接时连接校验的sql查询语句

validation-query: SELECT 'x'

#验证连接的有效性

test-while-idle: true

#获取连接时候验证，会影响性能(不建议true)

test-on-borrow: false

#打开PSCache，并指定每个连接上PSCache的大小。oracle设为true，mysql设为false。分库分表较多推荐设置为false

pool-prepared-statements: false

max-pool-prepared-statement-per-connection-size: 20

# 配置监控统计拦截的filters，去掉后监控界面sql无法统计，'wall'用于防火墙

filters: config,wall,stat

slave1:

xaDataSourceClassName: com.alibaba.druid.pool.xa.DruidXADataSource

uniqueResourceName: slave1

xaDataSource:

url: jdbc:mysql://127.0.0.1:3306/test?charset=utf8mb4&useUnicode=true&characterEncoding=utf8&useSSL=false&autoReconnect=true&serverTimezone=Asia/Shanghai

username: root

password: 123456

driver-class-name: com.mysql.cj.jdbc.Driver

initial-size: 5

min-idle: 5

max-active: 20

#获取连接等待超时时间

max-wait: 60000

#间隔多久进行一次检测，检测需要关闭的空闲连接

time-between-eviction-runs-millis: 60000

#一个连接在池中最小生存的时间

min-evictable-idle-time-millis: 300000

#指定获取连接时连接校验的sql查询语句

validation-query: SELECT 'x'

#验证连接的有效性

test-while-idle: true

#获取连接时候验证，会影响性能(不建议true)

test-on-borrow: false

#打开PSCache，并指定每个连接上PSCache的大小。oracle设为true，mysql设为false。分库分表较多推荐设置为false

pool-prepared-statements: false

max-pool-prepared-statement-per-connection-size: 20

# 配置监控统计拦截的filters，去掉后监控界面sql无法统计，'wall'用于防火墙

filters: config,wall,stat

web-stat-filter:

enabled: true

url-pattern: /*

exclusions: /druid/*,*.js,*.gif,*.jpg,*.bmp,*.png,*.css,*.ico

session-stat-enable: true

session-stat-max-count: 10

stat-view-servlet:

enabled: true

url-pattern: /druid/*

reset-enable: true

login-username: admin

login-password: admin

jta:

atomikos:

properties:

log-base-dir: ../logs

transaction-manager-id: txManager #默认取计算机的IP地址 需保证生产环境值唯一

6、复制SqlSessionTemplate.java里所有代码新建到MySqlSessionTemplate.java然后继承SqlSessionTemplate.java，并按照下方替换相应的方法(只贴出了更改了的地方)

import lombok.Getter;

import lombok.Setter;

public class MySqlSessionTemplate extends SqlSessionTemplate {

@Getter

@Setter

private Map targetSqlSessionFactories;

@Getter

@Setter

private SqlSessionFactory defaultTargetSqlSessionFactory;

public MySqlSessionTemplate(SqlSessionFactory sqlSessionFactory, ExecutorType executorType,

PersistenceExceptionTranslator exceptionTranslator) {

super(sqlSessionFactory, executorType, exceptionTranslator);

notNull(sqlSessionFactory, "Property 'sqlSessionFactory' is required");

notNull(executorType, "Property 'executorType' is required");

this.sqlSessionFactory = sqlSessionFactory;

this.executorType = executorType;

this.exceptionTranslator = exceptionTranslator;

this.sqlSessionProxy = (SqlSession) newProxyInstance(SqlSessionFactory.class.getClassLoader(),

new Class[]{SqlSession.class}, new MySqlSessionTemplate.SqlSessionInterceptor());

this.defaultTargetSqlSessionFactory = sqlSessionFactory;

}

//TODO 主要修改了这一块，并且用到sqlSessionFactory的地方都改调用该方法获取

public SqlSessionFactory getSqlSessionFactory() {

SqlSessionFactory targetSqlSessionFactory = targetSqlSessionFactories.get(DataSourceContextHolder.getDataSource());

if (targetSqlSessionFactory != null) {

return targetSqlSessionFactory;

} else if (defaultTargetSqlSessionFactory != null) {

return defaultTargetSqlSessionFactory;

} else {

Assert.notNull(targetSqlSessionFactories, "Property 'targetSqlSessionFactories' or 'defaultTargetSqlSessionFactory' are required");

}

return this.sqlSessionFactory;

}

/**

* {@inheritDoc}

*/

@Override

public Configuration getConfiguration() {

return this.getSqlSessionFactory().getConfiguration();

}

private class SqlSessionInterceptor implements InvocationHandler {

@Override

public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

SqlSession sqlSession = getSqlSession(MySqlSessionTemplate.this.getSqlSessionFactory(),

MySqlSessionTemplate.this.executorType, MySqlSessionTemplate.this.exceptionTranslator);

try {无锡人流哪家好 http://www.wxbhffk.com/

Object result = method.invoke(sqlSession, args);

if (!isSqlSessionTransactional(sqlSession, MySqlSessionTemplate.this.getSqlSessionFactory())) {

// force commit even on non-dirty sessions because some databases require

// a commit/rollback before calling close()

sqlSession.commit(true);

}

return result;

} catch (Throwable t) {

Throwable unwrapped = unwrapThrowable(t);

if (MySqlSessionTemplate.this.exceptionTranslator != null && unwrapped instanceof PersistenceException) {

// release the connection to avoid a deadlock if the translator is no loaded. See issue #22

closeSqlSession(sqlSession, MySqlSessionTemplate.this.sqlSessionFactory);

sqlSession = null;

Throwable translated = MySqlSessionTemplate.this.exceptionTranslator

.translateExceptionIfPossible((PersistenceException) unwrapped);

if (translated != null) {

unwrapped = translated;

}

}

throw unwrapped;

} finally {

if (sqlSession != null) {

closeSqlSession(sqlSession, MySqlSessionTemplate.this.getSqlSessionFactory());

}

}

}

}

}

7、新建MyBatisPlusConfiguration.java，mybatisplus配置 和多数据源配置

import com.alibaba.druid.pool.xa.DruidXADataSource;

import org.springframework.boot.jta.atomikos.AtomikosDataSourceBean;

import com.baomidou.mybatisplus.autoconfigure.SpringBootVFS;

import com.baomidou.mybatisplus.core.MybatisConfiguration;

import com.baomidou.mybatisplus.core.parser.ISqlParser;

import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;

import com.baomidou.mybatisplus.extension.plugins.tenant.TenantHandler;

import com.baomidou.mybatisplus.extension.plugins.tenant.TenantSqlParser;

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

import com.admin.util.datasource.DataSourceKeyEnum;

import net.sf.jsqlparser.expression.Expression;

import net.sf.jsqlparser.expression.LongValue;

import org.apache.ibatis.mapping.BoundSql;

import org.apache.ibatis.session.SqlSessionFactory;

import org.apache.ibatis.type.JdbcType;

import org.mybatis.spring.annotation.MapperScan;

import org.springframework.beans.factory.annotation.Value;

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 org.springframework.context.annotation.Primary;

import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

@Configuration

@MapperScan(basePackages = {"com.admin.*.dao", "com.baomidou.mybatisplus.samples.quickstart.mapper"}, sqlSessionTemplateRef = "sqlSessionTemplate")

public class MyBatisPlusConfiguration {

@Bean

@Primary //多数据源时需要加上该注解，加一个即可

@ConfigurationProperties(prefix = "spring.datasource.druid.master")

public AtomikosDataSourceBean userMaster() {

return new AtomikosDataSourceBean();

}

@Bean

@ConfigurationProperties(prefix = "spring.datasource.druid.slave0")

public AtomikosDataSourceBean userSlave0() {

return new AtomikosDataSourceBean();

}

@Bean

@ConfigurationProperties(prefix = "spring.datasource.druid.slave1")

public AtomikosDataSourceBean userSlave1() {

return new AtomikosDataSourceBean();

}

@Bean(name = "sqlSessionTemplate")

public MySqlSessionTemplate customSqlSessionTemplate() throws Exception {

Map sqlSessionFactoryMap = new HashMap() {{

put(DataSourceKeyEnum.MASTER.getValue(), createSqlSessionFactory(userMaster()));

put(DataSourceKeyEnum.SLAVE0.getValue(), createSqlSessionFactory(userSlave0()));

put(DataSourceKeyEnum.SLAVE1.getValue(), createSqlSessionFactory(userSlave1()));

}};

MySqlSessionTemplate sqlSessionTemplate = new MySqlSessionTemplate(sqlSessionFactoryMap.get(DataSourceKeyEnum.MASTER.getValue()));

sqlSessionTemplate.setTargetSqlSessionFactories(sqlSessionFactoryMap);

return sqlSessionTemplate;

}

/**

* 创建数据源

*

* @param dataSource

* @return

*/

private SqlSessionFactory createSqlSessionFactory(AtomikosDataSourceBean dataSource) throws Exception {

dataSource.setMaxPoolSize(10);

dataSource.setMinPoolSize(2);

dataSource.setPoolSize(2);

dataSource.setMaxIdleTime(60);//最大闲置时间，超过最小连接池的连接将关闭

dataSource.setMaxLifetime(1200);//连接最大闲置时间 单位s 全部的连接超时将关闭

dataSource.setTestQuery(druidDataSource.getValidationQuery());//前期先每次请求前都执行该操作保证连接有效，后期可用定时任务执行

dataSource.setMaintenanceInterval(60);//定时维护线程周期 单位秒

//以上配置可提取到.yml内通过ConfigurationProperties注解注入

dataSource.init();//项目启动则初始化连接

MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean();

sqlSessionFactory.setDataSource(dataSource);

sqlSessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:/com/admin/*/dao/xml/*.xml"));

sqlSessionFactory.setVfs(SpringBootVFS.class);

MybatisConfiguration configuration = new MybatisConfiguration();

//configuration.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);

configuration.setJdbcTypeForNull(JdbcType.NULL);

configuration.setMapUnderscoreToCamelCase(false);

configuration.setCacheEnabled(false);

sqlSessionFactory.setConfiguration(configuration);

sqlSessionFactory.setPlugins(paginationInterceptor());

sqlSessionFactory.afterPropertiesSet();

return sqlSessionFactory.getObject();

}

/*

* 自定义的分页插件，自动识别数据库类型

*/

@Bean

public PaginationInterceptor paginationInterceptor() {

return new PaginationInterceptor();

}

}

事务使用方法如单数据源一样：对应的方法或类上加@Transactional注解即可

以上是“springboot+mybatisplus+druid如何实现多数据源+分布式事务”这篇文章的所有内容，感谢各位的阅读！希望分享的内容对大家有帮助，更多相关知识，欢迎关注创新互联行业资讯频道！