spring boot+atomikos+mybatis+druid分布式事务实现

Love The Way You Lie 2022-06-02 01:55 246阅读 0赞

## 配置数据源 ##

新建配置类DataSourceConfig：

public class DataSourceConfig { 
    
        private static final String PRIMARY_MAPPER_BASE_PACKAGE = "io.github.yidasanqian.mapper.master";
        private static final String BUSINESS_MAPPER_BASE_PACKAGE = "io.github.yidasanqian.mapper.business";
    
        private static final String DATASOURCE_DRUID_PROPERTIES = "datasource/druid.properties";
        private static final String DATASOURCE_DRUID_PRIMARY_PROPERTIES = "datasource/druid-primary.properties";
        private static final String DATASOURCE_DRUID_BUSINESS_PROPERTIES = "datasource/druid-business.properties";
    
        private static final String CLASSPATH_MAPPER_XML = "classpath:mapper/*/*.xml";
    
        /** * druid 公共配置 */
        private static Properties commonProperties;
    
        static {
            commonProperties = new Properties();
            InputStream in = DataSourceConfig.class.getClassLoader().getResourceAsStream(DATASOURCE_DRUID_PROPERTIES);
            try {
                commonProperties.load(in);
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }
    
        private Properties loadDruidProperties(String path) throws IOException {
            Properties properties = new Properties();
            InputStream in = getClass().getClassLoader().getResourceAsStream(path);
            properties.load(in);
            for (Map.Entry<Object, Object> entry : commonProperties.entrySet()) {
                properties.setProperty(entry.getKey().toString(), entry.getValue().toString());
            }
            return properties;
        }
    
        /** * 设置数据源 * * @return * @throws IOException */
        @Primary
        @Bean
        public AtomikosDataSourceBean primaryDataSource() throws IOException {
            return getAtomikosDataSourceBean(DATASOURCE_DRUID_PRIMARY_PROPERTIES);
        }
    
    
        @Bean
        public AtomikosDataSourceBean businessDataSource() throws IOException {
            return getAtomikosDataSourceBean(DATASOURCE_DRUID_BUSINESS_PROPERTIES);
        }
    
        private AtomikosDataSourceBean getAtomikosDataSourceBean(String dataSourceProperties) throws IOException {
            Properties properties = loadDruidProperties(dataSourceProperties);
            AtomikosDataSourceBean dataSourceBean = new AtomikosDataSourceBean();
            // 配置DruidXADataSource
            DruidXADataSource xaDataSource = new DruidXADataSource();
            xaDataSource.configFromPropety(properties);
            // 设置置AtomikosDataSourceBean XADataSource
            dataSourceBean.setXaDataSource(xaDataSource);
            return dataSourceBean;
        }
    
        /** * 设置{@link SqlSessionFactoryBean}的数据源 * @param primaryDataSource 主数据源 * @return */
        @Primary
        @Bean
        public SqlSessionFactoryBean primarySqlSessionFactoryBean(@Qualifier("primaryDataSource") AtomikosDataSourceBean primaryDataSource) {
            return getSqlSessionFactoryBean(primaryDataSource);
        }
    
        @Bean
        public SqlSessionFactoryBean businessSqlSessionFactoryBean(@Qualifier("businessDataSource") AtomikosDataSourceBean businessDataSource) {
            return getSqlSessionFactoryBean(businessDataSource);
        }
    
        private SqlSessionFactoryBean getSqlSessionFactoryBean(AtomikosDataSourceBean dataSource) {
            SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
            sqlSessionFactoryBean.setDataSource(dataSource);
            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            try {
                sqlSessionFactoryBean.setMapperLocations(resolver.getResources(CLASSPATH_MAPPER_XML));
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            return sqlSessionFactoryBean;
        }
    
        /** * 搜索{@link DataSourceConfig#PRIMARY_MAPPER_BASE_PACKAGE} 包下的Mapper接口，并且将这些接口 * 交由{@link MapperScannerConfigurer#sqlSessionFactoryBeanName} 属性设置的SqlSessionFactoryBean管理 * @return */
        @Bean
        public MapperScannerConfigurer primaryMapperScannerConfigurer() {
            MapperScannerConfigurer mapperScannerConfigurer = new MapperScannerConfigurer();
            mapperScannerConfigurer.setBasePackage(PRIMARY_MAPPER_BASE_PACKAGE);
            mapperScannerConfigurer.setSqlSessionFactoryBeanName("primarySqlSessionFactoryBean");
            return mapperScannerConfigurer;
        }
    
        /** * 搜索{@link DataSourceConfig#BUSINESS_MAPPER_BASE_PACKAGE} 包下的Mapper接口，并且将这些接口 * 交由{@link MapperScannerConfigurer#sqlSessionFactoryBeanName} 属性设置的SqlSessionFactoryBean管理 * @return */
        @Bean
        public MapperScannerConfigurer businessMapperScannerConfigurer() {
            MapperScannerConfigurer mapperScannerConfigurer = new MapperScannerConfigurer();
            mapperScannerConfigurer.setBasePackage(BUSINESS_MAPPER_BASE_PACKAGE);
            mapperScannerConfigurer.setSqlSessionFactoryBeanName("businessSqlSessionFactoryBean");
            return mapperScannerConfigurer;
        }
    }

代码上都有注释，如有疑问，评论指出。

## 代码实现 ##

结构：

![这里写图片描述][SouthEast]

两个Mapper接口定义一样的`save`方法

service实现类`DemoServiceImpl`：

@Transactional(rollbackFor = Exception.class)
    @Service
    public class DemoServiceImpl implements DemoService { 
    
        private final Logger log = LoggerFactory.getLogger(getClass());
    
        @Resource
        private MasterDemoMapper masterDemoMapper;
    
        @Resource
        private BusinessDemoMapper businessDemoMapper;
    
        /** * 正常测试分布式事务 * * @return * @throws Exception */
        @Override
        public int save() throws Exception {
            log.info("save");
            Demo dsDemo = new Demo();
            dsDemo.setName("xa事务测试");
            int row = masterDemoMapper.save(dsDemo);
            log.info("保存之后");
            Demo dsDemo1 = new Demo();
            dsDemo1.setName("xa事务测试2");
            int row2 = businessDemoMapper.save(dsDemo1);
            return row + row2;
        }
    
        /** * 测试分布式事务回滚 * @return * @throws Exception */
        @Override
        public int save2() throws Exception {
            log.info("save2");
            Demo dsDemo = new Demo();
            dsDemo.setName("xa事务回滚测试");
            int row = masterDemoMapper.save(dsDemo);
            log.info("保存之后异常");
            int a = 1 / 0;
    
            Demo dsDemo1 = new Demo();
            dsDemo1.setName("xa事务回滚测试2");
            int row2 = businessDemoMapper.save(dsDemo1);
            return row + row2;
        }
    }

DemoController：

@RestController
    public class DemoController { 
    
        @Resource
        private DemoService dsService;
    
        @RequestMapping("/testXaDatasource")
        public String testXaDatasource() {
            int result = 0;
            try {
                result = dsService.save();
            } catch (Exception e) {
                e.printStackTrace();
            }
            return String.valueOf(result);
        }
    
        @RequestMapping("/testXaDatasource2")
        public String testXaDatasource2() {
            int result = 0;
            try {
                result = dsService.save2();
            } catch (Exception e) {
                e.printStackTrace();
            }
            return String.valueOf(result);
        }
    }

访问[http://localhost:8080/testXaDatasource][http_localhost_8080_testXaDatasource]返回2，表明多数据源及分布式事务可用。

访问[http://localhost:8080/testXaDatasource2][http_localhost_8080_testXaDatasource2]返回0，表明分布式事务回滚可用

## 源码地址 ##

[https://gitee.com/yidasanqian/spring-boot-distributed-transaction-demo.git][https_gitee.com_yidasanqian_spring-boot-distributed-transaction-demo.git]

[SouthEast]: /images/20220602/16ba06b37a1844d18f47e3580552a830.png
[http_localhost_8080_testXaDatasource]: http://localhost:8080/testXaDatasource
[http_localhost_8080_testXaDatasource2]: http://localhost:8080/testXaDatasource2
[https_gitee.com_yidasanqian_spring-boot-distributed-transaction-demo.git]: https://gitee.com/yidasanqian/spring-boot-distributed-transaction-demo.git