springboot+mybatis多数据源实现

我不是女神ヾ 2023-07-14 05:41 1阅读 0赞

# 前言 #

这次使用到多数据源的原因是因为重构一个项目，旧数据又需要迁移到新数据库，数据库表设计有差异，所以需要编写程序进行迁移。

# 实现 #

**目录结构**

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhYnlsb3Zld2Vp_size_16_color_FFFFFF_t_70][]

**application.yml **

server:
      port: 8080
    spring:
      profiles:
        active: dev
      application:
        name: case-multi-dataSource
      datasource:
        table1:
          driverClassName: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://localhost:3306/table1?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
          username: root
          password: root
          initialSize: 5
          minIdle: 5
          maxActive: 50
          maxWait: 10000
          timeBetweenEvictionRunsMillis: 10000
          minEvictableIdleTimeMillis: 50000
          validationQuery: SELECT 1 FROM DUAL
          testWhileIdle: true
          testOnBorrow: false
          testOnReturn: false
          poolPreparedStatements: true
          maxPoolPreparedStatementPerConnectionSize: 20
          connectionProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500
          filters: stat,wall,log4j
        table2:
          driverClassName: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://localhost:3306/table2?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
          username: root
          password: root
          initialSize: 5
          minIdle: 5
          maxActive: 50
          maxWait: 10000
          timeBetweenEvictionRunsMillis: 10000
          minEvictableIdleTimeMillis: 50000
          validationQuery: SELECT 1 FROM DUAL
          testWhileIdle: true
          testOnBorrow: false
          testOnReturn: false
          poolPreparedStatements: true
          maxPoolPreparedStatementPerConnectionSize: 20
          connectionProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500
          filters: stat,wall,log4j

** 配置两个 DataSourceConfig**

package com.example.multi.config;
    
    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.autoconfigure.jdbc.DataSourceBuilder;
    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.core.io.support.PathMatchingResourcePatternResolver;
    import org.springframework.jdbc.datasource.DataSourceTransactionManager;
    
    import javax.sql.DataSource;
    
    @Configuration
    //basePackages 对应dao目录
    @MapperScan(basePackages = "com.example.utils.dao.table1.**", sqlSessionTemplateRef = "table1SqlSessionTemplate")
    public class DataSource1Config {
    
        @Bean(name = "table1DataSource")
        //prefix 对应yml中数据库配置信息
        @ConfigurationProperties(prefix = "spring.datasource.table1")
        @Primary
        public DataSource testDataSource() {
            return DataSourceBuilder.create().build();
        }
    
        @Bean(name = "table1SqlSessionFactory")
        @Primary
        public SqlSessionFactory testSqlSessionFactory(@Qualifier("table1DataSource") DataSource dataSource) throws Exception {
            SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
            bean.setDataSource(dataSource);
            // 如果有用到mybatis-config.xml配置，就需要配置一下两行代码，否则可以不用
            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            bean.setConfigLocation(resolver.getResource("classpath:mybatis-config.xml"));
            // 配置mapper xml文件位置
            bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/table1/*.xml"));
            return bean.getObject();
        }
    
        @Bean(name = "table1TransactionManager")
        @Primary
        public DataSourceTransactionManager testTransactionManager(@Qualifier("table1DataSource") DataSource dataSource) {
            return new DataSourceTransactionManager(dataSource);
        }
    
        @Bean(name = "table1SqlSessionTemplate")
        @Primary
        public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("table1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
            return new SqlSessionTemplate(sqlSessionFactory);
        }
    
    }

package com.example.multi.config;
    
    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.autoconfigure.jdbc.DataSourceBuilder;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
    import org.springframework.jdbc.datasource.DataSourceTransactionManager;
    
    import javax.sql.DataSource;
    
    @Configuration
    //basePackages 对应dao目录
    @MapperScan(basePackages = "com.example.utils.dao.table1.**", sqlSessionTemplateRef = "table2SqlSessionTemplate")
    public class DataSource2Config {
    
        @Bean(name = "table2DataSource")
        //prefix 对应yml中数据库配置信息
        @ConfigurationProperties(prefix = "spring.datasource.table2")
        public DataSource testDataSource() {
            return DataSourceBuilder.create().build();
        }
    
        @Bean(name = "table2SqlSessionFactory")
        public SqlSessionFactory testSqlSessionFactory(@Qualifier("table2DataSource") DataSource dataSource) throws Exception {
            SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
            bean.setDataSource(dataSource);
            // 如果有用到mybatis-config.xml配置，就需要配置一下两行代码，否则可以不用
            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            bean.setConfigLocation(resolver.getResource("classpath:mybatis-config.xml"));
            // 配置mapper xml文件位置
            bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/table2/*.xml"));
            return bean.getObject();
        }
    
        @Bean(name = "table2TransactionManager")
        public DataSourceTransactionManager testTransactionManager(@Qualifier("table2DataSource") DataSource dataSource) {
            return new DataSourceTransactionManager(dataSource);
        }
    
        @Bean(name = "table2SqlSessionTemplate")
        public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("table2SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
            return new SqlSessionTemplate(sqlSessionFactory);
        }
    }

** 主要部分已经完成，其他操作只要在dao和mapper中两个不同的实现即可**

# 事务处理 #

**单数据源事务**

//此处transactionManager 指向config中的Bean
    @Transactional(transactionManager = "table1TransactionManager")
    public void insert() {
        table1Dao.insertTable1();
    }

注意：以下操作两个数据源，第一个成功插入数据库，第二个失败进行回滚

//此处transactionManager 指向config中的Bean
    @Transactional(transactionManager = "table1TransactionManager")
    public void insert() {
        table1Dao.insertTable1();
        table2Dao.insertTable2();
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhYnlsb3Zld2Vp_size_16_color_FFFFFF_t_70]: /images/20230529/5ec2bba75a114757ad300bb4eedf74e6.png