在Spring Boot中配置和使用多个数据源

2023/05/18

1. 概述

Spring Boot应用程序的典型场景是将数据存储在单个关系数据库中。但是我们有时需要访问多个数据库。

在本教程中,我们将学习如何使用Spring Boot配置和使用多个数据源。

要了解如何处理单个数据源,请查看我们对Spring Data JPA的介绍

2. 默认行为

让我们记住在application.yml中声明Spring Boot中的数据源是什么样的:

spring:
    datasource:
        url: ...
        username: ...
        password: ...
        driverClassname: ...

在内部,Spring将这些设置映射到org.springframework.boot.autoconfigure.jdbc.DataSourceProperties的实例。

让我们看一下实现:

@ConfigurationProperties(prefix = "spring.datasource")
public class DataSourceProperties implements BeanClassLoaderAware, InitializingBean {

    // ...

    /**
     * Fully qualified name of the JDBC driver. Auto-detected based on the URL by default.
     */
    private String driverClassName;

    /**
     * JDBC URL of the database.
     */
    private String url;

    /**
     * Login username of the database.
     */
    private String username;

    /**
     * Login password of the database.
     */
    private String password;

    // ...
}

我们应该指出自动将配置属性映射到Java对象的@ConfigurationProperties注解。

3. 扩展默认值

因此,要使用多个数据源,我们需要在Spring的应用程序上下文中声明多个具有不同映射的beans。

我们可以通过使用配置类来做到这一点:

@Configuration
public class TodoDatasourceConfiguration {

    @Bean
    @ConfigurationProperties("spring.datasource.todos")
    public DataSourceProperties todosDataSourceProperties() {
        return new DataSourceProperties();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.topics")
    public DataSourceProperties topicsDataSourceProperties() {
        return new DataSourceProperties();
    }
}

数据源的配置必须如下所示:

spring:
    datasource:
        todos:
            url: ...
            username: ...
            password: ...
            driverClassName: ...
        topics:
            url: ...
            username: ...
            password: ...
            driverClassName: ...

然后我们可以使用DataSourceProperties对象创建数据源:

@Bean
public DataSource todosDataSource() {
    return todosDataSourceProperties()
        .initializeDataSourceBuilder()
        .build();
}

@Bean
public DataSource topicsDataSource() {
    return topicsDataSourceProperties()
        .initializeDataSourceBuilder()
        .build();
}

4. Spring Data JDBC

在使用Spring Data JDBC时,我们还需要为每个DataSource配置一个JdbcTemplate实例:

@Bean
public JdbcTemplate todosJdbcTemplate(@Qualifier("todosDataSource") DataSource dataSource) {
    return new JdbcTemplate(dataSource);
}

@Bean
public JdbcTemplate topicsJdbcTemplate(@Qualifier("topicsDataSource") DataSource dataSource) {
    return new JdbcTemplate(dataSource);
}

然后我们也可以通过指定@Qualifier来使用它们:

@Autowired
@Qualifier("topicsJdbcTemplate")
JdbcTemplate jdbcTemplate;

5. Spring Data JPA

当使用Spring Data JPA时,我们希望使用如下所示的Repository,其中Todo是实体:

public interface TodoRepository extends JpaRepository<Todo, Long> {
}

因此,我们需要为每个数据源声明EntityManager工厂:

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
      basePackageClasses = Todo.class,
      entityManagerFactoryRef = "todosEntityManagerFactory",
      transactionManagerRef = "todosTransactionManager"
)
public class TodoJpaConfiguration {

    @Bean
    public LocalContainerEntityManagerFactoryBean todosEntityManagerFactory(
          Qualifier("todosDataSource") DataSource dataSource,
    EntityManagerFactoryBuilder builder) {
        return builder
              .dataSource(todosDataSource())
              .packages(Todo.class)
              .build();
    }

    @Bean
    public PlatformTransactionManager todosTransactionManager(
          @Qualifier("todosEntityManagerFactory") LocalContainerEntityManagerFactoryBean todosEntityManagerFactory) {
        return new JpaTransactionManager(Objects.requireNonNull(todosEntityManagerFactory.getObject()));
    }
}

让我们看一下我们应该注意的一些限制

我们需要对包进行拆分,以便为每个数据源提供一个@EnableJpaRepositories。

不幸的是,要注入EntityManagerFactoryBuilder,我们需要将其中一个数据源声明为@Primary。

这是因为EntityManagerFactoryBuilder是在org.springframework.boot.autoconfigure.orm.jpa.JpaBaseConfiguration中声明的,并且这个类需要注入单个数据源。通常,框架的某些部分可能不需要配置多个数据源。

6. 配置Hikari连接池

如果我们想配置Hikari,只需要在数据源定义中添加一个@ConfigurationProperties:

@Bean
@ConfigurationProperties("spring.datasource.todos.hikari")
public DataSource todosDataSource() {
    return todosDataSourceProperties()
        .initializeDataSourceBuilder()
        .build();
}

然后我们可以将以下内容添加到application.properties文件中:

spring.datasource.todos.hikari.connectionTimeout=30000
spring.datasource.todos.hikari.idleTimeout=600000
spring.datasource.todos.hikari.maxLifetime=1800000

7. 总结

在本文中,我们学习了如何使用Spring Boot配置多个数据源。

我们看到我们需要一些配置,并且在偏离标准时可能存在陷阱,但最终是可能的。

与往常一样,本教程的完整源代码可在GitHub上获得。

Show Disqus Comments

Post Directory

扫码关注公众号:Taketoday
发送 290992
即可立即永久解锁本站全部文章