在一個Spring Boot專案中，連線多個數據源還是比較常見的。之前也介紹瞭如何在幾種常用框架的場景下配置多資料來源，具體可見： - [Spring Boot 2.x基礎教程：JdbcTemplate的多資料來源配置](http://blog.didispace.com/spring-boot-learning-21-3-7/) - [Spring Boot 2.x基礎教程：Spring Data JPA的多資料來源配置](http://blog.didispace.com/spring-boot-learning-21-3-8/) - [Spring Boot 2.x基礎教程：MyBatis的多資料來源配置](http://blog.didispace.com/spring-boot-learning-21-3-9/) 當我們採用多資料來源的時候，同時也會出現一個這樣的特殊場景：我們希望對A資料來源的更新和B資料來源的更新具備事務性。這樣的例子很常見，比如：在訂單庫中建立一條訂單記錄，同時還需要在商品庫中扣減商品庫存。如果庫存扣減失敗，那麼我們希望訂單建立也能夠回滾。 如果這兩條資料在一個數據庫中，那麼通過之前介紹的[事務管理](http://blog.didispace.com/spring-boot-learning-21-3-10/)就能輕鬆解決了。但是，當這兩個操作位於不同的資料庫中，那麼就無法實現了。 本文就來介紹一種解決這類問題的方法：JTA事務。 ## 什麼是JTA JTA，全稱：Java Transaction API。JTA事務比JDBC事務更強大。一個JTA事務可以有多個參與者，而一個JDBC事務則被限定在一個單一的資料庫連線。所以，當我們在同時操作多個數據庫的時候，使用JTA事務就可以彌補JDBC事務的不足。 在Spring Boot 2.x中，整合了這兩個JTA的實現： - Atomikos：可以通過引入`spring-boot-starter-jta-atomikos`依賴來使用 - Bitronix：可以通過引入`spring-boot-starter-jta-bitronix`依賴來使用 由於Bitronix自Spring Boot 2.3.0開始不推薦使用，所以在下面的動手環節中，我們將使用Atomikos作為例子來介紹JTA的使用。 ## 動手試試 下面我們就來實操一下，如何在Spring Boot中使用JTA來實現多資料來源下的事務管理。 **準備工作** 1. 這裡我們將使用最基礎的JdbcTemplate來實現資料訪問，所以如果你還不會使用JdbcTemplate配置多資料來源，建議先看一下[JdbcTemplate的多資料來源配置](http://blog.didispace.com/spring-boot-learning-21-3-7/)。 2. 場景設定： - 假設我們有兩個庫，分別為：test1和test2 - 這兩個庫中都有一張User表，我們希望這兩張表中的資料是一致的 - 假設這兩張表中都已經有一條資料：name=aaa，age=30；因為這兩張表中資料是一致的，所以要update的時候，就必須兩個庫中的User表更新時候，要麼都成功，要麼都失敗。  > 本文首發：[https://blog.didispace.com/spring-boot-learning-24-3-12/](https://blog.didispace.com/spring-boot-learning-24-3-12/) ，後期修改更新主要以原文為主。 **操作詳細** 1. 在`pom.xml`中加入JTA的實現Atomikos的Starter ```xml ``` 2. 在`application.properties`配置檔案中配置兩個test1和test2資料來源 ```properties spring.jta.enabled=true spring.jta.atomikos.datasource.primary.xa-properties.url=jdbc:mysql://localhost:3306/test1 spring.jta.atomikos.datasource.primary.xa-properties.user=root spring.jta.atomikos.datasource.primary.xa-properties.password=12345678 spring.jta.atomikos.datasource.primary.xa-data-source-class-name=com.mysql.cj.jdbc.MysqlXADataSource spring.jta.atomikos.datasource.primary.unique-resource-name=test1 spring.jta.atomikos.datasource.primary.max-pool-size=25 spring.jta.atomikos.datasource.primary.min-pool-size=3 spring.jta.atomikos.datasource.primary.max-lifetime=20000 spring.jta.atomikos.datasource.primary.borrow-connection-timeout=10000 spring.jta.atomikos.datasource.secondary.xa-properties.url=jdbc:mysql://localhost:3306/test2 spring.jta.atomikos.datasource.secondary.xa-properties.user=root spring.jta.atomikos.datasource.secondary.xa-properties.password=12345678 spring.jta.atomikos.datasource.secondary.xa-data-source-class-name=com.mysql.cj.jdbc.MysqlXADataSource spring.jta.atomikos.datasource.secondary.unique-resource-name=test2 spring.jta.atomikos.datasource.secondary.max-pool-size=25 spring.jta.atomikos.datasource.secondary.min-pool-size=3 spring.jta.atomikos.datasource.secondary.max-lifetime=20000 spring.jta.atomikos.datasource.secondary.borrow-connection-timeout=10000 ``` 3. 建立多資料來源配置類 ```java @Configuration public class DataSourceConfiguration { @Primary @Bean @ConfigurationProperties(prefix = "spring.jta.atomikos.datasource.primary") public DataSource primaryDataSource() { return new AtomikosDataSourceBean(); } @Bean @ConfigurationProperties(prefix = "spring.jta.atomikos.datasource.secondary") public DataSource secondaryDataSource() { return new AtomikosDataSourceBean(); } @Bean public JdbcTemplate primaryJdbcTemplate(@Qualifier("primaryDataSource") DataSource primaryDataSource) { return new JdbcTemplate(primaryDataSource); } @Bean public JdbcTemplate secondaryJdbcTemplate(@Qualifier("secondaryDataSource") DataSource secondaryDataSource) { return new JdbcTemplate(secondaryDataSource); } } ``` 注意，這裡除了家在的配置不同之外，`DataSource`也採用了`AtomikosDataSourceBean`注意與之前配置多資料來源使用的配置和實現類的區別。 4. 建立一個Service實現，模擬兩種不同的情況。 ```java @Service public class TestService { private JdbcTemplate primaryJdbcTemplate; private JdbcTemplate secondaryJdbcTemplate; public TestService(JdbcTemplate primaryJdbcTemplate, JdbcTemplate secondaryJdbcTemplate) { this.primaryJdbcTemplate = primaryJdbcTemplate; this.secondaryJdbcTemplate = secondaryJdbcTemplate; } @Transactional public void tx() { // 修改test1庫中的資料 primaryJdbcTemplate.update("update user set age = ? where name = ?", 30, "aaa"); // 修改test2庫中的資料 secondaryJdbcTemplate.update("update user set age = ? where name = ?", 30, "aaa"); } @Transactional public void tx2() { // 修改test1庫中的資料 primaryJdbcTemplate.update("update user set age = ? where name = ?", 40, "aaa"); // 模擬：修改test2庫之前丟擲異常 throw new RuntimeException(); } } ``` 這裡tx函式，是兩句update操作，一般都會成功；而tx2函式中，我們人為的製造了一個異常，這個異常是在test1庫中的資料更新後才產生的，這樣就可以測試一下test1更新成功，之後是否還能在JTA的幫助下實現回滾。 5. 建立測試類，編寫測試用例 ```java @SpringBootTest(classes = Chapter312Application.class) public class Chapter312ApplicationTests { @Autowired protected JdbcTemplate primaryJdbcTemplate; @Autowired protected JdbcTemplate secondaryJdbcTemplate; @Autowired private TestService testService; @Test public void test1() throws Exception { // 正確更新的情況 testService.tx(); Assertions.assertEquals(30, primaryJdbcTemplate.queryForObject("select age from user where name=?", Integer.class, "aaa")); Assertions.assertEquals(30, secondaryJdbcTemplate.queryForObject("select age from user where name=?", Integer.class, "aaa")); } @Test public void test2() throws Exception { // 更新失敗的情況 try { testService.tx2(); } catch (Exception e) { e.printStackTrace(); } finally { // 部分更新失敗，test1中的更新應該回滾 Assertions.assertEquals(30, primaryJdbcTemplate.queryForObject("select age from user where name=?", Integer.class, "aaa")); Assertions.assertEquals(30, secondaryJdbcTemplate.queryForObject("select age from user where name=?", Integer.class, "aaa")); } } } ``` 這裡有兩個測試用例： - test1：因為沒有故意製造的異常，不出意外兩個庫的update都會成功，所以根據name=aaa去把兩個資料查出來，看age是否都被更新到了30。 - test2：tx2函式會把test1中name=aaa的使用者age更新為40，然後丟擲異常，JTA事務生效的話，會把age回滾回30，所以這裡的檢查也是兩個庫的aaa使用者的age應該都為30，這樣就意味著JTA事務生效，保證了test1和test2兩個庫中的User表資料更新一致，沒有製造出髒資料。 ## 測試驗證 將上面編寫的單元測試執行起來:  觀察一下啟動階段的日誌，可以看到這些Atomikos初始化日誌輸出： ``` 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.default_max_wait_time_on_shutdown = 9223372036854775807 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.allow_subtransactions = true 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.recovery_delay = 10000 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.automatic_resource_registration = true 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.oltp_max_retries = 5 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.client_demarcation = false 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.threaded_2pc = false 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.serial_jta_transactions = true 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.log_base_dir = /Users/didi/Documents/GitHub/SpringBoot-Learning/2.x/chapter3-12/transaction-logs 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.rmi_export_class = none 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.max_actives = 50 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.checkpoint_interval = 500 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.enable_logging = true 2021-02-02 19:00:36.145 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.log_base_name = tmlog 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.max_timeout = 300000 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.trust_client_tm = false 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: java.naming.factory.initial = com.sun.jndi.rmi.registry.RegistryContextFactory 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.tm_unique_name = 127.0.0.1.tm 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.forget_orphaned_log_entries_delay = 86400000 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.oltp_retry_interval = 10000 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: java.naming.provider.url = rmi://localhost:1099 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.force_shutdown_on_vm_exit = false 2021-02-02 19:00:36.146 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : USING: com.atomikos.icatch.default_jta_timeout = 10000 2021-02-02 19:00:36.147 INFO 8868 --- [ main] c.a.icatch.provider.imp.AssemblerImp : Using default (local) logging and recovery... 2021-02-02 19:00:36.184 INFO 8868 --- [ main] c.a.d.xa.XATransactionalResource : test1: refreshed XAResource 2021-02-02 19:00:36.203 INFO 8868 --- [ main] c.a.d.xa.XATransactionalResource : test2: refreshed XAResource ``` 同時，我們在`transaction-logs`目錄下，還能找到關於事務的日誌資訊：  ```json {"id":"127.0.0.1.tm161226409083100001","wasCommitted":true,"participants":[{"uri":"127.0.0.1.tm1","state":"COMMITTING","expires":1612264100801,"resourceName":"test1"},{"uri":"127.0.0.1.tm2","state":"COMMITTING","expires":1612264100801,"resourceName":"test2"}]} {"id":"127.0.0.1.tm161226409083100001","wasCommitted":true,"participants":[{"uri":"127.0.0.1.tm1","state":"TERMINATED","expires":1612264100804,"resourceName":"test1"},{"uri":"127.0.0.1.tm2","state":"TERMINATED","expires":1612264100804,"resourceName":"test2"}]} {"id":"127.0.0.1.tm161226409092800002","wasCommitted":false,"participants":[{"uri":"127.0.0.1.tm3","state":"TERMINATED","expires":1612264100832,"resourceName":"test1"}]} ``` **更多本系列免費教程連載[「點選進入彙總目錄」](http://blog.didispace.com/spring-boot-learning-2x/)** ## 程式碼示例 本文的相關例子可以檢視下面倉庫中的`chapter3-12`目錄： - Github：[https://github.com/dyc87112/SpringBoot-Learning/](https://github.com/dyc87112/SpringBoot-Learning/tree/master/2.x) - Gitee：[https://gitee.com/didispace/SpringBoot-Learning/](https://gitee.com/didispace/SpringBoot-Learning/tree/master/2.x) **如果您覺得本文不錯，歡迎`Star`支援，您的關注是我堅持的動力！** > 歡迎關注我的公眾號：程式猿DD，獲得獨家整理的免費學習資源助力你的Java學習之路！另每週贈書不