spring data jpa 多数据源

本文介绍如何在Spring Boot项目中配置并使用Spring Data JPA实现多数据源,包括配置文件设置、Java配置类编写等步骤,并解决了@Autowired默认按类型注入导致的数据源选择问题。
由于业务需要把原来项目的单数据源变成了两个数据源,方式和mybatis的差不多但是还是要在这里记录一下,顺便记录一下花了一个多小时解决的坑。首先看一下配置文件:
spring.datasource.primary.url=ip1
spring.datasource.primary.username=username1
spring.datasource.primary.password=password1
spring.datasource.primary.driver-class-name=com.mysql.jdbc.Driver

spring.datasource.secondary.url=ip2
spring.datasource.secondary.username=username2
spring.datasource.secondary.password=password2
spring.datasource.secondary.driver-class-name=com.mysql.jdbc.Driver
三个Java文件:
@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();
    }
}


@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef="entityManagerFactoryPrimary",
        transactionManagerRef="transactionManagerPrimary",
        basePackages= { "com.txxs.dao1" }) //设置Repository所在位置
public class PrimaryConfig {


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


    @Autowired
    private JpaProperties jpaProperties;


    @Primary
    @Bean(name = "entityManagerPrimary")
    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
        return entityManagerFactoryPrimary(builder).getObject().createEntityManager();
    }


    @Primary
    @Bean(name = "entityManagerFactoryPrimary")
    public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary (EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(primaryDataSource)
                .properties(getVendorProperties(primaryDataSource))
                .packages("com.txxs.po1") //设置实体类所在位置
                .persistenceUnit("primaryPersistenceUnit")
                .build();
    }


    private Map<String, String> getVendorProperties(DataSource dataSource) {
        return jpaProperties.getHibernateProperties(dataSource);
    }


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


}


@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef="entityManagerFactorySecondary",
        transactionManagerRef="transactionManagerSecondary",
        basePackages= { "com.txxs.dao2" }) //设置Repository所在位置
public class SecondaryConfig {


    @Autowired
    @Qualifier("secondaryDataSource")
    private DataSource secondaryDataSource;


    @Autowired
    private JpaProperties jpaProperties;


    @Bean(name = "entityManagerSecondary")
    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
        return entityManagerFactorySecondary(builder).getObject().createEntityManager();
    }


    @Bean(name = "entityManagerFactorySecondary")
    public LocalContainerEntityManagerFactoryBean entityManagerFactorySecondary (EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(secondaryDataSource)
                .properties(getVendorProperties(secondaryDataSource))
                .packages("com.txxs.po2") //设置实体类所在位置
                .persistenceUnit("secondaryPersistenceUnit")
                .build();
    }


    private Map<String, String> getVendorProperties(DataSource dataSource) {
        return jpaProperties.getHibernateProperties(dataSource);
    }


    @Bean(name = "transactionManagerSecondary")
    PlatformTransactionManager transactionManagerSecondary(EntityManagerFactoryBuilder builder) {
        return new JpaTransactionManager(entityManagerFactorySecondary(builder).getObject());
    }


}
这样就可以实现spring data jpa的多数据源了,还是很简单的,有个问题是在实现的过程中总是会去加载第一个数据源,而不加载第二个数据源,报的错误如下:
14:49:22.316 [http-nio-9102-exec-2] INFO  c.txxs.task.ask - com.Aspect#aspect:50 - action=GET/v1/
Hibernate: select t.* from table1 t where ENAME like 'like%'
14:49:26.466 [http-nio-9102-exec-2] WARN  o.h.e.jdbc.spi.SqlExceptionHelper - SQL Error: 1146, SQLState: 42S02
14:49:26.467 [http-nio-9102-exec-2] ERROR o.h.e.jdbc.spi.SqlExceptionHelper - Table 'datasource.table1' doesn't exist
table1是数据源2中的表,怎么解决呢,添加如下注解,原因是@Autowired默认按照类型进行注入,加上@Qualifier后spring注解按照里边的字符串的名字进行注入了
    @Qualifier("secondaryDataSource")
    private DataSource secondaryDataSource;
好的,下面是一个简单的 Spring Data JPA 多数据源的实例: 1. 首先,需要在 pom.xml 文件中引入 Spring Data JPA 的依赖: ```xml <dependency> <groupId>org.springframework.data</groupId> <artifactId>spring-data-jpa</artifactId> <version>2.4.2</version> </dependency> ``` 2. 在 application.yml 文件中配置多个数据源: ```yml spring: datasource: primary: url: jdbc:mysql://localhost:3306/primary_db username: root password: root secondary: url: jdbc:mysql://localhost:3306/secondary_db username: root password: root ``` 3. 创建两个 DataSource Bean: ```java @Configuration public class DataSourceConfig { @Primary @Bean(name = "primaryDataSource") @ConfigurationProperties(prefix = "spring.datasource.primary") public DataSource primaryDataSource() { return DataSourceBuilder.create().build(); } @Bean(name = "secondaryDataSource") @ConfigurationProperties(prefix = "spring.datasource.secondary") public DataSource secondaryDataSource() { return DataSourceBuilder.create().build(); } } ``` 4. 创建 JPA EntityManagerFactory Bean: ```java @Configuration @EnableJpaRepositories( basePackages = "com.example.primary.repository", entityManagerFactoryRef = "primaryEntityManagerFactory", transactionManagerRef = "primaryTransactionManager" ) public class PrimaryDataSourceConfig { @Autowired @Qualifier("primaryDataSource") private DataSource primaryDataSource; @Bean(name = "primaryEntityManagerFactory") public LocalContainerEntityManagerFactoryBean primaryEntityManagerFactory(EntityManagerFactoryBuilder builder) { return builder .dataSource(primaryDataSource) .packages("com.example.primary.entity") .persistenceUnit("primary") .build(); } @Bean(name = "primaryTransactionManager") public PlatformTransactionManager primaryTransactionManager( @Qualifier("primaryEntityManagerFactory") EntityManagerFactory entityManagerFactory) { return new JpaTransactionManager(entityManagerFactory); } } ``` 5. 创建第二个 JPA EntityManagerFactory Bean: ```java @Configuration @EnableJpaRepositories( basePackages = "com.example.secondary.repository", entityManagerFactoryRef = "secondaryEntityManagerFactory", transactionManagerRef = "secondaryTransactionManager" ) public class SecondaryDataSourceConfig { @Autowired @Qualifier("secondaryDataSource") private DataSource secondaryDataSource; @Bean(name = "secondaryEntityManagerFactory") public LocalContainerEntityManagerFactoryBean secondaryEntityManagerFactory(EntityManagerFactoryBuilder builder) { return builder .dataSource(secondaryDataSource) .packages("com.example.secondary.entity") .persistenceUnit("secondary") .build(); } @Bean(name = "secondaryTransactionManager") public PlatformTransactionManager secondaryTransactionManager( @Qualifier("secondaryEntityManagerFactory") EntityManagerFactory entityManagerFactory) { return new JpaTransactionManager(entityManagerFactory); } } ``` 6. 创建两个 Repository: ```java @Repository public interface PrimaryRepository extends JpaRepository<PrimaryEntity, Long> { } @Repository public interface SecondaryRepository extends JpaRepository<SecondaryEntity, Long> { } ``` 这样,就完成了 Spring Data JPA 多数据源的配置。在使用时,只需要在 Repository 上指定对应的数据源即可。例如: ```java @Service public class MyService { @Autowired private PrimaryRepository primaryRepository; @Autowired private SecondaryRepository secondaryRepository; public void doSomething() { // 使用 primary 数据源 List<PrimaryEntity> primaryEntities = primaryRepository.findAll(); // 使用 secondary 数据源 List<SecondaryEntity> secondaryEntities = secondaryRepository.findAll(); } } ``` 以上就是一个简单的 Spring Data JPA 多数据源的实例,希望能够帮助到你。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值