数据源切换从此不求人，SpringBoot动态切换数据源

后端

2023-05-13 15:04:52

在 Spring Boot 中实现动态数据源切换

在 Spring Boot 项目中，我们经常需要根据不同的情况使用不同的数据源。例如，我们可能有一个主数据库用于存储一般数据，还有一个次级数据库用于存储敏感或历史数据。为了实现这种灵活性，我们需要一种方法来动态切换数据源。

SpringBoot 提供了 DynamicDataSource 来实现动态数据源切换。该类有一个方法 determineCurrentLookupKey()，它返回一个字符串，指示当前要使用哪个数据源。

重写 determineCurrentLookupKey() 方法

要重写 determineCurrentLookupKey() 方法，我们需要：

获取当前线程的上下文数据 ，例如用户 ID、请求 URL 等。
根据上下文数据判断 要使用哪个数据源。
将数据源名称作为字符串返回 。

示例代码

以下示例代码展示了如何重写 determineCurrentLookupKey() 方法：

import java.util.Map;

import javax.sql.DataSource;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {

    private Map<Object, Object> targetDataSources;

    public DynamicDataSource(Map<Object, Object> targetDataSources) {
        super.setTargetDataSources(targetDataSources);
        super.setDefaultTargetDataSource(targetDataSources.get("primary"));
    }

    @Override
    protected Object determineCurrentLookupKey() {
        // 获取当前线程的用户 ID
        Long userId = getCurrentUserId();

        // 根据用户 ID 判断要使用哪个数据源
        if (userId % 2 == 0) {
            return "even";
        } else {
            return "odd";
        }
    }

    private Long getCurrentUserId() {
        // 此处省略获取用户 ID 的代码
        return 1L;
    }
}

在需要切换数据源的方法上使用 @DataSourceSelector 注解

在需要切换数据源的方法上，我们可以使用 @DataSourceSelector 注解，并指定要切换的数据源名称。

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceSelector {

    String value();
}

示例代码

以下示例代码展示了如何在方法上使用 @DataSourceSelector 注解：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.example.demo.annotation.DataSourceSelector;
import com.example.demo.repository.UserRepository;

@Service
public class UserService {

    @Autowired
    private UserRepository userRepository;

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