SpringBoot中實現(xiàn)多數(shù)據(jù)源連接和切換的方案

更新時間：2024年11月04日 08:26:15 作者：碼到三十五

在Spring Boot中,通過AbstractRoutingDataSource實現(xiàn)多數(shù)據(jù)源連接是一種常見的做法,這種技術(shù)允許你在運行時動態(tài)地切換數(shù)據(jù)源,從而支持對多個數(shù)據(jù)庫的操作,本文給大家介紹了SpringBoot中實現(xiàn)多數(shù)據(jù)源連接和切換的方案,需要的朋友可以參考下

引言
1. 添加依賴
2. 配置數(shù)據(jù)源屬性
3. 創(chuàng)建數(shù)據(jù)源配置類
4. 創(chuàng)建自定義數(shù)據(jù)源路由類
5. 創(chuàng)建數(shù)據(jù)源上下文持有者
6. 配置多數(shù)據(jù)源
7. 使用AOP切換數(shù)據(jù)源
8. 使用自定義注解切換數(shù)據(jù)源

引言

在Spring Boot中，通過AbstractRoutingDataSource實現(xiàn)多數(shù)據(jù)源連接是一種常見的做法。這種技術(shù)允許你在運行時動態(tài)地切換數(shù)據(jù)源，從而支持對多個數(shù)據(jù)庫的操作。Spring Boot中配置和使用AbstractRoutingDataSource來實現(xiàn)多數(shù)據(jù)源連接。

1. 添加依賴

pom.xml文件的依賴，比如Spring Data JPA和數(shù)據(jù)庫驅(qū)動：

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-jpa</artifactId>
    </dependency>
    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-java</artifactId>
    </dependency>
    <!-- 其他依賴 -->
</dependencies>

2. 配置數(shù)據(jù)源屬性

在application.yml或application.properties中配置多個數(shù)據(jù)源的信息。例如：

spring:
  datasource:
    primary:
      url: jdbc:mysql://localhost:3306/primary_db
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver
    secondary:
      url: jdbc:mysql://localhost:3306/secondary_db
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver

3. 創(chuàng)建數(shù)據(jù)源配置類

創(chuàng)建兩個數(shù)據(jù)源配置類，分別用于配置主數(shù)據(jù)源和次數(shù)據(jù)源。

@Configuration
public class DataSourceConfig {

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

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

4. 創(chuàng)建自定義數(shù)據(jù)源路由類

擴展AbstractRoutingDataSource類，并根據(jù)上下文信息動態(tài)返回數(shù)據(jù)源。

public class DynamicRoutingDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSourceType();
    }
}

5. 創(chuàng)建數(shù)據(jù)源上下文持有者

用于在運行時設(shè)置和獲取當(dāng)前的數(shù)據(jù)源類型。

public class DataSourceContextHolder {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    public static void setDataSourceType(String dataSourceType) {
        contextHolder.set(dataSourceType);
    }

    public static String getDataSourceType() {
        return contextHolder.get();
    }

    public static void clearDataSourceType() {
        contextHolder.remove();
    }
}

6. 配置多數(shù)據(jù)源

將數(shù)據(jù)源配置到Spring上下文中，并指定默認的數(shù)據(jù)源。

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
    basePackages = "com.example.repository",
    entityManagerFactoryRef = "entityManagerFactory",
    transactionManagerRef = "transactionManager"
)
public class DataSourceRoutingConfig {

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

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

    @Bean
    public DataSource dataSource() {
        DynamicRoutingDataSource routingDataSource = new DynamicRoutingDataSource();
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("primary", primaryDataSource);
        targetDataSources.put("secondary", secondaryDataSource);
        routingDataSource.setTargetDataSources(targetDataSources);
        routingDataSource.setDefaultTargetDataSource(primaryDataSource);
        return routingDataSource;
    }

    @Bean
    public LocalContainerEntityManagerFactoryBean entityManagerFactory(
            EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(dataSource())
                .packages("com.example.entity")
                .persistenceUnit("multiple-pu")
                .build();
    }

    @Bean
    public PlatformTransactionManager transactionManager(EntityManagerFactory entityManagerFactory) {
        return new JpaTransactionManager(entityManagerFactory);
    }
}

7. 使用AOP切換數(shù)據(jù)源

通過AOP在方法執(zhí)行前設(shè)置數(shù)據(jù)源類型，并在方法執(zhí)行后清除。

@Aspect
@Component
public class DataSourceAspect {

    @Before("@annotation(targetDataSource)")
    public void changeDataSource(JoinPoint point, TargetDataSource targetDataSource) throws Throwable {
        DataSourceContextHolder.setDataSourceType(targetDataSource.value());
    }

    @After("@annotation(targetDataSource)")
    public void clearDataSource(JoinPoint point, TargetDataSource targetDataSource) {
        DataSourceContextHolder.clearDataSourceType();
    }
}

自定義注解TargetDataSource：

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface TargetDataSource {
    String value();
}

8. 使用自定義注解切換數(shù)據(jù)源

在需要使用特定數(shù)據(jù)源的方法或類上使用@TargetDataSource注解。

@Service
public class UserService {

    @Autowired
    private UserRepository userRepository;

    @TargetDataSource("primary")
    public User findUserById(Long id) {
        return userRepository.findById(id).orElse(null);
    }

    @TargetDataSource("secondary")
    public User findUserBySecondaryId(Long id) {
        // 假設(shè)secondary數(shù)據(jù)庫有一個類似的表結(jié)構(gòu)
        return userRepository.findById(id).orElse(null);
    }
}

到此這篇關(guān)于SpringBoot中實現(xiàn)多數(shù)據(jù)源連接和切換的方案的文章就介紹到這了,更多相關(guān)SpringBoot多數(shù)據(jù)源連接和切換內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: