在Spring Boot中配置双数据源的诀窍

后端

2023-12-18 14:59:39

管理 Spring Boot 应用程序中的多个数据源

概述

在企业级应用程序中，经常需要访问多个数据源，例如：

不同的数据存储类型（如 MySQL 和 MongoDB）
不同应用程序的特定数据库
分布在不同地理位置的数据库

Spring Boot 提供了灵活的方法来配置和管理多个数据源，使开发人员可以轻松处理此类场景。

配置数据源

应用程序属性文件

创建一个应用程序属性文件（通常为 application.properties），并添加以下配置以指定数据源属性：

spring.datasource.default.url=jdbc:mysql://localhost:3306/default
spring.datasource.default.username=root
spring.datasource.default.password=root
spring.datasource.secondary.url=jdbc:mysql://localhost:3306/secondary
spring.datasource.secondary.username=root
spring.datasource.secondary.password=root

default : 默认数据源
secondary : 辅助数据源

配置应用程序类

在应用程序类中，使用 DataSourceBuilder 创建数据源 bean：

@SpringBootApplication
public class Application {

    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }

    @Bean
    public DataSource defaultDataSource() {
        return DataSourceBuilder.create()
                .url("jdbc:mysql://localhost:3306/default")
                .username("root")
                .password("root")
                .build();
    }

    @Bean
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create()
                .url("jdbc:mysql://localhost:3306/secondary")
                .username("root")
                .password("root")
                .build();
    }
}

使用数据源

通过自动装配机制，在应用程序类中注入数据源 bean：

@SpringBootApplication
public class Application {

    @Autowired
    private DataSource defaultDataSource;

    @Autowired
    private DataSource secondaryDataSource;

    // ...
}

自动配置

Spring Boot 通过 spring-boot-autoconfigure 模块提供自动配置功能。添加相应的依赖项（例如 spring-boot-starter-jdbc）以启用自动配置。

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>

JdbcTemplate

JdbcTemplate 是 Spring Boot 中用于与数据库交互的便捷类：

查询

JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

List<User> users = jdbcTemplate.query("SELECT * FROM users", new RowMapper<User>() {
    @Override
    public User mapRow(ResultSet resultSet, int rowNum) throws SQLException {
        return new User(resultSet.getLong("id"), resultSet.getString("name"), resultSet.getString("email"));
    }
});

更新

JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

int rowCount = jdbcTemplate.update("UPDATE users SET name = ? WHERE id = ?", "John Doe", 1);

Spring Data JPA

Spring Data JPA 是一个对象-关系映射框架，它简化了与数据库的交互：

实体类

@Entity
public class User {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private String name;

    private String email;

    // ...
}

仓库接口

public interface UserRepository extends JpaRepository<User, Long> {}

查询

@Autowired
private UserRepository userRepository;

List<User> users = userRepository.findAll();

更新

@Autowired
private UserRepository userRepository;

User user = new User();
user.setName("John Doe");
user.setEmail("john.doe@example.com");

userRepository.save(user);