Spring Boot整合持久層之JPA多資料來源

2022-08-13 18:01:32

JPA 多資料來源

JPA 和 MyBatis 設定多資料來源類似，不同的是，JPA 設定時主要提供不同的 LocalContainerEntityManagerFactoryBean 以及事務管理器。

專案目錄結構如下：

準備工作

建立 Spring Boot Web 專案，新增如下依賴：

<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>
  <scope>runtime</scope>
</dependency>

application.properties 設定

注意：這裡的設定與設定單獨的 JPA 有區別，因為在後文的設定中要從 JpaProperties 中的 getProperties 方法中獲取所有的 JPA 相關設定，因此這裡的屬性字首都是 spring.jpa.properties。

server.port=8081
# 資料來源1
spring.datasource.one.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.one.url=jdbc:mysql://localhost:3306/chapter05-1
spring.datasource.one.username=root
spring.datasource.one.password=root
# 資料來源2
spring.datasource.two.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.two.url=jdbc:mysql://localhost:3306/chapter05-2
spring.datasource.two.username=root
spring.datasource.two.password=root
# JPA 設定
spring.jpa.properties.hibernate.dialect=org.hibernate.dialect.MySQL57InnoDBDialect
spring.jpa.properties.database=mysql
spring.jpa.properties.hibernate.hbm2ddl.auto=update
spring.jpa.properties.show-sql= true

資料來源設定 DataSourceConfig

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

實體類 User

@Entity(name = "t_user")
public class User {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;
    private String name;
    private String gender;
    private Integer age;
    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", name='" + name + ''' +
                ", gender='" + gender + ''' +
                ", age=" + age +
                '}';
    }
    public Integer getId() {
        return id;
    }
    public void setId(Integer id) {
        this.id = id;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public String getGender() {
        return gender;
    }
    public void setGender(String gender) {
        this.gender = gender;
    }
    public Integer getAge() {
        return age;
    }
    public void setAge(Integer age) {
        this.age = age;
    }
}

建立 JPA 設定

接下來是核心設定，根據兩個設定好的資料來源建立兩個不同的 JPA 設定

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = "com.sang.dao1",
        entityManagerFactoryRef = "entityManagerFactoryBeanOne",
        transactionManagerRef = "platformTransactionManagerOne")
public class JpaConfigOne {
    @Resource(name = "dsOne")
    DataSource dsOne;
    @Autowired
    JpaProperties jpaProperties;
    @Bean
    @Primary
    LocalContainerEntityManagerFactoryBean entityManagerFactoryBeanOne(
            EntityManagerFactoryBuilder builder) {
        return builder.dataSource(dsOne)
                .properties(jpaProperties.getProperties())
                .packages("com.sang.model")
                .persistenceUnit("pu1")
                .build();
    }
    @Bean
    PlatformTransactionManager platformTransactionManagerOne(
            EntityManagerFactoryBuilder builder) {
        LocalContainerEntityManagerFactoryBean factoryOne = entityManagerFactoryBeanOne(builder);
        return new JpaTransactionManager(factoryOne.getObject());
    }
}

程式碼解釋：

使用 @EnableJpaRepositories 註解進行 JPA 設定，該註解中主要設定三個屬性：basePackages、entityManagerFactoryRef、transactionManagerRef。其中 basePackages 用來指定 Repository 所在的位置，entityManagerFactoryRef 用來指定實體類管理工廠 Bean 的名稱，transactionManagerRef 用來指定事務管理器的參照名稱，這裡的參照名稱就是 JpaConfigOne 類中註冊的 Bean 的名稱（預設的 Bean 名稱為方法名）
建立 LocalContainerEntityManagerFactoryBean ，該 Bean 將用來提供 EntityManager 範例，在該類的建立過程中，首先設定資料來源，然後設定 JPA 相關設定（JpaProperties 由系統自動載入），再設定實體類所在的位置，最後設定持久化單位名，若專案中只有一個 EntityManagerFactory，則 persistenceUnit 可以省略掉，若有多個，則必須明確指定持久化單元名
由於專案中會提供兩個 LocalContainerEntityManagerFactoryBean 範例，@Primary 註解表示當存在多個 LocalContainerEntityManagerFactoryBean 範例時，該範例將被優先使用
platformTransactionManagerOne() 方法表示建立一個事務管理器。JpaTransactionManager 提供對單個 EntityManagerFactory 的事務支援，專門用於解決 JPA 中的事務管理

以下是第二個 JPA 設定

注意：LocalContainerEntityManagerFactoryBean 範例不需要新增 @Primary 註解

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = "com.sang.dao2",
entityManagerFactoryRef = "entityManagerFactoryBeanTwo",
transactionManagerRef = "platformTransactionManagerTwo")
public class JpaConfigTwo {
    @Resource(name = "dsTwo")
    DataSource dsTwo;
    @Autowired
    JpaProperties jpaProperties;
    @Bean
    LocalContainerEntityManagerFactoryBean entityManagerFactoryBeanTwo(
            EntityManagerFactoryBuilder builder) {
        return builder.dataSource(dsTwo)
                .properties(jpaProperties.getProperties())
                .packages("com.sang.model")
                .persistenceUnit("pu2")
                .build();
    }
    @Bean
    PlatformTransactionManager platformTransactionManagerTwo(
            EntityManagerFactoryBuilder builder) {
        LocalContainerEntityManagerFactoryBean factoryTwo = entityManagerFactoryBeanTwo(builder);
        return new JpaTransactionManager(factoryTwo.getObject());
    }
}

建立 Repository

分別在 com.sang.dao1 和 com.sang.dao2 包下建立兩個 Repository

UserDao

public interface UserDao extends JpaRepository<User,Integer> {
}

UserDao2

public interface UserDao2 extends JpaRepository<User,Integer> {
}

建立 Controller

簡便起見，這裡省略掉 Service 層，將 UserDao 直接注入 Controller 中

@RestController
public class UserController {
    @Autowired
    UserDao userDao;
    @Autowired
    UserDao2 userDao2;
    @GetMapping("/test1")
    public void test1() {
        User u1 = new User();
        u1.setAge(55);
        u1.setName("魯迅");
        u1.setGender("男");
        userDao.save(u1);
        User u2 = new User();
        u2.setAge(80);
        u2.setName("泰戈爾");
        u2.setGender("男");
        userDao2.save(u2);
    }
}

http://localhost:8081/test1，檢視資料庫，即可看到資料庫中的表和資料都已經存在了，如下

到此這篇關於Spring Boot整合持久層之JPA多資料來源的文章就介紹到這了,更多相關Spring Boot JPA多資料來源內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！