Spring boot Mybatis多数据源支持

Bertha 。 2021-11-11 03:46 314阅读 0赞

## 前言 ##

由于项目的数据存储在多源关系型数据库中，oracle、mysql等，需要提供对多源数据库链接的支持。同时支持MySQL jpa操作。

## application.yml配置 ##

**需要去掉spring boot默认的jdbc链接配置**

spring:
    #    url: jdbc:mysql://127.0.0.1:3306/testdb?useUnicode=true&characterEncoding=utf-8&useSSL=false
    #    username: root
    #    password: 123456
    #    type: com.alibaba.druid.pool.DruidDataSource
    #    driverClassName: com.mysql.jdbc.Driver

**添加自定义数据库配置**  
注意：配置名称需要跟配置的连接池提供的默认名称匹配，案例采用 DruidDataSource数据源作为连接池

#mysql配置
    mysql:
      url:  jdbc:mysql://127.0.0.1:3306/testdb?useUnicode=true&characterEncoding=utf-8&useSSL=false
      username: root
      password: 123456
      type: com.alibaba.druid.pool.DruidDataSource
      driverClassName: com.mysql.jdbc.Driver
    #oracle配置
    oracle:
      url: jdbc:oracle:thin:@127.0.0.1:1521/orcl11g
      username: tiger
      password: tiger
      type: com.alibaba.druid.pool.DruidDataSource
      driverClassName: oracle.jdbc.driver.OracleDriver

## spring boot 配置类 ##

**mysql配置**

**MySqlConfig.java**

import com.alibaba.druid.pool.DruidDataSource;
    import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
    import org.apache.ibatis.session.SqlSessionFactory;
    import org.mybatis.spring.SqlSessionFactoryBean;
    import org.mybatis.spring.annotation.MapperScan;
    import org.springframework.beans.factory.annotation.Qualifier;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    import org.springframework.boot.jdbc.DataSourceBuilder;
    import org.springframework.boot.orm.jpa.hibernate.SpringImplicitNamingStrategy;
    import org.springframework.boot.orm.jpa.hibernate.SpringPhysicalNamingStrategy;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.context.annotation.Primary;
    import org.springframework.core.io.ClassPathResource;
    import org.springframework.core.io.Resource;
    import org.springframework.core.io.ResourceLoader;
    import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
    import org.springframework.core.io.support.ResourcePatternResolver;
    import org.springframework.orm.jpa.JpaTransactionManager;
    import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
    import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
    import org.springframework.transaction.PlatformTransactionManager;
    
    import javax.persistence.EntityManagerFactory;
    import javax.sql.DataSource;
    import java.util.ArrayList;
    import java.util.List;
    import java.util.Properties;
    
    /**
     * mysql链接配置
     *
     * 支持多源配置， 注意只有默认主库中的支持jpa默认加载配置
     *
     * @version 2019-08-02
     * @author 小米虾
     */
    @Configuration
    @MapperScan(value = "com.test.modules.*.*.mysql.mapper", sqlSessionFactoryRef = "mysqlSessionFactory")
    public class MySqlConfig {
    
        @Bean
        //副数据库需要配置@Qualifier
        @Qualifier("mysql")
        @ConfigurationProperties(prefix="mysql")
         @Primary //当其它地方也用数据源的时候 默认加载它
        public DataSource mysqlDataSource() {
            return DataSourceBuilder.create().type(DruidDataSource.class).build();
        }
    
        //此处将 MySQL 作为第一链接  @Primary
        @Bean("mysqlSessionFactory")
        @Primary
        public SqlSessionFactory mysqlSessionFactory(@Qualifier("mysql") DataSource mysqlDataSource) throws Exception {
    
            MybatisSqlSessionFactoryBean mybatisSqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
            mybatisSqlSessionFactoryBean.setDataSource(mysqlDataSource);
    
           //加载 mappber.xml 配置文件
            ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
    
            Resource[] resources = resolver.getResources("classpath:mapper/mysql/*.xml");
    
            mybatisSqlSessionFactoryBean.setMapperLocations(resources);
    
            return mybatisSqlSessionFactoryBean.getObject();
        }
    
    
    
        /**
         *  jpa管理类，配置加载JpaRepository 相关
         */
        @Bean
        @Qualifier("mysqlEntityManagerFactory")
        public LocalContainerEntityManagerFactoryBean entityManagerFactory(@Qualifier("mysql") DataSource mysqlDataSource) {
    
            HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
            vendorAdapter.setGenerateDdl(false);
            vendorAdapter.setShowSql(true);
    
    
            LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
            factory.setJpaVendorAdapter(vendorAdapter);
            factory.setPackagesToScan("com.test.*");
            factory.setDataSource(mysqlDataSource);
    
            //配置jpa _ 驼峰转换
            Properties properties = new Properties();
            properties.setProperty("hibernate.physical_naming_strategy", SpringPhysicalNamingStrategy.class.getName());
            properties.setProperty("hibernate.implicit_naming_strategy", SpringImplicitNamingStrategy.class.getName());
    
            factory.setJpaProperties(properties);
            return factory;
        }
    
        @Bean
        public PlatformTransactionManager transactionManager(@Qualifier("mysqlEntityManagerFactory") EntityManagerFactory entityManagerFactory) {
    
            JpaTransactionManager txManager = new JpaTransactionManager();
            txManager.setEntityManagerFactory(entityManagerFactory);
            return txManager;
        }
    
    
    }

**oracle配置**  
**OracleConfig.class**