SpringBoot使用jta+atomikos解决多数据源分布式事务

最新推荐文章于 2024-08-16 16:49:47 发布

不想当程序汪的第N天

最新推荐文章于 2024-08-16 16:49:47 发布

阅读量2k

点赞数 1

分类专栏： SpringBoot 文章标签： java mysql spring

本文链接：https://blog.csdn.net/qq_36525300/article/details/103466340

版权

SpringBoot 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

多数据源的两种情况

SpringBoot 中多数据源大致可以分为两种：

垂直分割模式
类似水平分割模式

类似水平分割模式，不同区域都拥有数据库，每个区域的数据库数据相同，通过不同区域的账户选择登录哪个数据库。

垂直分割模式

垂直分割模式，将整个数据库根据模块拆分成多个数据库，每个数据库是单独一个数据源，这样就涉及到多数据源的分布式事务问题。

application.properties

这里配置了两个数据库。

mybatis.type-aliases-package = com.yellow.etc.mapper
mybatis.mapper-location = classpath:mapping/*.xml

spring.datasource.test1.driver-class-name = com.mysql.jdbc.Driver
spring.datasource.test1.jdbc-url = jdbc:mysql://localhost:3306/test01?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8
spring.datasource.test1.username = root
spring.datasource.test1.password = 123456

spring.datasource.test2.driver-class-name = com.mysql.jdbc.Driver
spring.datasource.test2.jdbc-url = jdbc:mysql://localhost:3306/test02?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8
spring.datasource.test2.username = root
spring.datasource.test2.password = 123456

DataSourceConfig

通过 Java 配置，注册两个数据源，分别为 test1DataSource 与 test2DataSource。

注意！配置两个数据源，需要在主数据源上打@Primary注解，否则会报以下异常

org.springframework.beans.factory.NoUniqueBeanDefinitionException: No qualifying bean of type ‘org.springframework.transaction.TransactionManager’ available: expected single matching bean but found 2: test1TransactionManager,test2TransactionManager

import javax.sql.DataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
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 org.springframework.jdbc.datasource.DataSourceTransactionManager;

@Configuration
@MapperScan(basePackages = "com.yellow.etc.mapper.test01", sqlSessionFactoryRef = "test1SqlSessionFactory")
public class DataSource1Config {
	/**
	 * @methodDesc:功能描述:(配置test01数据库)
	 * @return
	 */
	@Bean(name = "test1DataSource")
	@ConfigurationProperties(prefix = "spring.datasource.test1")
	@Primary //没有该注解会报异常
	public DataSource testDataSource() {
		return DataSourceBuilder.create().build();
	}
	/**
	 * @methodDesc:功能描述:(test1 sql会话工厂)
	 * @param dataSource
	 * @return
	 * @throws Exception 
	 */
	@Bean(name = "test1SqlSessionFactory")
	@Primary
	public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) 
			throws Exception {
		SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
		bean.setDataSource(dataSource);
		return bean.getObject();
	}
	/**
	 * @methodDesc:功能描述:(test1 事务管理)
	 * @param dataSource
	 * @return
	 */
	@Bean(name = "test1TransactionManager")
	@Primary
	public DataSourceTransactionManager tesTransactionManager(@Qualifier("test1DataSource") DataSource dataSource) {
		return new DataSourceTransactionManager(dataSource);
	}
	/**
	 * @methodDesc:功能描述:()
	 * @param sqlSessionFactory
	 * @return
	 */
	@Bean(name = "test1SqlSessionTemplate")
	@Primary
	public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
		return new SqlSessionTemplate(sqlSessionFactory);
	}
}

import javax.sql.DataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
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 org.springframework.jdbc.datasource.DataSourceTransactionManager;

@Configuration
@MapperScan(basePackages = "com.yellow.etc.mapper.test02", sqlSessionFactoryRef = "test2SqlSessionFactory")
public class DataSource2Config {
	/**
	 * @methodDesc:功能描述:(配置test01数据库)
	 * @return
	 */
	@Bean(name = "test2DataSource")
	@ConfigurationProperties(prefix = "spring.datasource.test2")
	public DataSource testDataSource() {
		return DataSourceBuilder.create().build();
	}
	/**
	 * @methodDesc:功能描述:(test1 sql会话工厂)
	 * @param dataSource
	 * @return
	 * @throws Exception 
	 */
	@Bean(name = "test2SqlSessionFactory")
	public SqlSessionFactory testSqlSessionFactory(@Qualifier("test2DataSource") DataSource dataSource) 
			throws Exception {
		SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
		bean.setDataSource(dataSource);
		return bean.getObject();
	}
	/**
	 * @methodDesc:功能描述:(test1 事务管理)
	 * @param dataSource
	 * @return
	 */
	@Bean(name = "test2TransactionManager")
	public DataSourceTransactionManager tesTransactionManager(@Qualifier("test2DataSource") DataSource dataSource) {
		return new DataSourceTransactionManager(dataSource);
	}
	/**
	 * @methodDesc:功能描述:()
	 * @param sqlSessionFactory
	 * @return
	 */
	@Bean(name = "test2SqlSessionTemplate")
	public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test2SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
		return new SqlSessionTemplate(sqlSessionFactory);
	}
}

测试

编写两个 mapper 对应两个数据库，都是简单的插入方法，在 service 中调用这两个方法，手动添加一个除0异常，查看数据库状态。

@Transactional
public int addUser(String name, Integer age) {
	userMapper01.insert(name, age);
	userMapper02.insert(name, age);
	int i = 1/0;
	return 2;
}

调用方法后发现，使用 test01 作为主数据源，当发生异常回滚时，只会回滚该数据源对应的数据库数据，对 test02 数据不会进行回滚操作。这就不符合事务的原子性，要么同时成功，要么同时失败。

解决方法

这里我们可以使用 jta+atomikos 处理多数据源事务问题。

jta+atomikos 相当于把多个数据源的事务管理在一起，进行简单的代理，帮助我们在事务提交或回滚时进行操作。

pom.xml

在 pom.xml 中添加以下依赖。

<!-- jta+atomic -->
<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-jta-atomikos</artifactId>
</dependency>

application.properties

mybatis.type-aliases-package = com.yellow.etc.mapper
mybatis.mapper-location = classpath:mapping/*.xml

spring.datasource.test1.driver-class-name = com.mysql.jdbc.Driver
spring.datasource.test1.jdbc-url = jdbc:mysql://localhost:3306/test01?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8
spring.datasource.test1.username = root
spring.datasource.test1.password = 123456
spring.datasource.test1.minPoolSize = 3
spring.datasource.test1.maxPoolSize = 25
spring.datasource.test1.maxLifetime = 20000
spring.datasource.test1.borrowConnectionTimeout = 30
spring.datasource.test1.loginTimeout = 30
spring.datasource.test1.maintenanceInterval = 60
spring.datasource.test1.maxIdleTime = 60

spring.datasource.test2.driver-class-name = com.mysql.jdbc.Driver
spring.datasource.test2.jdbc-url = jdbc:mysql://localhost:3306/test02?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8
spring.datasource.test2.username = root
spring.datasource.test2.password = 123456
spring.datasource.test2.minPoolSize = 3
spring.datasource.test2.maxPoolSize = 25
spring.datasource.test2.maxLifetime = 20000
spring.datasource.test2.borrowConnectionTimeout = 30
spring.datasource.test2.loginTimeout = 30
spring.datasource.test2.maintenanceInterval = 60

配置类

import org.springframework.boot.context.properties.ConfigurationProperties;
import lombok.Data;

@Data
@ConfigurationProperties(prefix = "spring.datasource.test1")
public class DBConfig01 {
	private String url;
	private String username;
	private String password;
	private int minPoolSize;
	private int maxPoolSize;
	private int maxLifetime;
	private int borrowConnectionTimeout;
	private int loginTimeout;
	private int maintenanceInterval;
	private int maxIdleTime;
	private String testQuery;
}

import org.springframework.boot.context.properties.ConfigurationProperties;
import lombok.Data;

@Data
@ConfigurationProperties(prefix = "spring.datasource.test2")
public class DBConfig02 {
	private String url;
	private String username;
	private String password;
	private int minPoolSize;
	private int maxPoolSize;
	private int maxLifetime;
	private int borrowConnectionTimeout;
	private int loginTimeout;
	private int maintenanceInterval;
	private int maxIdleTime;
	private String testQuery;
}

配置数据源

import javax.sql.DataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.jta.atomikos.AtomikosDataSourceBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
/**
  * 类名称：JTAConfiguration01      
  * 创建人：98   
  * 创建时间：2019年12月18日 上午11:15:11   
 * @version
 */
@Configuration
@MapperScan(basePackages = "com.yellow.etc.mapper.test01", sqlSessionTemplateRef = "test1SqlSessionTemplate")
public class JTAConfiguration01 {
	
	/**
	 * @methodDesc:功能描述:(配置数据源)
	 * @param testConfig
	 * @return
	 */
	@Primary
	@Bean(name = "test1DataSource")
	public DataSource testDataSource(DBConfig01 testConfig) {
		MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
		mysqlXaDataSource.setUrl(testConfig.getUrl());
		mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
		mysqlXaDataSource.setPassword(testConfig.getPassword());
		mysqlXaDataSource.setUser(testConfig.getUsername());
		mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
		// 将本地事务注册到Atomikos
		AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
		xaDataSource.setXaDataSource(mysqlXaDataSource);
		xaDataSource.setUniqueResourceName("test1DataSource");
		// 配置信息
		xaDataSource.setMinPoolSize(testConfig.getMinPoolSize());
		xaDataSource.setMaxPoolSize(testConfig.getMaxPoolSize());
		xaDataSource.setMaxLifetime(testConfig.getMaxLifetime());
		xaDataSource.setBorrowConnectionTimeout(testConfig.getBorrowConnectionTimeout());
		xaDataSource.setLoginTimeout(testConfig.getLoginTimeout());
		xaDataSource.setMaintenanceInterval(testConfig.getMaintenanceInterval());
		xaDataSource.setMaxIdleTime(testConfig.getMaxIdleTime());
		xaDataSource.setTestQuery(testConfig.getTestQuery());
		return xaDataSource;
	}
	
	@Bean(name = "test1SqlSessionFactory")
	public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) throws Exception {
		SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
		bean.setDataSource(dataSource);
		return bean.getObject();
	}
 
	@Bean(name = "test1SqlSessionTemplate")
	public SqlSessionTemplate testSqlSessionTemplate(
			@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
		return new SqlSessionTemplate(sqlSessionFactory);
	}
}

import javax.sql.DataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.jta.atomikos.AtomikosDataSourceBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
/**
  * 类名称：JTAConfiguration02      
  * 创建人：98   
  * 创建时间：2019年12月18日 上午11:14:36   
 * @version
 */
@Configuration
@MapperScan(basePackages = "com.yellow.etc.mapper.test02", sqlSessionTemplateRef = "test2SqlSessionTemplate")
public class JTAConfiguration02 {
	/**
	 * @methodDesc:功能描述:(配置数据源)
	 * @param testConfig
	 * @return
	 */
	@Primary
	@Bean(name = "test2DataSource")
	public DataSource testDataSource(DBConfig02 testConfig) {
		MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
		mysqlXaDataSource.setUrl(testConfig.getUrl());
		mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
		mysqlXaDataSource.setPassword(testConfig.getPassword());
		mysqlXaDataSource.setUser(testConfig.getUsername());
		mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
		// 将本地事务注册到Atomikos
		AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
		xaDataSource.setXaDataSource(mysqlXaDataSource);
		xaDataSource.setUniqueResourceName("test2DataSource");
		// 配置信息
		xaDataSource.setMinPoolSize(testConfig.getMinPoolSize());
		xaDataSource.setMaxPoolSize(testConfig.getMaxPoolSize());
		xaDataSource.setMaxLifetime(testConfig.getMaxLifetime());
		xaDataSource.setBorrowConnectionTimeout(testConfig.getBorrowConnectionTimeout());
		xaDataSource.setLoginTimeout(testConfig.getLoginTimeout());
		xaDataSource.setMaintenanceInterval(testConfig.getMaintenanceInterval());
		xaDataSource.setMaxIdleTime(testConfig.getMaxIdleTime());
		xaDataSource.setTestQuery(testConfig.getTestQuery());
		return xaDataSource;
	}
	
	@Bean(name = "test2SqlSessionFactory")
	public SqlSessionFactory testSqlSessionFactory(@Qualifier("test2DataSource") DataSource dataSource) throws Exception {
		SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
		bean.setDataSource(dataSource);
		return bean.getObject();
	}
 
	@Bean(name = "test2SqlSessionTemplate")
	public SqlSessionTemplate testSqlSessionTemplate(
			@Qualifier("test2SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
		return new SqlSessionTemplate(sqlSessionFactory);
	}
}

启动类

@SpringBootApplication
// 加载配置类
@EnableConfigurationProperties(value = {DBConfig01.class, DBConfig02.class})
public class SpringBootDataSourceApplication {
	public static void main(String[] args) {
		SpringApplication.run(SpringBootDataSourceApplication.class, args);
	}
}