本文是 Spring Boot 整合資料持久化方案的最後一篇，主要和大夥來聊聊 Spring Boot 整合 Jpa 多資料來源問題。在 Spring Boot 整合JbdcTemplate 多資料來源、Spring Boot 整合 MyBatis 多資料來源以及 Spring Boot 整合 Jpa 多資料來源這三個知識點中，整合 Jpa 多資料來源算是最複雜的一種，也是很多人在配置時最容易出錯的一種。本文大夥就跟著鬆哥的教程，一步一步整合 Jpa 多資料來源。

工程建立

首先是建立一個 Spring Boot 工程，建立時新增基本的 Web、Jpa 以及 MySQL 依賴，如下：

建立完成後，新增 Druid 依賴，這裡和前文的要求一樣，要使用專為 Spring Boot 打造的 Druid，大夥可能發現了，如果整合多資料來源一定要使用這個依賴，因為這個依賴中才有 DruidDataSourceBuilder，最後還要記得鎖定資料庫依賴的版本，因為可能大部分人用的還是 5.x 的 MySQL 而不是 8.x。完整依賴如下：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-jpa</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid-spring-boot-starter</artifactId>
    <version>1.1.10</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>5.1.28</version>
    <scope>runtime</scope>
</dependency>
 

如此之後，工程就建立成功了。

基本配置

在基本配置中，我們首先來配置多資料來源基本資訊以及 DataSource，首先在 application.properties 中新增如下配置資訊：

#  資料來源一
spring.datasource.one.username=root
spring.datasource.one.password=root
spring.datasource.one.url=jdbc:mysql:///test01?useUnicode=true&characterEncoding=UTF-8
spring.datasource.one.type=com.alibaba.druid.pool.DruidDataSource

#  資料來源二
spring.datasource.two.username=root
spring.datasource.two.password=root
spring.datasource.two.url=jdbc:mysql:///test02?useUnicode=true&characterEncoding=UTF-8
spring.datasource.two.type=com.alibaba.druid.pool.DruidDataSource

# Jpa配置
spring.jpa.properties.database=mysql
spring.jpa.properties.show-sql=true
spring.jpa.properties.database-platform=mysql
spring.jpa.properties.hibernate.ddl-auto=update
spring.jpa.properties.hibernate.dialect=org.hibernate.dialect.MySQL57Dialect
 

這裡 Jpa 的配置和上文相比 key 中多了 properties，多資料來源的配置和前文一致，然後接下來配置兩個 DataSource，如下：

@Configuration
public class DataSourceConfig {
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.one")
    @Primary
    DataSource dsOne() {
        return DruidDataSourceBuilder.create().build();
    }
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.two")
    DataSource dsTwo() {
        return DruidDataSourceBuilder.create().build();
    }
}

這裡的配置和前文的多資料來源配置基本一致，但是注意多了一個在 Spring 中使用較少的註解 @Primary，這個註解一定不能少，否則在專案啟動時會出錯，@Primary 表示當某一個類存在多個例項時，優先使用哪個例項。

好了，這樣，DataSource 就有了。

多資料來源配置

接下來配置 Jpa 的基本資訊，這裡兩個資料來源，我分別在兩個類中來配置，先來看第一個配置：

@Configuration
@EnableJpaRepositories(basePackages = "org.javaboy.jpa.dao",entityManagerFactoryRef = "localContainerEntityManagerFactoryBeanOne",transactionManagerRef = "platformTransactionManagerOne")
public class JpaConfigOne {
    @Autowired
    @Qualifier(value = "dsOne")
    DataSource dsOne;
    @Autowired
    JpaProperties jpaProperties;
    @Bean
    @Primary
    LocalContainerEntityManagerFactoryBean localContainerEntityManagerFactoryBeanOne(EntityManagerFactoryBuilder builder) {
        return builder.dataSource(dsOne)
                .packages("org.javaboy.jpa.model")
                .properties(jpaProperties.getProperties())
                .persistenceUnit("pu1")
                .build();
    }
    @Bean
    PlatformTransactionManager platformTransactionManagerOne(EntityManagerFactoryBuilder builder) {
        LocalContainerEntityManagerFactoryBean factoryBeanOne = localContainerEntityManagerFactoryBeanOne(builder);
        return new JpaTransactionManager(factoryBeanOne.getObject());
    }
}

首先這裡注入 dsOne，再注入 JpaProperties，JpaProperties 是系統提供的一個例項，裡邊的資料就是我們在 application.properties 中配置的 jpa 相關的配置。然後我們提供兩個 Bean，分別是 LocalContainerEntityManagerFactoryBean 和 PlatformTransactionManager 事務管理器，不同於 MyBatis 和 JdbcTemplate，在 Jpa 中，事務一定要配置。在提供 LocalContainerEntityManagerFactoryBean 的時候，需要指定 packages，這裡的 packages 指定的包就是這個資料來源對應的實體類所在的位置，另外在這裡配置類上通過 @EnableJpaRepositories 註解指定 dao 所在的位置，以及 LocalContainerEntityManagerFactoryBean 和 PlatformTransactionManager 分別對應的引用的名字。

好了，這樣第一個就配置好了，第二個基本和這個類似，主要有幾個不同點：

• dao 的位置不同
• persistenceUnit 不同
• 相關 bean 的名稱不同

注意實體類可以共用。

程式碼如下：

@Configuration
@EnableJpaRepositories(basePackages = "org.javaboy.jpa.dao2",entityManagerFactoryRef = "localContainerEntityManagerFactoryBeanTwo",transactionManagerRef = "platformTransactionManagerTwo")
public class JpaConfigTwo {
    @Autowired
    @Qualifier(value = "dsTwo")
    DataSource dsTwo;
    @Autowired
    JpaProperties jpaProperties;
    @Bean
    LocalContainerEntityManagerFactoryBean localContainerEntityManagerFactoryBeanTwo(EntityManagerFactoryBuilder builder) {
        return builder.dataSource(dsTwo)
                .packages("org.javaboy.jpa.model")
                .properties(jpaProperties.getProperties())
                .persistenceUnit("pu2")
                .build();
    }
    @Bean
    PlatformTransactionManager platformTransactionManagerTwo(EntityManagerFactoryBuilder builder) {
        LocalContainerEntityManagerFactoryBean factoryBeanTwo = localContainerEntityManagerFactoryBeanTwo(builder);
        return new JpaTransactionManager(factoryBeanTwo.getObject());
    }
}

接下來，在對應位置分別提供相關的實體類和 dao 即可，資料來源一的 dao 如下：

package org.javaboy.jpa.dao;
public interface UserDao extends JpaRepository<User,Integer> {
    List<User> getUserByAddressEqualsAndIdLessThanEqual(String address, Integer id);
    @Query(value = "select * from t_user where id=(select max(id) from t_user)",nativeQuery = true)
    User maxIdUser();
}

資料來源二的 dao 如下：

package org.javaboy.jpa.dao2;
public interface UserDao2 extends JpaRepository<User,Integer> {
    List<User> getUserByAddressEqualsAndIdLessThanEqual(String address, Integer id);

    @Query(value = "select * from t_user where id=(select max(id) from t_user)",nativeQuery = true)
    User maxIdUser();
}

共同的實體類如下：

package org.javaboy.jpa.model;
@Entity(name = "t_user")
public class User {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;
    private String username;
    private String address;
    //省略getter/setter
}

到此，所有的配置就算完成了，接下來就可以在 Service 中注入不同的 UserDao，不同的 UserDao 操作不同的資料來源。

其實整合 Jpa 多資料來源也不算難，就是有幾個細節問題，這些細節問題解決，其實前面介紹的其他多資料來源整個都差不多。

好了，本文就先介紹到這裡。

相關案例已經上傳到 GitHub，歡迎小夥伴們們下載：https://github.com/lenve/javaboy-code-samples

掃碼關注鬆哥，公眾號後臺回覆 2TB，獲取鬆哥獨家 超2TB 免費 Java 學習乾貨

相關推薦