spring-data-mongodb 在一个 Mongo 实例中连接多个数据库

Question

我正在使用最新的 spring-data-mongodb (1.1.0.M2) 和最新的 Mongo Driver (2.9.0-RC1)。我有一种情况，我有多个客户端连接到我的应用程序，我想在同一个 Mongo 服务器中为每个客户端提供自己的“模式/数据库”。如果我直接使用驱动程序，这不是一项很难完成的任务：

Mongo mongo = new Mongo( new DBAddress( "localhost", 127017 ) );

DB client1DB = mongo.getDB( "client1" );
DBCollection client1TTestCollection = client1DB.getCollection( "test" );
long client1TestCollectionCount = client1TTestCollection.count();

DB client2DB = mongo.getDB( "client2" );
DBCollection client2TTestCollection = client2DB.getCollection( "test" );
long client2TestCollectionCount = client2TTestCollection.count();

看，很简单。但是 spring-data-mongodb 不允许使用多个数据库的简单方法。建立与Mongo 的连接的首选方法是扩展AbstractMongoConfiguration 类：

你会看到你重写了下面的方法：

getDatabaseName()

所以它迫使您使用一个数据库名称。然后，您构建的存储库接口使用 MongoTemplate 中的该数据库名称，该名称被传递到 SimpleMongoRepository 类。

究竟我会在哪里粘贴多个数据库名称？我必须创建多个数据库名称、多个MongoTempates（每个数据库名称一个）和多个其他配置类。这仍然没有让我的存储库接口使用正确的模板。如果有人尝试过这样的事情，请告诉我。如果我弄清楚了，我会在这里发布答案。

谢谢。

Answer 1

这是我认为您正在寻找的文章的链接http://michaelbarnesjr.wordpress.com/2012/01/19/spring-data-mongo/

关键是提供多个模板

为每个数据库配置一个模板。

<bean id="vehicleTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
    <constructor-arg ref="mongoConnection"/>
    <constructor-arg name="databaseName" value="vehicledatabase"/>
</bean>

为每个数据库配置一个模板。

<bean id="imageTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
        <constructor-arg ref="mongoConnection"/>
        <constructor-arg name="databaseName" value="imagedatabase"/>
</bean>

<bean id="vehicleTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
    <constructor-arg ref="mongoConnection"/>
    <constructor-arg name="databaseName" value="vehicledatabase"/>
</bean>

现在，您需要告诉 Spring 您的存储库在哪里，以便它可以注入它们。它们必须都在同一个目录中。我试图将它们放在不同的子目录中，但它无法正常工作。所以它们都在存储库目录中。

<mongo:repositories base-package="my.package.repository">
    <mongo:repository id="imageRepository" mongo-template-ref="imageTemplate"/>
    <mongo:repository id="carRepository" mongo-template-ref="vehicleTemplate"/>
    <mongo:repository id="truckRepository" mongo-template-ref="vehicleTemplate"/>
</mongo:repositories>

每个存储库都是一个接口，编写如下（是的，您可以将它们留空）：

@Repository
public interface ImageRepository extends MongoRepository<Image, String> {

}

@Repository
public interface TruckRepository extends MongoRepository<Truck, String> {

}

私有变量imageRepository的名字就是集合！ Image.java 将被保存到 imagedb 数据库中的图像集合中。

以下是查找、插入和删除记录的方法：

@Service
public class ImageService {

    @Autowired
    private ImageRepository imageRepository;
}

通过自动装配，您可以将变量名称与配置中的名称 (id) 匹配。

Answer 2

因此，经过大量研究和实验，我得出的结论是，目前的spring-data-mongodb 项目还不可能做到这一点。我在上面尝试了 baja 的方法并遇到了一个特定的障碍。 MongoTemplate 在其构造函数中运行其ensureIndexes() 方法。此方法调用数据库以确保数据库中存在带注释的索引。 MongoTemplate 的构造函数在Spring 启动时被调用，所以我什至没有机会设置ThreadLocal 变量。我必须在Spring 启动时设置一个默认值，然后在收到请求时更改它。这是不允许的，因为我不想要也没有默认数据库。

尽管如此，一切都没有丢失。我们最初的计划是让每个客户端在自己的应用服务器上运行，指向MongoDB 服务器上自己的MongoDB 数据库。然后我们可以提供一个-Dprovider= 系统变量，每个服务器运行时只指向一个数据库。

我们被指示拥有一个多租户应用程序，因此尝试使用 ThreadLocal 变量。但由于它不起作用，我们能够按照我们最初设计的方式运行应用程序。

我相信有一种方法可以使这一切正常工作，它只需要比其他帖子中描述的更多。您必须创建自己的RepositoryFactoryBean。这是来自Spring Data MongoDB Reference Docs 的示例。您仍然需要实现自己的 MongoTemplate 并延迟或删除 ensureIndexes() 调用。但是你必须重写一些类来确保你的MongoTemplate被调用而不是Spring's。换句话说，工作量很大。我想看到或什至做的工作，我只是没有时间。

感谢您的回复。

Answer 3

您可能希望对SimpleMongoDbFactory 进行子类化，并制定如何返回getDb 返回的默认数据库的策略。一种选择是使用线程局部变量来决定要使用的 Db，而不是使用多个 MongoTemplate。

类似这样的：

public class ThreadLocalDbNameMongoDbFactory extends SimpleMongoDbFactory {
    private static final ThreadLocal<String> dbName = new ThreadLocal<String>();
    private final String defaultName; // init in c'tor before calling super

    // omitted constructor for clarity

    public static void setDefaultNameForCurrentThread(String tlName) {
        dbName.set(tlName);
    }
    public static void clearDefaultNameForCurrentThread() {
        dbName.remove();
    }

    public DB getDb() {
        String tlName = dbName.get();
        return super.getDb(tlName != null ? tlName : defaultName);
    }
}

然后，在从 AbstractMongoConfiguration 扩展的 @Configuration 类中覆盖 mongoDBFactory()，如下所示：

@Bean
@Override
public MongoDbFactory mongoDbFactory() throws Exception {
  if (getUserCredentials() == null) {
      return new ThreadLocalDbNameMongoDbFactory(mongo(), getDatabaseName());
  } else {
      return new ThreadLocalDbNameMongoDbFactory(mongo(), getDatabaseName(), getUserCredentials());
  }
}

在您的客户端代码（可能是 ServletFilter 等）中，您需要调用： ThreadLocalDBNameMongoRepository.setDefaultNameForCurrentThread() 在进行任何 Mongo 工作并随后将其重置为： ThreadLocalDBNameMongoRepository.clearDefaultNameForCurrentThread() 完成后。

Answer 4

要看的地方是MongoDbFactory 接口。它的基本实现需要一个 Mongo 实例，并在整个应用程序生命周期中使用它。要实现每个线程（以及每个请求）的数据库使用，您可能必须按照AbstractRoutingDataSource 的方式实现一些东西。这个想法几乎是你有一个模板方法必须在每次调用时查找租户（我猜是ThreadLocal绑定），然后从一组预定义的实例或一些自定义逻辑中选择一个Mongo实例来提出新租户等的新的。

请记住，MongoDbFactory 通常通过getDb() 方法使用。但是，MongoDB 中有一些功能需要我们提供getDb(String name)。 DBRefs（就像关系世界中的外键）可以指向一个完全不同的数据库。因此，如果您正在进行委托，要么避免使用该功能（我认为指向另一个数据库的 DBRefs 是唯一调用 getDb(name) 的地方）或明确处理它。

从配置的角度来看，您既可以简单地完全覆盖 mongoDbFactory()，也可以根本不扩展基类并提出您自己的基于 Java 的配置。

Answer 5

我使用 java Config 使用了不同的数据库，我就是这样做的：

@Bean 
public MongoDbFactory mongoRestDbFactory() throws Exception { 
    MongoClientURI uri=new MongoClientURI(environment.getProperty("mongo.uri")); 
    return new SimpleMongoDbFactory(uri);
}

@Override
public String getDatabaseName() {
    return "rest";
}

@Override
public @Bean(name = "secondaryMongoTemplate") MongoTemplate mongoTemplate() throws Exception{ //hay que cambiar el nombre de los templates para que el contendor de beans sepa la diferencia  
    return new MongoTemplate(mongoRestDbFactory());    
}

另一个是这样的：

@Bean 
public MongoDbFactory restDbFactory() throws Exception {
    MongoClientURI uri = new MongoClientURI(environment.getProperty("mongo.urirestaurants")); 
    return new SimpleMongoDbFactory(uri);
}

@Override
public String getDatabaseName() {
    return "rest";
}

@Override
public @Bean(name = "primaryMongoTemplate") MongoTemplate mongoTemplate() throws Exception{ 
    return new MongoTemplate(restDbFactory());    
}

所以当我需要更改我的数据库时，我只选择要使用的配置

Answer 6

据我了解，您希望更灵活地动态更改当前数据库。

我已经链接了一个以简单方式实现多租户的项目。

它可以用作应用程序的起点。

它实现了 SimpleMongoDbFactory 并提供了一个自定义的 getDB 方法来解析正确的数据库以在特定时刻使用。它可以通过多种方式进行改进，例如，通过从 SpringSession 对象的 HttpSession 检索数据库详细信息，例如可以由 Redis 缓存。

要让不同的 mongoTemplates 同时使用不同的 db，可以将 mongoDbFactory 的范围更改为 session。

参考资料：

multi-tenant-spring-mongodb