Spring Boot集成MongoDB运行时动态切换集合名称的方案咨询

方案B的优势 #

• 逻辑直观无黑魔法，上手门槛低，排查问题不需要深入Spring Data MongoDB底层实现
• 不需要修改现有Repository的查询逻辑，原有方法可以直接复用

方案B的潜在坑点 #

• 手动创建Repository实例需要处理Spring依赖注入问题：Spring默认生成的Repository实例是代理对象，包含事务拦截、类型转换等内置切面逻辑，自行创建实例如果没有正确关联ApplicationContext，会导致依赖注入失败、事务不生效等问题
• 可扩展性受限：如果后续realm数量增长到几十上百个，对象池会占用过多堆内存，同时并发请求新增realm时需要处理线程安全问题，避免重复创建同集合的Repository实例
• 跨realm事务完全不支持：如果业务存在跨用户realm的数据操作，该方案无法实现事务管控

推荐两种符合Spring Boot规范的线程安全实现，不需要自行维护实例池：

方案1：SpEL + ThreadLocal 动态解析集合名（代码侵入最小） #

该方案是SpEL方案的线程安全改进版，完全避免用户数据串扰：

注意：原SpEL方案的线程安全问题源于将集合名作为Repository的成员变量存储，多线程请求会覆盖变量值，本方案通过ThreadLocal实现每个请求线程的realm隔离，完全避免该问题。

1. 编写Realm持有者类，用ThreadLocal存储当前请求的用户realm标识：

public class RealmHolder {
    private static final ThreadLocal<String> CURRENT_REALM = new ThreadLocal<>();

    public static void setCurrentRealm(String realm) {
        CURRENT_REALM.set(realm);
    }

    public static String getCurrentRealm() {
        return CURRENT_REALM.get();
    }

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

2. 编写Spring MVC拦截器，请求进来时解析用户身份，设置对应的realm，请求结束后清空ThreadLocal：

@Component
public class RealmInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 从请求上下文获取当前用户的realm标识，比如从JWT、session中解析
        String userRealm = getUserRealmFromRequest(request);
        RealmHolder.setCurrentRealm(userRealm);
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        RealmHolder.clear();
    }
}

3. 修改Product实体类的@Document注解，用SpEL表达式动态读取realm：

@Document(collection = "products_#{T(com.xxx.RealmHolder).getCurrentRealm()}")
public class Product {
    // 原有实体字段不变
}

该方案完全复用Spring Data MongoDB的原生机制，原有ProductRepository的代码不需要做任何修改，天然线程安全，不会出现数据串扰。

方案2：直接使用MongoTemplate指定集合名（最灵活） #

如果业务存在大量自定义查询、跨realm操作的场景，直接放弃MongoRepository，改用MongoTemplate手动指定集合名：

@Service
public class ProductService {
    @Autowired
    private MongoTemplate mongoTemplate;

    public List<Product> listUserProducts(String userId) {
        String userRealm = getUserRealmByUserId(userId);
        String collectionName = "products_" + userRealm;
        Query query = Query.query(Criteria.where("userId").is(userId));
        return mongoTemplate.find(query, Product.class, collectionName);
    }
}

该方案完全可控，没有任何线程安全问题，适合复杂业务场景。

• 若realm数量不多、查询逻辑都是通用的Repository方法，优先选SpEL+ThreadLocal方案，代码量最少，符合Spring规范
• 若业务灵活度要求高、存在跨realm操作，优先选MongoTemplate方案
• 只有当不同realm有严格的合规隔离要求、资源配额需要单独管控时，再考虑方案C

内容的提问来源于stack exchange，提问作者sascha10000