Spring Boot实现多数据源，引爆多层次数据流操作体验

2023-08-14 20:41:53

解锁 Spring Boot 的多数据源功能

1. 多数据源的必要性

想象你正在构建一个复杂的应用程序，该应用程序需要与多个数据库交互。例如，你可能需要管理客户数据、订单信息和产品库存。将这些数据分散在不同的数据源中不仅有利于组织，还能提高性能和可扩展性。

2. Spring Boot 实现多数据源

Spring Boot 提供了一种简便的方法来配置多数据源。只需在 application.properties 文件中定义多个数据源，然后使用 @DataSource 注释指定要注入哪个数据源。

# 数据源 1
spring.datasource.primary.url=jdbc:mysql://localhost:3306/db1
spring.datasource.primary.username=user1
spring.datasource.primary.password=password1

# 数据源 2
spring.datasource.secondary.url=jdbc:oracle://localhost:1521/db2
spring.datasource.secondary.username=user2
spring.datasource.secondary.password=password2

3. SqlSessionFactory 配置

要使用 SqlSessionFactory 配置多数据源，你需要为每个数据源创建一个 SqlSessionFactoryBean 对象。此对象负责创建和管理连接池。

@Bean
public SqlSessionFactoryBean primarySqlSessionFactory() {
    SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
    sqlSessionFactoryBean.setDataSource(dataSource1());
    return sqlSessionFactoryBean;
}

@Bean
public SqlSessionFactoryBean secondarySqlSessionFactory() {
    SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
    sqlSessionFactoryBean.setDataSource(dataSource2());
    return sqlSessionFactoryBean;
}

4. 注解方式配置

你还可以使用注释方式配置多数据源。使用 @DataSource 注释指定数据源，并使用 @Repository 注释指定要使用哪个数据源。

@Repository
@DataSource("secondaryDataSource")
public class UserRepository {

    // ...
}

5. 实际操作步骤

在 application.properties 文件中配置数据源。
在 SqlSessionFactoryBean 中配置数据源。
使用 @DataSource 注释指定数据源。
使用 @Repository 注释指定要使用哪个数据源。

示例代码

@SpringBootApplication
public class App {

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

    @Bean
    @Qualifier("dataSource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create()
                .url("jdbc:h2:mem:test1")
                .username("sa")
                .password("")
                .build();
    }

    @Bean
    @Qualifier("dataSource2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create()
                .url("jdbc:h2:mem:test2")
                .username("sa")
                .password("")
                .build();
    }

    @Bean
    public SqlSessionFactoryBean primarySqlSessionFactory(@Qualifier("dataSource1") DataSource dataSource) {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        return sqlSessionFactoryBean;
    }

    @Bean
    public SqlSessionFactoryBean secondarySqlSessionFactory(@Qualifier("dataSource2") DataSource dataSource) {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        return sqlSessionFactoryBean;
    }

    @Repository
    @DataSource("dataSource1")
    public class UserRepository {

        @Autowired
        private JdbcTemplate jdbcTemplate;

        public List<User> findAll() {
            return jdbcTemplate.query("select * from users",
                    (rs, rowNum) -> new User(rs.getLong("id"),
                            rs.getString("name")));
        }
    }
}