Spring Boot WebFlux Reactive MongoDB：如何在每个请求中切换数据库？

Question

我正在使用Spring WebFlux和Reactive MongoDB开发一个SaaS项目。它需要是一个多租户应用程序，每个租户必须使用自己的数据库。

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-mongodb-reactive</artifactId>
    <version>2.2.6.RELEASE</version>
</dependency>

然后我扩展了AbstractReactiveMongoConfiguration，以便提供MongoClient和DatabaseName：

import com.mongodb.reactivestreams.client.MongoClient;
import com.mongodb.reactivestreams.client.MongoClients;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.config.AbstractReactiveMongoConfiguration;

@Configuration
public class DatabaseConfiguration extends AbstractReactiveMongoConfiguration {
    @Override
    public MongoClient reactiveMongoClient() {
        System.out.println("ReactiveMongoClient");
        return MongoClients.create();
    }

    @Override
    protected String getDatabaseName() {
        System.out.println("DataBase name");
        return "gateway";
    }
}

整个应用程序是一个OAuth2.0资源服务器，我可以从ReactiveSecurityContextHolder中的身份验证中检索TenantID。

public Mono<String> tenantID() {
    return ReactiveSecurityContextHolder.getContext()
            .switchIfEmpty(Mono.empty())
            .flatMap((securityContext) -> {
                Authentication authentication = securityContext.getAuthentication();

                if (authentication instanceof CustomAuthenticationToken) {
                    CustomAuthenticationToken customAuthenticationToken = (customAuthenticationToken) authentication;
                    Jwt jwt = customAuthenticationToken.getToken();

                    String issuer = jwt.getIssuer().toString();
                    return Mono.justOrEmpty(issuer);
                }

                return Mono.empty();
            });
}

为了根据执行请求的用户（身份验证）切换数据库，下一步是什么？

Answer 1

几个月前我也遇到过同样的问题，想分享一下我是如何解决的。

Spring没有为这个问题提供任何开箱即用的解决方案。几个月前，我创建了一个概念证明，如何解决它，并刚刚在GitHub上发布了示例项目。

Spring-MongoDB-多租户-示例

简而言之：我创建了一个自定义的MultiTenancyReactiveMongoTemplate，它根据来自SubscriberContext的tenantId在内部委托给实际的ReactiveMongoTemplate。tenantId是从定制WebFilter中的HTTP请求头中提取的，它将其放在SubscriberContext中。

此解决方案适用于大多数情况，还支持自动创建索引和使用reactivemongorepository。

但也有一些限制,因为MultiTenancyReactiveMongoTemplate或ReactiveGridFStemplate上的事务、IndexOp不能与提供的解决方案一起工作。有些事情可以用其他委托“模板”来实现，但有些事情永远不会起作用，因为这些操作只是返回标量值（没有publisher)，并且在这些情况下无法访问subscriberContext。

如果您不需要这些功能，您可能会使用这个解决方案。

您可以为每个租户/客户升级和配置服务实例，并在这些服务的‘前面’放置一个反向代理。反向代理决定应该使用哪个服务来处理请求。反向代理可以很容易地实现，例如Spring Cloud Gateway，它允许您很容易地实现决定（例如，基于身份验证令牌）应该使用哪种服务的谓词。

有了CloudFoundry或Kubernetes这样的技术，协调这些特定于承租人的服务的部署就不再那么困难了，这个解决方案也使得进行特定于承租人的监视、警报甚至计费变得很容易。

我们选择解决方案2是因为总体而言，这对我们来说更容易，规模更大。