【Transaction】jta-atomikos + Springboot JPA 多数据源事务回滚

依赖

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

配置

# 多数据源配置 spring: datasource: ds1: driverClassName: org.postgresql.Driver jdbcUrl: jdbc:postgresql://xxx/bigdata?currentSchema=public username: xxx password: xxx ds3: driverClassName: org.postgresql.Driver jdbcUrl: jdbc:postgresql://xxx/xxx?currentSchema=public username: xxxx password: xxxx jpa: hibernate: dialect: org.hibernate.dialect.PostgreSQLDialect open-in-view: true show-sql: true database-platform: org.hibernate.dialect.PostgreSQLDialect

配置类

@Configuration @ConfigurationProperties(prefix = "spring.datasource.ds2") @Data public class GPProperties { private String jdbcUrl; private String username; private String password; private String driverClassName; } @Configuration @ConfigurationProperties(prefix = "spring.datasource.ds1") @Data public class PGProperties { private String jdbcUrl; private String username; private String password; private String driverClassName; } @Configuration @EnableTransactionManagement @DependsOn("transactionManager") @EnableJpaRepositories( entityManagerFactoryRef = "entityManagerPrimary", transactionManagerRef = "transactionManager", basePackages = {"com.kanq.bigdata.greenplum.dao.ds1"}) //设置Repository所在位置 public class PrimaryConfig { @Autowired private JpaVendorAdapter jpaVendorAdapter; @Autowired @Qualifier("primaryDataSource") private DataSource primaryDataSource; @Bean(name = "entityManagerPrimary") public LocalContainerEntityManagerFactoryBean orderEntityManager() throws Throwable { HashMap<String, Object> properties = new HashMap<String, Object>(); properties.put("hibernate.transaction.jta.platform", AtomikosJtaPlatform.class.getName()); properties.put("javax.persistence.transactionType", "JTA"); LocalContainerEntityManagerFactoryBean entityManager = new LocalContainerEntityManagerFactoryBean(); entityManager.setJtaDataSource(primaryDataSource); entityManager.setJpaVendorAdapter(jpaVendorAdapter); entityManager.setPackagesToScan("com.kanq.bigdata.greenplum.entity.pg"); entityManager.setPersistenceUnitName("layerPersistenceUnit"); entityManager.setJpaPropertyMap(properties); return entityManager; } } @Configuration @DependsOn("transactionManager") @EnableTransactionManagement @EnableJpaRepositories( entityManagerFactoryRef = "entityManagerSecond", transactionManagerRef = "transactionManager", basePackages = {"com.kanq.bigdata.greenplum.dao.ds2"}) //设置Repository所在位置 public class SecondConfig { @Autowired private JpaVendorAdapter jpaVendorAdapter; @Autowired @Qualifier("secondDataSource") private DataSource secondDataSource; @Bean(name = "entityManagerSecond") public LocalContainerEntityManagerFactoryBean orderEntityManager() throws Throwable { HashMap<String, Object> properties = new HashMap<String, Object>(); properties.put("hibernate.transaction.jta.platform", AtomikosJtaPlatform.class.getName()); properties.put("javax.persistence.transactionType", "JTA"); LocalContainerEntityManagerFactoryBean entityManager = new LocalContainerEntityManagerFactoryBean(); entityManager.setJtaDataSource(secondDataSource); entityManager.setJpaVendorAdapter(jpaVendorAdapter); entityManager.setPackagesToScan("com.kanq.bigdata.greenplum.entity.gp"); entityManager.setPersistenceUnitName("metaPersistenceUnit"); entityManager.setJpaPropertyMap(properties); return entityManager; } } @Configuration @ComponentScan @EnableTransactionManagement public class DataSourceConfig { // @ConfigurationProperties(prefix="spring.datasource.ds1") @Bean(name = "primaryDataSource",initMethod = "init", destroyMethod = "close") @Qualifier("primaryDataSource") @Primary public DataSource primaryDataSource(PGProperties pgProperties) { // DataSource pg = DataSourceBuilder.create().build(); PGXADataSource pgxa=new PGXADataSource(); pgxa.setURL(pgProperties.getJdbcUrl()); pgxa.setUser(pgProperties.getUsername()); pgxa.setPassword(pgProperties.getPassword()); AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean(); xaDataSource.setXaDataSource(pgxa); xaDataSource.setUniqueResourceName("primaryDataSource"); return xaDataSource; } @Bean(name = "secondDataSource") @Qualifier("secondDataSource") public DataSource secondaryDataSource(GPProperties gpProperties) { PGXADataSource gpxa=new PGXADataSource(); gpxa.setURL(gpProperties.getJdbcUrl()); gpxa.setUser(gpProperties.getUsername()); gpxa.setPassword(gpProperties.getPassword()); AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean(); xaDataSource.setXaDataSource(gpxa); xaDataSource.setUniqueResourceName("secondDataSource"); return xaDataSource; } @Bean public PropertySourcesPlaceholderConfigurer propertySourcesPlaceholderConfigurer() { return new PropertySourcesPlaceholderConfigurer(); } //设置JPA特性 @Bean public JpaVendorAdapter jpaVendorAdapter() { HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter(); //显示sql hibernateJpaVendorAdapter.setShowSql(true); //自动生成/更新表 hibernateJpaVendorAdapter.setGenerateDdl(false); //设置数据库类型 hibernateJpaVendorAdapter.setDatabase(Database.POSTGRESQL); return hibernateJpaVendorAdapter; } @Bean(name = "atomikosTransactionManager",initMethod = "init", destroyMethod = "close") public TransactionManager atomikosTransactionManager() throws Throwable { UserTransactionManager userTransactionManager = new UserTransactionManager(); userTransactionManager.setForceShutdown(false); AtomikosJtaPlatform.transactionManager = userTransactionManager; return userTransactionManager; } @Bean(name = "userTransaction") public UserTransaction userTransaction() throws Throwable { UserTransactionImp userTransactionImp = new UserTransactionImp(); userTransactionImp.setTransactionTimeout(10000); return userTransactionImp; } @Bean(name = "transactionManager") @DependsOn({"userTransaction", "atomikosTransactionManager"}) public PlatformTransactionManager transactionManager() throws Throwable { UserTransaction userTransaction = userTransaction(); AtomikosJtaPlatform.transaction = userTransaction; TransactionManager atomikosTransactionManager = atomikosTransactionManager(); return new JtaTransactionManager(userTransaction, atomikosTransactionManager); } } import org.hibernate.engine.transaction.jta.platform.internal.AbstractJtaPlatform; import javax.transaction.TransactionManager; import javax.transaction.UserTransaction; public class AtomikosJtaPlatform extends AbstractJtaPlatform { private static final long serialVersionUID = 1L; static TransactionManager transactionManager; static UserTransaction transaction; @Override protected TransactionManager locateTransactionManager() { return transactionManager; } @Override protected UserTransaction locateUserTransaction() { return transaction; } }

Entity、Repository

Entity、Repository都需要分2个文件夹，配置中配置了不同数据源使用不同的文件夹

略

Controller

需要加事务@Transactional

@RequestMapping(value = "/test", method = RequestMethod.POST) @Transactional public ResultModel test(){ Map<String,Object> res=null; res=service.test(); return Message.success(res); }

Service

如果Controller没有@Transactional注解，这里回滚会出错，提示没有事务。

//atomikos事务 @Transactional public boolean test(){ try { XXX xxx=new XXX("wo",1); YYY yyy=new YYY("qaq",2); xxxDao.saveAndFlush(xxx); yyyDao.saveAndFlush(yyy); }catch (Exception e){ log.error("数据更新异常",e); TransactionAspectSupport.currentTransactionStatus().setRollbackOnly(); return false; } return true; }

2个不同数据源的事务都会回滚。