Springboot 整合jta-atomikos實現多數據源及分佈式事務管理——SpringBoot學習

一、POM 文件引入 JAR

  在 pom.xml 中引入 jta-atomikos 以及測試包

<!-- mybatis 依賴 -->
<dependency>
	<groupId>org.mybatis.spring.boot</groupId>
	<artifactId>mybatis-spring-boot-starter</artifactId>
	<version>1.1.1</version>
</dependency>

<!-- atomikos 多數據源管理 -->
<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-jta-atomikos</artifactId>
</dependency>

<!-- mysql 依賴 -->
<dependency>
	<groupId>mysql</groupId>
	<artifactId>mysql-connector-java</artifactId>
</dependency>

<!-- 測試 -->
<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-test</artifactId>
	<scope>test</scope>
	<exclusions>
		<exclusion>
			<groupId>org.springframework</groupId>
			<artifactId>spring-test</artifactId>
		</exclusion>
	</exclusions>
</dependency>

<!-- SpringBoot 2.0 的 spring-boot-starter-test 中的spring-test 是5.0.4 版本的,
		而,5.0.4 版沒有 SpringJUnit4ClassRunner 和 SpringRunner 所以去除,另加入5.0.5 及以上版本的spring-test-->
<dependency>
	<groupId>org.springframework</groupId>
	<artifactId>spring-test</artifactId>
	<version>5.0.5.RELEASE</version>
	<scope>test</scope>
</dependency>

二、application.properties 新增配置

  在 application.properties 中添加兩個數據源的配置

mysql.datasource.test1.url = jdbc:mysql://localhost:3306/mysql
mysql.datasource.test1.username = root
mysql.datasource.test1.password = 123456
mysql.datasource.test1.minPoolSize = 3
mysql.datasource.test1.maxPoolSize = 25
mysql.datasource.test1.maxLifetime = 20000
mysql.datasource.test1.borrowConnectionTimeout = 30
mysql.datasource.test1.loginTimeout = 30
mysql.datasource.test1.maintenanceInterval = 60
mysql.datasource.test1.maxIdleTime = 60

mysql.datasource.test2.url = jdbc:mysql://localhost:3306/mybatis
mysql.datasource.test2.username =root
mysql.datasource.test2.password = 123456
mysql.datasource.test2.minPoolSize = 3
mysql.datasource.test2.maxPoolSize = 25
mysql.datasource.test2.maxLifetime = 20000
mysql.datasource.test2.borrowConnectionTimeout = 30
mysql.datasource.test2.loginTimeout = 30
mysql.datasource.test2.maintenanceInterval = 60
mysql.datasource.test2.maxIdleTime = 60


使用 application.yml 的配置爲

### two datasource Configuration
mysql: 
  datasource:
    test1: 
      url: jdbc:mysql://localhost:3306/mysql
      driver-class-name: com.mysql.jdbc.Driver
      username: root
      password: 123456
      minPoolSize: 3
      maxPoolSize: 25
      maxLifetime: 20000
      borrowConnectionTimeout: 30
      loginTimeout: 30
      maintenanceInterval: 60
      maxIdleTime: 60
    test2: 
      url: jdbc:mysql://localhost:3306/mybatis
      driver-class-name: com.mysql.jdbc.Driver
      username: root
      password: 123456
      minPoolSize: 3
      maxPoolSize: 25
      maxLifetime: 20000
      borrowConnectionTimeout: 30
      loginTimeout: 30
      maintenanceInterval: 60
      maxIdleTime: 60
  

三、編寫 Dao

  添加兩個 bean 文件 src/main/java/com/bean/MybatisUser.javasrc/main/java/com/bean/MysqlUser.java

src/main/java/com/bean/MybatisUser.java

package com.bean;

import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

/**
* @Description 類描述
* @author 歐陽
* @since 2019年4月7日 下午12:29:13
* @version V1.0
*/

@Getter
@Setter
@ToString
public class MybatisUser {
	private String id;
	private String name;
	
	public MybatisUser() {
		super();
	}
	
	public MybatisUser(String id, String name) {
		super();
		this.id = id;
		this.name = name;
	}	
}

src/main/java/com/bean/MysqlUser.java

package com.bean;

import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

/**
* @Description 類描述
* @author 歐陽
* @since 2019年4月7日 下午12:29:13
* @version V1.0
*/

@Getter
@Setter
@ToString
public class MysqlUser {
	private String id;
	private String name;
	
	public MysqlUser() {
		super();
	}
	
	public MysqlUser(String id, String name) {
		super();
		this.id = id;
		this.name = name;
	}
}

  根據庫的區別,創建兩個包 com.mysqlcom.mybatis ,分別在包下創建 Mapper 。

  在 com.mysql 包下創建 src/main/java/com/mysql/UserMapper2.javasrc/main/java/com/mysql/userMapper2.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.mysql.UserMapper2">

	<insert id="insertUser" parameterType="com.bean.MysqlUser">
		insert into t_user(id, name) values(#{id}, #{name})
	</insert>

	<select id="selectUsersAll" resultType="com.bean.MysqlUser">
		select id,name from t_user
	</select>

	<update id="updateUser" parameterType="com.bean.MysqlUser">
		update t_user set name=#{name} where id=#{id}
	</update>

	<delete id="deleteUserById" parameterType="string">
		delete from t_user where id = #{id}
	</delete>
</mapper>
package com.mysql;

import java.util.List;

import com.bean.MysqlUser;

/**
* @Description springboot 整合jta-atomikos 數據源2
* @author 歐陽
* @since 2019年4月11日 下午4:26:00
* @version V1.0
*/

public interface UserMapper2 {
	/**
	 * 添加用戶
	 * @param user
	 */
	public void insertUser(MysqlUser mysqlUser);
	
	/**
	 * 更新用戶
	 * @param user
	 */
	public void updateUser(MysqlUser mysqlUser);
	
	/**
	 * 查詢用戶
	 * @param user
	 * @return
	 */
	public List<MysqlUser> selectUsersAll();
	
	/**
	 * 刪除用戶
	 * @param id
	 */
	public void deleteUserById(String id);
}

  在 com.mysql 包下創建 src/main/java/com/mybatis/UserMapper1.javasrc/main/java/com/mybatis/userMapper1.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.mybatis.UserMapper1">

	<insert id="insertUser" parameterType="com.bean.MybatisUser">
		insert into t_user(id, name) values(#{id}, #{name})
	</insert>

	<select id="selectUsersAll" resultType="com.bean.MybatisUser">
		select id,name from t_user
	</select>

	<update id="updateUser" parameterType="com.bean.MybatisUser">
		update t_user set name=#{name} where id=#{id}
	</update>

	<delete id="deleteUserById" parameterType="string">
		delete from t_user where id = #{id}
	</delete>
</mapper>
package com.mybatis;

import java.util.List;

import com.bean.MybatisUser;

/**
* @Description springboot 整合jta-atomikos 數據源1
* @author 歐陽
* @since 2019年4月11日 下午4:26:00
* @version V1.0
*/

public interface UserMapper1 {
	/**
	 * 添加用戶
	 * @param user
	 */
	public void insertUser(MybatisUser mybatisUser);
	
	/**
	 * 更新用戶
	 * @param user
	 */
	public void updateUser(MybatisUser mybatisUser);
	
	/**
	 * 查詢用戶
	 * @param user
	 * @return
	 */
	public List<MybatisUser> selectUsersAll();
	
	/**
	 * 刪除用戶
	 * @param id
	 */
	public void deleteUserById(String id);
}

四、配置數據源

package com.config;

import java.sql.SQLException;

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.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Component;

import com.atomikos.jdbc.AtomikosDataSourceBean;
import com.mysql.jdbc.jdbc2.optional.MysqlXADataSource;

import lombok.Data;

/**
* @Description 配置多數據源
* @author 歐陽
* @since 2019年6月15日 下午5:16:47
* @version V1.0
*/
@Configuration
public class MultiDBConfig {
	
	@Bean
	@ConfigurationProperties(prefix = "mysql.datasource.test1")
	public DBConfig1 dbConfig1() {
		return new DBConfig1();
	}
	
	@Bean
	@ConfigurationProperties(prefix = "mysql.datasource.test2")
	public DBConfig2 dbConfig2() {
		return new DBConfig2();
	}
	
	@Component
	@MapperScan(basePackages = {"com.mysql", "com.bean"}, sqlSessionTemplateRef = "db1SqlSessionTemplate")
	public class MyBatisConfig1 {
		// 配置數據源
		@Primary
		@Bean(name = "db1DataSource")
		public DataSource db1DataSource(DBConfig1 dbConfig1) throws SQLException {
			MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
			mysqlXaDataSource.setUrl(dbConfig1.getUrl());
			mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
			mysqlXaDataSource.setPassword(dbConfig1.getPassword());
			mysqlXaDataSource.setUser(dbConfig1.getUsername());
			mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);

			AtomikosDataSourceBean db1DataSource = new AtomikosDataSourceBean();
			db1DataSource.setXaDataSource(mysqlXaDataSource);
			db1DataSource.setUniqueResourceName("db1DataSource");

			db1DataSource.setMinPoolSize(dbConfig1.getMinPoolSize());
			db1DataSource.setMaxPoolSize(dbConfig1.getMaxPoolSize());
			db1DataSource.setMaxLifetime(dbConfig1.getMaxLifetime());
			db1DataSource.setBorrowConnectionTimeout(dbConfig1.getBorrowConnectionTimeout());
			db1DataSource.setLoginTimeout(dbConfig1.getLoginTimeout());
			db1DataSource.setMaintenanceInterval(dbConfig1.getMaintenanceInterval());
			db1DataSource.setMaxIdleTime(dbConfig1.getMaxIdleTime());
			db1DataSource.setTestQuery(dbConfig1.getTestQuery());
			return db1DataSource;
		}
		
		@Primary
		@Bean(name = "db1SqlSessionFactory")
		public SqlSessionFactory db1SqlSessionFactory(@Qualifier("db1DataSource") DataSource db1DataSource)
				throws Exception {
			SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
			bean.setDataSource(db1DataSource);
			return bean.getObject();
		}
		
		@Primary
		@Bean(name = "db1SqlSessionTemplate")
		public SqlSessionTemplate db1SqlSessionTemplate(@Qualifier("db1SqlSessionFactory") SqlSessionFactory db1SqlSessionFactory) throws Exception {
			return new SqlSessionTemplate(db1SqlSessionFactory);
		}
	}
	
	@Component
	@MapperScan(basePackages = {"com.mybatis", "com.bean"}, sqlSessionTemplateRef = "db2SqlSessionTemplate")
	public class MyBatisConfig2 {
		// 配置數據源
		@Bean(name = "db2DataSource")
		public DataSource db2DataSource(DBConfig2 dbConfig2) throws SQLException {
			MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
			mysqlXaDataSource.setUrl(dbConfig2.getUrl());
			mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
			mysqlXaDataSource.setPassword(dbConfig2.getPassword());
			mysqlXaDataSource.setUser(dbConfig2.getUsername());
			mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);

			AtomikosDataSourceBean db2DataSource = new AtomikosDataSourceBean();
			db2DataSource.setXaDataSource(mysqlXaDataSource);
			db2DataSource.setUniqueResourceName("db2DataSource");

			db2DataSource.setMinPoolSize(dbConfig2.getMinPoolSize());
			db2DataSource.setMaxPoolSize(dbConfig2.getMaxPoolSize());
			db2DataSource.setMaxLifetime(dbConfig2.getMaxLifetime());
			db2DataSource.setBorrowConnectionTimeout(dbConfig2.getBorrowConnectionTimeout());
			db2DataSource.setLoginTimeout(dbConfig2.getLoginTimeout());
			db2DataSource.setMaintenanceInterval(dbConfig2.getMaintenanceInterval());
			db2DataSource.setMaxIdleTime(dbConfig2.getMaxIdleTime());
			db2DataSource.setTestQuery(dbConfig2.getTestQuery());
			return db2DataSource;
		}
		
		@Bean(name = "db2SqlSessionFactory")
		public SqlSessionFactory db2SqlSessionFactory(@Qualifier("db2DataSource") DataSource db2DataSource)
				throws Exception {
			SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
			bean.setDataSource(db2DataSource);
			return bean.getObject();
		}
		
		@Bean(name = "db2SqlSessionTemplate")
		public SqlSessionTemplate db2SqlSessionTemplate(@Qualifier("db2SqlSessionFactory") SqlSessionFactory db2SqlSessionFactory) throws Exception {
			return new SqlSessionTemplate(db2SqlSessionFactory);
		}
	}
	
	
	
	@Data
	private class DBConfig1 {
		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;
	}
	
	@Data
	private class DBConfig2 {
		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;
	}
	

}

五、添加 Service

  添加一個統一的 Service 接口 src/main/java/com/service/MoreDateSourceService.java 並實現該接口,實現類 src/main/java/com/service/impl/MoreDateSourceServiceImpl.java 統一調用兩個數據源,測試整合 jta-atomikos 實現分佈式事務管理的正確性,以向兩個數據庫中添加用戶爲例。
MoreDateSourceService 接口代碼

package com.service;

import com.bean.MybatisUser;
import com.bean.MysqlUser;

/**
* @Description 多數據源事務管理 Service
* @author 歐陽
* @since 2019年4月11日 下午8:34:50
* @version V1.0
*/

public interface MoreDateSourceService {
	public void addUser(MybatisUser mybatisUser, MysqlUser mysqlUser);
}

MoreDateSourceServiceImpl 實現類

package com.service.impl;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.bean.MybatisUser;
import com.bean.MysqlUser;
import com.mybatis.UserMapper1;
import com.mysql.UserMapper2;
import com.service.MoreDateSourceService;

/**
* @Description 多數據源事務管理 Service
* @author 歐陽
* @since 2019年4月11日 下午8:35:42
* @version V1.0
*/
@Service
public class MoreDateSourceServiceImpl implements MoreDateSourceService {
	
	@Autowired
	private UserMapper2 userMapper2;
	
	@Autowired
	private UserMapper1 userMapper1;
	
	@Override
	@Transactional
	public void addUser(MybatisUser mybatisUser, MysqlUser mysqlUser) {
		
		userMapper1.insertUser(mybatisUser);
		userMapper2.insertUser(mysqlUser);
		
//		int num = 1/0;  //放開測試事務。預期結果爲兩個庫都添加失敗,庫中無數據。測試結果與預期一致
	}

}

  注意:通過使用 @Transactional 註解添加事務管理。第一種情況,在 addUser 方法中的插入數據代碼後增加異常代碼 int num = 1/0; ,測試事務是否起作用,如果不起作用,則兩個庫中都有剛插入的數據,反之兩個庫中都沒有剛插入的數據;第二種情況,註釋異常代碼 int num = 1/0; ,數據會正常插入到兩個庫中。測試後實際測試結果表明測試結果與預期一致,整合 jta-atomikos 後可通過使用 @Transactional 註解可管理分佈式事務。

六、測試類

  添加測試類測試 MoreDateSourceServiceImplTest

package com.service.impl;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import com.App;
import com.bean.MybatisUser;
import com.bean.MysqlUser;
import com.service.MoreDateSourceService;

/**
* @Description 多數據源事務管理 測試
* @author 歐陽
* @since 2019年4月11日 下午8:45:21
* @version V1.0
*/

@RunWith(SpringRunner.class)
@SpringBootTest(classes={App.class})
public class MoreDateSourceServiceImplTest {

	@Autowired
	private MoreDateSourceService moreDateSourceService;
	
	@Test
	public void testAddUser() {
		MybatisUser mybatisUser = new MybatisUser("12", "張三");
		MysqlUser mysqlUser = new MysqlUser("12", "張三");
		
		moreDateSourceService.addUser(mybatisUser, mysqlUser);
		System.out.println("Yes");
	}

}

七、測試結果

  測試結果通過,通過整合 jta-atomikos 實現分佈式事務管理功能正常,springboot 整合 jta-atomikos 完成。

測試結果成功

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章