Spring?Boot整合持久層之JPA多數(shù)據(jù)源

更新時(shí)間：2022年08月13日 11:15:29 作者：一只小熊貓呀

JPA（Java Persistence API）Java 持久化 API，是 Java 持久化的標(biāo)準(zhǔn)規(guī)范，Hibernate 是持久化規(guī)范的技術(shù)實(shí)現(xiàn)，而 Spring Data JPA 是在 Hibernate 基礎(chǔ)上封裝的一款框架

JPA 多數(shù)據(jù)源

JPA 和 MyBatis 配置多數(shù)據(jù)源類似，不同的是，JPA 配置時(shí)主要提供不同的 LocalContainerEntityManagerFactoryBean 以及事務(wù)管理器。

項(xiàng)目目錄結(jié)構(gòu)如下：

準(zhǔn)備工作

創(chuàng)建 Spring Boot Web 項(xiàng)目，添加如下依賴：

<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 配置

注意：這里的配置與配置單獨(dú)的 JPA 有區(qū)別，因?yàn)樵诤笪牡呐渲弥幸獜?JpaProperties 中的 getProperties 方法中獲取所有的 JPA 相關(guān)配置，因此這里的屬性前綴都是 spring.jpa.properties。

server.port=8081
# 數(shù)據(jù)源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
# 數(shù)據(jù)源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

數(shù)據(jù)源配置 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();
    }
}

實(shí)體類 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;
    }
}

創(chuàng)建 JPA 配置

接下來是核心配置，根據(jù)兩個(gè)配置好的數(shù)據(jù)源創(chuàng)建兩個(gè)不同的 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 注解進(jìn)行 JPA 配置，該注解中主要配置三個(gè)屬性：basePackages、entityManagerFactoryRef、transactionManagerRef。其中 basePackages 用來指定 Repository 所在的位置，entityManagerFactoryRef 用來指定實(shí)體類管理工廠 Bean 的名稱，transactionManagerRef 用來指定事務(wù)管理器的引用名稱，這里的引用名稱就是 JpaConfigOne 類中注冊的 Bean 的名稱（默認(rèn)的 Bean 名稱為方法名）
創(chuàng)建 LocalContainerEntityManagerFactoryBean ，該 Bean 將用來提供 EntityManager 實(shí)例，在該類的創(chuàng)建過程中，首先配置數(shù)據(jù)源，然后設(shè)置 JPA 相關(guān)配置（JpaProperties 由系統(tǒng)自動加載），再設(shè)置實(shí)體類所在的位置，最后配置持久化單位名，若項(xiàng)目中只有一個(gè) EntityManagerFactory，則 persistenceUnit 可以省略掉，若有多個(gè)，則必須明確指定持久化單元名
由于項(xiàng)目中會提供兩個(gè) LocalContainerEntityManagerFactoryBean 實(shí)例，@Primary 注解表示當(dāng)存在多個(gè) LocalContainerEntityManagerFactoryBean 實(shí)例時(shí)，該實(shí)例將被優(yōu)先使用
platformTransactionManagerOne() 方法表示創(chuàng)建一個(gè)事務(wù)管理器。JpaTransactionManager 提供對單個(gè) EntityManagerFactory 的事務(wù)支持，專門用于解決 JPA 中的事務(wù)管理

以下是第二個(gè) JPA 配置

注意：LocalContainerEntityManagerFactoryBean 實(shí)例不需要添加 @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());
    }
}

創(chuàng)建 Repository

分別在 com.sang.dao1 和 com.sang.dao2 包下創(chuàng)建兩個(gè) Repository

UserDao

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

UserDao2

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

創(chuàng)建 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("泰戈?duì)?);
        u2.setGender("男");
        userDao2.save(u2);
    }
}

http://localhost:8081/test1，查看數(shù)據(jù)庫，即可看到數(shù)據(jù)庫中的表和數(shù)據(jù)都已經(jīng)存在了，如下