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

1.概述

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

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

2. 默认行为

让我们回忆一下在 Spring Boot 中在application.yml中声明数据源的样子:

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;

// ...

}

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

3.扩展默认值

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

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

@Configuration

public class TodoDatasourceConfiguration {

@Bean

@ConfigurationProperties("spring.datasource.todos")

public DataSourceProperties todosDataSourceProperties() {

return new DataSourceProperties();

}

}

@Configuration

public class TopicDatasourceConfiguration {

@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 时,我们希望使用如下所示的存储库,其中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(dataSource)

.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 配置多个数据源。

我们发现我们需要一些配置,并且偏离标准时可能会出现陷阱,但最终这是可能的。

如果您喜欢此文章,请收藏、点赞、评论,谢谢,祝您快乐每一天。