SpringBoot使用MyBatis 集成Atomikos 实现分布式事务管理多数据源

￡神魔★判官ぃ 2023-06-15 13:29 34阅读 0赞

### 1.项目结构： ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9sdWNpZmVyLmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70][]

###  2.配置：（最重要） ###

**由于只用了两个数据源，这里就有两个读取springboot主配置类的两个类：**

package com.lucifer.config;
    
    import lombok.Data;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    
    
    /**
     * 数据源1配置类
     * 用于读取配置文件自定义配置
     *
     * @author Lucifer
     */
    @Data
    @ConfigurationProperties(prefix = "spring.datasource.druid.master")
    public class DBConfig {
    
        private String url;
        private String username;
        private String password;
    
    }

package com.lucifer.config;
    
    import lombok.Data;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    
    /**
     * 数据源2配置类
     * 用于读取配置文件自定义配置
     */
    @Data
    @ConfigurationProperties(prefix = "spring.datasource.druid.slave")
    public class DBConfig2 {
    
        private String url;
        private String username;
        private String password;
    
    }

**      使用xa数据源统一管理配置信息，每一个数据源对应一个SqlSessionFactoryBean;并且管理各自的mapper，然后配置全局的事务管理器，并将事务管理器纳入到spring容器当中；**

package com.lucifer.config;
    
    import com.mysql.cj.jdbc.MysqlXADataSource;
    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;
    import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
    
    import javax.sql.DataSource;
    import java.sql.SQLException;
    
    /**
     * @author Lucifer
     */
    @Configuration
    @MapperScan(basePackages = "com.lucifer.order.mapper", sqlSessionFactoryRef = "firstSqlSessionFactory")
    public class MybatisDatasourceConfig {
    
        @Primary
        @Bean(name = "firstDataSource")
        public DataSource firstDataSource(DBConfig dbConfig) throws SQLException {
            //获取mysqlXaDataSource
            MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
            //数据源配置
            mysqlXaDataSource.setUrl(dbConfig.getUrl());
            mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
            mysqlXaDataSource.setPassword(dbConfig.getPassword());
            mysqlXaDataSource.setUser(dbConfig.getUsername());
            mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
            //将mysql数据源交由Atomikos统一管理
            AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
            xaDataSource.setXaDataSource(mysqlXaDataSource);
            xaDataSource.setUniqueResourceName("firstDataSource");
            return xaDataSource;
        }
    
        /**
         * 设置SqlSessionFactory，每一个数据源一个SqlSessionFactoryBean 
         *
         * @param dataSource
         * @return
         * @throws Exception
         */
        @Bean(name = "firstSqlSessionFactory")
        public SqlSessionFactory firstSqlSessionFactory(@Qualifier("firstDataSource") DataSource dataSource) throws Exception {
            SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
            bean.setDataSource(dataSource);
            bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:/mapper/order/*Mapper.xml"));
            return bean.getObject();
        }
    
        @Bean(name = "firstSqlSessionTemplate")
        public SqlSessionTemplate testSqlSessionTemplate(
                @Qualifier("firstSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
            return new SqlSessionTemplate(sqlSessionFactory);
        }
    
    
    }

package com.lucifer.config;
    
    import com.mysql.cj.jdbc.MysqlXADataSource;
    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.core.io.support.PathMatchingResourcePatternResolver;
    
    import javax.sql.DataSource;
    import java.sql.SQLException;
    
    /**
     * @author Lucifer
     */
    @Configuration
    @MapperScan(basePackages = "com.lucifer.log.mapper", sqlSessionFactoryRef = "secondSqlSessionFactory")
    public class MybatisDatasourceConfig2 {
    
        @Bean(name = "secondDataSource")
        public DataSource secondDataSource(DBConfig2 dbConfig2) throws SQLException {
            //获取mysqlXaDataSource
            MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
            //配置数据源
            mysqlXaDataSource.setUrl(dbConfig2.getUrl());
            mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
            mysqlXaDataSource.setPassword(dbConfig2.getPassword());
            mysqlXaDataSource.setUser(dbConfig2.getUsername());
            mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
            //将mysql数据源交由Atomikos统一管理
            AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
            xaDataSource.setXaDataSource(mysqlXaDataSource);
            xaDataSource.setUniqueResourceName("secondDataSource");
            return xaDataSource;
        }
    
    
        /**
         * 设置SqlSessionFactory，每一个数据源一个SqlSessionFactoryBean 
         *
         * @param dataSource
         * @return
         * @throws Exception
         */
        @Bean(name = "secondSqlSessionFactory")
        public SqlSessionFactory secondSqlSessionFactory(@Qualifier("secondDataSource") DataSource dataSource) throws Exception {
            SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
            bean.setDataSource(dataSource);
            bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:/mapper/log/*Mapper.xml"));
            return bean.getObject();
        }
    
        @Bean(name = "secondSqlSessionTemplate")
        public SqlSessionTemplate testSqlSessionTemplate(
                @Qualifier("secondSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
            return new SqlSessionTemplate(sqlSessionFactory);
        }
    
    
    }

**application.yml：**springboot的配置文件；

spring:
      datasource:
        type: com.alibaba.druid.pool.xa.DruidXADataSource
        druid:
          master:
            driver-class-name: com.mysql.jdbc.Driver
            url: jdbc:mysql://192.168.87.130:3306/order?autoReconnect=true&useSSL=false&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull
            username: root
            password: 123456
            maxPoolSize: 5
          slave:
            driver-class-name: com.mysql.jdbc.Driver
            url: jdbc:mysql://192.168.87.130:3306/logs?autoReconnect=true&useSSL=false&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull
            username: root
            password: 123456
            maxPoolSize: 5

### 3.mapper接口 ###

package com.lucifer.log.mapper;
    
    import com.lucifer.pojo.LogInfo;
    import org.apache.ibatis.annotations.Param;
    
    import java.util.List;
    
    /**
     * @author Lucifer
     */
    public interface LogMapper {
    
    
        /**
         * 新增
         * @param logInfo
         * @return
         */
        int insertLogInfo(LogInfo logInfo);
    
        /**
         * 批量新增
         * @param logInfoList
         * @return
         */
        int insertBatchLogs(@Param("logInfoList") List<LogInfo> logInfoList);
    
    }

package com.lucifer.order.mapper;
    
    import com.lucifer.pojo.OrderInfo;
    import org.apache.ibatis.annotations.Param;
    
    import java.util.List;
    
    /**
     * @author Lucifer
     */
    public interface OrderMapper {
    
        /**
         * 新增
         *
         * @param orderInfo
         * @return
         */
        int insertOrderInfo(OrderInfo orderInfo);
    
        /**
         * 批量新增
         *
         * @param orderInfos
         * @return
         */
        int insertBatchOrder(@Param("orderInfos") List<OrderInfo> orderInfos);
    
    }

### 4.实体类 ###

package com.lucifer.pojo;
    
    import lombok.Data;
    
    import java.util.Date;
    
    /**
     * @author Lucifer
     */
    @Data
    public class OrderInfo {
    
        private Integer id;
    
        private Double money;
    
        private String userId;
    
        private String address;
    
        private Date createTime;
    
    }

package com.lucifer.pojo;
    
    import lombok.Data;
    
    import java.util.Date;
    
    /**
     * @author Lucifer
     */
    @Data
    public class LogInfo {
    
        private Integer id;
    
        private Date createTime;
    
        private String content;
    }

### 5.service接口及实现类 ###

package com.lucifer.service;
    
    import com.lucifer.pojo.OrderInfo;
    
    /**
     * @author Lucifer
     */
    public interface OrderService {
    
        /**
         * 新增
         * @param orderInfo
         */
        void addOrder(OrderInfo orderInfo);
    
    
    }

package com.lucifer.service.impl;
    
    
    import com.lucifer.log.mapper.LogMapper;
    import com.lucifer.order.mapper.OrderMapper;
    import com.lucifer.pojo.LogInfo;
    import com.lucifer.pojo.OrderInfo;
    import com.lucifer.service.OrderService;
    import lombok.extern.slf4j.Slf4j;
    import org.springframework.stereotype.Service;
    
    import javax.annotation.Resource;
    import javax.transaction.Transactional;
    import java.util.Date;
    import java.util.Random;
    
    /**
     * @author Lucifer
     */
    @Slf4j
    @Service
    @Transactional(rollbackOn = Exception.class)
    public class OrderServiceImpl implements OrderService {
    
        @Resource
        OrderMapper orderMapper;
    
        @Resource
        LogMapper logMapper;
    
    
        @Override
        public void addOrder(OrderInfo orderInfo) {
            int insert = orderMapper.insertOrderInfo(orderInfo);
            log.info("order库新增sql执行条数:{}",insert);
    
            //测试1 异常回滚
            //int i=1/0;
    
            LogInfo logInfo=new LogInfo();
            logInfo.setId((new Random().nextInt()));
            logInfo.setCreateTime(new Date());
            logInfo.setContent(orderInfo.toString());
            int insert1 = logMapper.insertLogInfo(logInfo);
            log.info("logs库新增sql执行条数:{}",insert1);
    
            //测试2 异常回滚
            int i=1/0;
        }
    
    }

### 6.spring boot主配置类 ###

package com.lucifer;
    
    import com.lucifer.config.DBConfig;
    import com.lucifer.config.DBConfig2;
    import org.mybatis.spring.annotation.MapperScan;
    import org.springframework.boot.SpringApplication;
    import org.springframework.boot.autoconfigure.SpringBootApplication;
    import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
    import org.springframework.boot.context.properties.EnableConfigurationProperties;
    
    /**
     * @author Lucifer
     */
    @SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
    @EnableConfigurationProperties(value = { DBConfig.class, DBConfig2.class})
    @MapperScan(value = {"com.lucifer.log.mapper","com.lucifer.order.mapper"})
    public class SpringbootAtomikosApplication {
    
        public static void main(String[] args) {
            SpringApplication.run(SpringbootAtomikosApplication.class, args);
        }
    
    }

### 7、mapper.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.lucifer.log.mapper.LogMapper">
    
        <insert id="insertLogInfo">
            insert into log_info(id,create_time,content) values(#{id},#{createTime},#{content})
        </insert>
    
        <insert id="insertBatchLogs">
            insert into log_info
                (id,create_time,content)
            values
            <foreach collection="logInfoList" item="log" index="index" separator=",">
                (#{log.id},#{log.createTime},#{log.content})
            </foreach>
        </insert>
    </mapper>

<?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.lucifer.order.mapper.OrderMapper">
    
        <insert id="insertOrderInfo">
            insert into order_info(id,money,user_id,address,create_time)
            values(#{id},#{money},#{userId},#{address},#{createTime})
        </insert>
    
    
        <insert id="insertBatchOrder">
            insert into order_info
                (id,money,user_id,address,create_time)
            values
            <foreach collection="orderInfos" item="order" index="index" separator=",">
                (#{order.id},#{order.money},#{order.userId},#{order.address},#{order.createTime})
            </foreach>
    
        </insert>
    
    
    </mapper>

# 附： #

### sql脚本： ###

#logs数据库中
    CREATE TABLE `log_info` (
      `id` int(11) NOT NULL,
      `create_time` datetime DEFAULT NULL,
      `content` longtext,
      PRIMARY KEY (`id`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
    
    
    #order数据库中
    CREATE TABLE `order_info` (
      `id` int(11) NOT NULL,
      `money` double NOT NULL,
      `user_id` varchar(20) DEFAULT NULL,
      `address` varchar(200) DEFAULT NULL,
      `create_time` datetime DEFAULT NULL,
      PRIMARY KEY (`id`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8;

pom.xml:

<?xml version="1.0" encoding="UTF-8"?>
    <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
             xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
        <modelVersion>4.0.0</modelVersion>
        <parent>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-parent</artifactId>
            <version>2.2.1.RELEASE</version>
            <relativePath/> 
        </parent>
        <groupId>com.lucifer</groupId>
        <artifactId>springboot-atomikos</artifactId>
        <version>0.0.1-SNAPSHOT</version>
        <name>springboot-atomikos</name>
        <description>Demo project for Spring Boot</description>
    
        <properties>
            <java.version>1.8</java.version>
            <mybatis-plus.version>3.2.0</mybatis-plus.version>
        </properties>
    
        <dependencies>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-web</artifactId>
            </dependency>
    
            <dependency>
                <groupId>mysql</groupId>
                <artifactId>mysql-connector-java</artifactId>
                <scope>runtime</scope>
            </dependency>
    
            
            <dependency>
                <groupId>org.projectlombok</groupId>
                <artifactId>lombok</artifactId>
                <optional>true</optional>
            </dependency>
    
            
            <dependency>
                <groupId>mysql</groupId>
                <artifactId>mysql-connector-java</artifactId>
            </dependency>
    
            
           
            <dependency>
                <groupId>org.mybatis.spring.boot</groupId>
                <artifactId>mybatis-spring-boot-starter</artifactId>
                <version>2.1.0</version>
            </dependency>
    
    
    
            
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-jta-atomikos</artifactId>
            </dependency>
    
    
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-test</artifactId>
                <scope>test</scope>
                <exclusions>
                    <exclusion>
                        <groupId>org.junit.vintage</groupId>
                        <artifactId>junit-vintage-engine</artifactId>
                    </exclusion>
                </exclusions>
            </dependency>
    
            <dependency>
                <groupId>com.alibaba</groupId>
                <artifactId>druid</artifactId>
                <version>1.1.20</version>
            </dependency>
    
        </dependencies>
    
        <build>
            <plugins>
                <plugin>
                    <groupId>org.springframework.boot</groupId>
                    <artifactId>spring-boot-maven-plugin</artifactId>
                </plugin>
            </plugins>
        </build>
    
    </project>

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9sdWNpZmVyLmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191122175648790.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9sdWNpZmVyLmJsb2cuY3Nkbi5uZXQ=,size_16,color_FFFFFF,t_70