Spring Boot多租户架构下控制器层切换租户标识符实现跨Schema查询的方案咨询

嘿，我完全懂你现在遇到的困扰——咱们用请求级租户模式实现了数据库/ Schema隔离的多租户架构，平时用着没问题，但碰到「先查默认Schema，找不到再查另一个Schema」的需求时，在控制器里切换租户上下文怎么都不生效对吧？咱们先理清楚你当前代码里的问题，再聊聊不用手动管理EntityManager或SessionFactory的可行方案。

先说说你现有代码的问题 #

你直接调用TenantContext.setTenantInfo("other_schema")后，又去操作sessionFactory和multiTenantConnectionConfig，这种方式是不对的：

• 当前请求的线程上下文已经绑定了默认租户的事务/连接，直接修改租户标识符不会让后续的personService.getById()自动切换到新的数据源
• 手动操作sessionFactory和连接配置的方式，没有和当前请求的线程上下文绑定，相当于做了无用功

正确的实现思路（无需手动管理EntityManager） #

核心思路是：通过线程安全的租户上下文存储，结合事务传播特性，让后续的查询操作读取新的租户标识符并创建新的数据库连接。

1. 确保租户上下文是线程安全的

首先你的TenantContext必须用ThreadLocal来存储租户标识符，这样每个请求线程的租户信息是隔离的，示例如下：

public class TenantContext {
    private static final ThreadLocal<String> CURRENT_TENANT = new ThreadLocal<>();

    public static void setTenantInfo(String tenantId) {
        CURRENT_TENANT.set(tenantId);
    }

    public static String getCurrentTenant() {
        return CURRENT_TENANT.get();
    }

    public static void clear() {
        CURRENT_TENANT.remove();
    }
}

2. 修改控制器逻辑：切换租户+恢复原租户+新事务查询

在控制器里，我们需要先保存原租户标识符，切换到目标租户后，触发一个新事务的查询（因为现有事务已经绑定了默认租户的连接），最后一定要恢复原租户，避免污染线程池里的后续请求：

public class PersonController {

    private final PersonService personService;

    public PersonController(PersonService personService) {
        this.personService = personService;
    }

    public PersonDto getPerson(@PathVariable String personId) {
        // 保存原租户标识符，用于后续恢复
        String originalTenant = TenantContext.getCurrentTenant();
        
        try {
            // 第一步：查询默认Schema
            return personService.getById(personId);
        } catch (EntityNotFoundException e) { // 用具体的异常，不要捕获所有Exception
            try {
                // 切换到目标租户
                TenantContext.setTenantInfo("other_schema");
                // 开启新事务查询，让操作使用新租户的数据源
                return personService.getByIdInNewTransaction(personId);
            } finally {
                // 必须恢复原租户，避免线程复用导致的租户混乱
                TenantContext.setTenantInfo(originalTenant);
            }
        }
    }
}

3. 在Service层添加新事务方法

在Service里，我们需要一个用Propagation.REQUIRES_NEW注解的方法，这个注解会强制创建一个新的事务，新事务会读取当前TenantContext的新租户标识符，从而获取对应的数据源：

import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;

@Service
public class PersonService {

    private final PersonRepository personRepository;

    public PersonService(PersonRepository personRepository) {
        this.personRepository = personRepository;
    }

    @Transactional(readOnly = true)
    public PersonDto getById(String personId) {
        Person person = personRepository.findById(personId)
                .orElseThrow(() -> new EntityNotFoundException("Person not found in current schema"));
        return convertToDto(person);
    }

    // 开启新事务的查询方法
    @Transactional(readOnly = true, propagation = Propagation.REQUIRES_NEW)
    public PersonDto getByIdInNewTransaction(String personId) {
        // 复用原查询逻辑，新事务会自动使用当前租户的数据源
        return getById(personId);
    }

    // 实体转DTO的工具方法
    private PersonDto convertToDto(Person person) {
        // ... 你的转换逻辑
        return new PersonDto();
    }
}

4. 确保租户标识符解析器正确读取上下文

最后要确认你的TenantIdentifierResolver是从TenantContext读取当前租户的，这样新事务才能正确识别租户：

import org.hibernate.context.spi.CurrentTenantIdentifierResolver;

public class CustomTenantIdentifierResolver implements CurrentTenantIdentifierResolver {

    @Override
    public String resolveCurrentTenantIdentifier() {
        // 优先取上下文的租户，没有则用默认Schema
        String tenant = TenantContext.getCurrentTenant();
        return tenant != null ? tenant : "default_schema";
    }

    @Override
    public boolean validateExistingCurrentSessions() {
        return true;
    }
}

关键注意点 #

• 不要捕获所有Exception：只捕获「数据不存在」的具体异常（比如EntityNotFoundException），避免把数据库连接异常、参数错误等问题当成「数据不存在」去查另一个Schema
• 必须恢复原租户：Servlet容器用线程池复用线程，如果不恢复原租户，下一个请求可能会使用错误的租户标识符
• 依赖事务传播特性：REQUIRES_NEW是核心，它会让新的查询脱离原事务，重新获取基于新租户的数据库连接

备注：内容来源于stack exchange，提问作者Narendra Pandey