本章概要
- JPA 多数据源
5.4.3 JPA 多数据源
JPA 和 MyBatis 配置多数据源类似，不同的是，JPA 配置时主要提供不同的 LocalContainerEntityManagerFactoryBean 以及事务管理器。
项目目录结构如下：
1. 准备工作
创建 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;
}
}
3. 创建 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());
}
}
4. 创建 Repository
分别在 com.sang.dao1 和 com.sang.dao2 包下创建两个 Repository
UserDao
public interface UserDao extends JpaRepository<User,Integer> {
}
UserDao2
public interface UserDao2 extends JpaRepository<User,Integer> {
}
5. 创建 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，查看数据库，即可看到数据库中的表和数据都已经存在了，如下
