精通SpringBoot —— 第十四篇：整合mybatis和jpa的多数据源

刚更新了一篇SpringBoot整合mybatis。突然想mybatis虽然功能强大，但是对sql的要求还是算高的。Spring Data Jpa对于某些用户来说 ，还是个比较简单的选择。毕竟不用写较复杂的sql语句，于是就想，来一个SpringBoot+mybatis + jpa的大杂烩。
说到底，多数据源整合其实就是要保证项目中有且仅有一个primaryDataSource，即主数据源，它是无可替代的。
先看application.yml 有多少个数据源就的配置多少个数据库链接

spring:
  ## multipart datasource configuration ##
  datasource:
    primary:
      jdbc-url: jdbc:mysql://localhost:3306/javashop
      username: root
      password: 123456
      driver-class-name: com.mysql.jdbc.Driver
      type: com.alibaba.druid.pool.DruidDataSource
    secondary:
      jdbc-url: jdbc:mysql://localhost:3306/test
      username: root
      password: 123456
      driver-class-name: com.mysql.jdbc.Driver
      type: com.alibaba.druid.pool.DruidDataSource
    master:
      jdbc-url: jdbc:mysql://localhost:3306/javashop
      username: root
      password: 123456
      driver-class-name: com.mysql.jdbc.Driver
      type: com.alibaba.druid.pool.DruidDataSource
    cluster:
      jdbc-url: jdbc:mysql://localhost:3306/test
      username: root
      password: 123456
      driver-class-name: com.mysql.jdbc.Driver
      type: com.alibaba.druid.pool.DruidDataSource

  jpa:
    hibernate:
      ddl-auto: update
      naming:
        physical-strategy: org.springframework.boot.orm.jpa.hibernate.SpringPhysicalNamingStrategy
    show-sql: true
    database-platform: org.hibernate.dialect.MySQL5InnoDBDialect

项目结构图

└─src
    ├─main
    │  ├─java
    │  │  └─com
    │  │      └─developlee
    │  │          └─multipartdatasource
    │  │              │  MultipartDatasourceApplication.java
    │  │              │  
    │  │              ├─config
    │  │              │  ├─jpa
    │  │              │  │      DataSourceConfig.java
    │  │              │  │      PrimaryDataSourceConfig.java
    │  │              │  │      SecondaryDataSourceConfig.java
    │  │              │  │      
    │  │              │  └─mybatis
    │  │              │          ClusterDataSourceConfig.java
    │  │              │          MasterDataSourceConfig.java
    │  │              │          
    │  │              ├─dao
    │  │              │  ├─jpa
    │  │              │  │  ├─primary
    │  │              │  │  │      UserRepository.java
    │  │              │  │  │      
    │  │              │  │  └─secondary
    │  │              │  │          CityRepository.java
    │  │              │  │          
    │  │              │  └─mybatis
    │  │              │      ├─cluster
    │  │              │      │      CityDao.java
    │  │              │      │      
    │  │              │      └─master
    │  │              │              UserDao.java
    │  │              │              
    │  │              ├─entity
    │  │              │  ├─jpa
    │  │              │  │  ├─primary
    │  │              │  │  │      UserEntity.java
    │  │              │  │  │      
    │  │              │  │  └─secondary
    │  │              │  │          CityEntity.java
    │  │              │  │          
    │  │              │  └─mybatis
    │  │              │      ├─cluster
    │  │              │      │      MybatisCityEntity.java
    │  │              │      │      
    │  │              │      └─master
    │  │              │              MybatisUserEntity.java
    │  │              │              
    │  │              └─service
    │  │                  │  CityService.java
    │  │                  │  UserService.java
    │  │                  │  
    │  │                  └─impl
    │  │                          CityServiceImpl.java
    │  │                          UserServiceImpl.java
    │  │                          
    │  └─resources
    │      │  application.yml
    │      │  
    │      ├─mapper
    │      │  ├─cluster
    │      │  │      city.xml
    │      │  │      
    │      │  └─master
    │      │          user.xml
    │      │          
    │      ├─static
    │      └─templates
    └─test
        └─java
            └─com
                └─developlee
                    └─multipartdatasource
                            MultipartDatasourceApplicationTests.java
                    

从文件树可以看到，对每个数据源要操作的实体类和数据接口都进行隔离。

jpa数据源配置

@Configuration
public class DataSourceConfig {

    @Bean(name = "primaryDataSource")
    @Qualifier("primaryDataSource")
    @Primary
    @ConfigurationProperties(prefix = "spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }

}

JPA-primary主数据源配置

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef = "primaryEntityManagerFactory",
        transactionManagerRef = "primaryTransactionManager",
        basePackages = { PrimaryDataSourceConfig.REPO_PACKAGE } // 设置 Repository 所在位置
        )
public class PrimaryDataSourceConfig {

    static final String REPO_PACKAGE = "com.developlee.multipartdatasource.dao.jpa.primary";
    static final String ENTITY_PACKAGE = "com.developlee.multipartdatasource.entity.jpa.primary";

    @Resource
    @Qualifier("primaryDataSource")
    private DataSource primaryDataSource;

    @Resource
    private JpaProperties jpaProperties;

    @Primary
    @Bean(name = "primaryEntityManager")
    public EntityManager primaryEntityManager(EntityManagerFactoryBuilder builder) {
       return primaryEntityManagerFactory(builder).getObject().createEntityManager();
    }

    @Primary
    @Bean(name = "primaryEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean primaryEntityManagerFactory(EntityManagerFactoryBuilder builder) {
        return builder.dataSource(primaryDataSource)
                .properties(jpaProperties.getHibernateProperties(new HibernateSettings()))
                .persistenceUnit("primaryPersistenceUnit")
                .packages(ENTITY_PACKAGE)
                .build();
    }

    @Primary
    @Bean(name = "primaryTransactionManager")
    public PlatformTransactionManager primaryTransactionManager(EntityManagerFactoryBuilder builder) {
        return new JpaTransactionManager(primaryEntityManagerFactory(builder).getObject());
    }
}

jpa-secondary的从数据源与此类似。

mybatis-master数据源配置

@Configuration
@MapperScan(basePackages = MasterDataSourceConfig.PACKAGE, sqlSessionFactoryRef = "masterSqlSessionFactory")
public class MasterDataSourceConfig {
    // 精确到master目录和其他数据源隔离
    static final String PACKAGE = "com.developlee.multipartdatasource.dao.mybatis.master";
    static final String MAPPER_LOCATION = "classpath:mapper/master/*.xml";

    @Value("${spring.datasource.master.jdbc-url}")
    private String url;

    @Value("${spring.datasource.master.username}")
    private String user;

    @Value("${spring.datasource.master.password}")
    private String password;

    @Value("${spring.datasource.master.driver-class-name}")
    private String driverClass;

    @Bean(name = "masterDataSource")
    public DataSource masterDataSource() {
        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setDriverClassName(driverClass);
        dataSource.setUrl(url);
        dataSource.setUsername(user);
        dataSource.setPassword(password);
        return dataSource;
    }

    @Bean(name = "masterTransactionManager")
    @Primary
    public DataSourceTransactionManager masterTransactionManager() {
        return new DataSourceTransactionManager(masterDataSource());
    }

    @Bean(name = "masterSqlSessionFactory")
    @Primary
    public SqlSessionFactory masterSqlSessionFactory(@Qualifier("masterDataSource") DataSource masterDataSource)
            throws Exception {
        final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(masterDataSource);
        sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources(MasterDataSourceConfig.MAPPER_LOCATION));
        return sessionFactory.getObject();
    }
}

mybatis-cluster数据源配置与此类似。

测试

说明下，业务逻辑代码就不贴了，可在我的github中找到。

@RunWith(SpringRunner.class)
@SpringBootTest
public class MultipartDatasourceApplicationTests {

    @Autowired
    UserServiceImpl userService;

    @Autowired
    CityServiceImpl cityService;


    @Test
    public void contextLoads() {
        UserEntity userEntity = new UserEntity();
        userEntity.setUsername("JPA-Lensen");
        userEntity.setPwd("123456");
        userEntity.setMobile("137****8661");

        CityEntity cityEntity = new CityEntity();
        cityEntity.setUserId(1024L);
        cityEntity.setDescription("JPA-Lensen-Desc");

        MybatisUserEntity mybatisUserEntity = new MybatisUserEntity();
        mybatisUserEntity.setUsername("Mybatis-Lensen");
        mybatisUserEntity.setPwd("654321");
        mybatisUserEntity.setMobile("137****1024");

        MybatisCityEntity mybatisCityEntity = new MybatisCityEntity();
        mybatisCityEntity.setDesc("MyBatis-Lensen-Desc");
        mybatisCityEntity.setUserId(2048L);
        mybatisCityEntity.setId(10L);

        userService.saveUserByJpa(userEntity);
        cityService.saveCityByJpa(cityEntity);
        userService.saveUserByMybatis(mybatisUserEntity);
        cityService.saveCityByMybatis(mybatisCityEntity);

    }

}

截图看看效果

看看数据库的数据
tb_user表

tb_city表

最后： 此项目代码可在我的github.com中找到。
我的个人公众号：developlee的潇洒人生。
关注了也不一定更新，更新就不得了了。