spring-boot jpa 配置两个数据源

spring-boot jpa 配置两个数据源

内容简介

1、配置两个mysql数据库
2、配置一个mysql,一个h2内存数据库

如何配置两个mysql数据库

1、配置文件
首先，连接数据的配置文件需要改变，即增加第二个数据库的url，username,password
下面给出我的测试配置，仅供参考:

spring:
    datasource:
        primary:
            driver-class-name: com.mysql.jdbc.Driver
            url: jdbc:mysql://localhost:3306/songsf_dev?characterEncoding=utf-8
            username: root
            password: 123456
        secondary:
            driver-class-name: com.mysql.jdbc.Driver
            url: jdbc:mysql://localhost:3306/test?characterEncoding=utf-8
            username: root
            password: 123456

2、数据源配置类
需要个数据源配置类，来生成数据源一、数据源二的dataSource，并命名为不同的bean name

@Configuration
public class DataSourceConfig {
    @Bean(name = "primaryDataSource")
    @Qualifier("primaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    @Primary
    @ConfigurationProperties(prefix="spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }
}

这里我们声明了DataSourceConfig类用于生成我们需要的dataSource，@ConfigurationProperties注解是跟数据库连接的相关配置。通过前缀可以找到 相关的配置项。@Bean name使spring-boot框架可以识别出不同DataSource所对应的数据库配置。
3、接下来我们需要实现JpaRepositories相关的管理类，用于关联我们的数据库访问层文件。
（1）第一个数据源的管理类

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

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

    @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.songsf.learn.domain.p") //设置实体类所在位置
                .persistenceUnit("primaryPersistenceUnit")
                .build();
    }

    @Autowired
    private JpaProperties jpaProperties;

    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());
    }

}

(2)第二个数据源的管理类

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

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

    @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.songsf.learn.domain.s") //设置实体类所在位置
                .persistenceUnit("secondaryPersistenceUnit")
                .build();
    }

    @Autowired
    private JpaProperties jpaProperties;

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

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

}

配置中的backPackages是数据库访问层代码包，主要是加了@Entity注解的bean类，和集成了JpaRepository的dao层文件 这里就不在做过的赘述了。

配置一个mysql,一个h2内存数据库

其实我们前边所做的大量工作，已经为第二个h2数据库配置做好了铺垫。只需要将配置文件中的第二个数据库连接配置改一下即可。

spring:
  datasource:
    primary:
      driver-class-name: com.mysql.jdbc.Driver
      url: jdbc:mysql://localhost:3306/songsf_dev?characterEncoding=utf-8
      username: root
      password: 123456
    secondary:
      url: jdbc:h2:file:D:\\test\\db\\testdb
      username: sa
      password: sa

h2数据库是纯java编写，一般用于嵌入式数据库。我的做法是将其集成到spring-boot工程中，作为内存数据库来用。用于替代缓存。
操作起来跟一般数据库一样，基于内存，访问速度上会比数据库快。而且缓存内容不是常规的key-value键值对，可以有多个字段，查询起来也比较 方便。
测试代码地址： https://github.com/learnerfan/spring-boot-restful.git