Spring-data-mongodb подключается к нескольким базам данных в одном экземпляре Mongo

Вот ссылка на статью, я думаю, это то, что вы ищете 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;
}

С помощью Autowiring вы сопоставляете имя переменной с именем (id) в вашей конфигурации.

Итак, после долгих исследований и экспериментов я пришел к выводу, что это еще не возможно с текущим проектом spring-data-mongodb. Я попробовал метод бахи выше и столкнулся с определенным препятствием. MongoTemplate запускает свой метод ensureIndexes() из своего конструктора. Этот метод вызывает базу данных, чтобы убедиться, что в базе данных существуют аннотированные индексы. Конструктор для MongoTemplate вызывается при запуске Spring, поэтому у меня даже нет возможности установить переменную ThreadLocal. Я должен иметь уже установленную базу данных по умолчанию при запуске Spring, а затем изменять ее при поступлении запроса. Это недопустимо, потому что я не хочу и не имею базу данных по умолчанию.

Однако не все было потеряно. Наш первоначальный план состоял в том, чтобы каждый клиент работал на своем собственном сервере приложений, указывающем на свою собственную базу данных MongoDB на сервере MongoDB. Затем мы можем предоставить системную переменную -Dprovider=, и каждый сервер будет работать, указывая только на одну базу данных.

Нам было приказано иметь мультитенантное приложение, отсюда и попытка использовать переменную ThreadLocal. Но так как это не сработало, мы смогли запустить приложение так, как мы его изначально разработали.

Я считаю, что есть способ заставить все это работать, просто это требует больше, чем описано в других постах. Вы должны сделать свой собственный RepositoryFactoryBean. Вот пример из Справочные документы Spring Data MongoDB. Вам все равно придется реализовать свой собственный MongoTemplate и отложить или удалить вызов ensureIndexes(). Но вам придется переписать несколько классов, чтобы убедиться, что ваш MongoTemplate вызывается вместо Spring's. Другими словами, много работы. Работа, которую я хотел бы увидеть или даже сделать, у меня просто не было времени.

Спасибо за ответы.

Вы можете создать подкласс SimpleMongoDbFactory и разработать стратегию возврата базы данных по умолчанию, возвращаемой getDb. Один из вариантов — использовать локальные переменные потока для выбора используемой базы данных вместо использования нескольких шаблонов 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);
    }
}

Затем переопределите mongoDBFactory() в своем классе @Configuration, который наследуется от AbstractMongoConfiguration, вот так:

@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() после того, как вы закончите.

Место, на которое стоит обратить внимание, — это интерфейс MongoDbFactory. Базовая реализация этого использует экземпляр Mongo и работает с ним на протяжении всего жизненного цикла приложения. Чтобы добиться использования базы данных для каждого потока (и, следовательно, для каждого запроса), вам, вероятно, придется реализовать что-то вроде строк AbstractRoutingDataSource. Идея в значительной степени состоит в том, что у вас есть шаблонный метод, который должен будет искать арендатора для каждого вызова (я думаю, ThreadLocal связанный), а затем выбирать Mongo экземпляр из набора предопределенных или некоторую пользовательскую логику, чтобы придумать новый для новый арендатор и т.д.

Имейте в виду, что MongoDbFactory обычно используется с помощью метода getDb(). Однако в MongoDB есть функции, которые требуют от нас предоставления файла getDb(String name). DBRefs (что-то вроде внешнего ключа в реляционном мире) может указывать на документы совершенно другой базы данных. Поэтому, если вы выполняете делегирование, либо избегайте использования этой функции (я думаю, что DBRefs, указывающие на другую БД, являются единственными местами, вызывающими getDb(name)), либо явно обрабатывайте ее.

С точки зрения конфигурации вы можете либо просто полностью переопределить mongoDbFactory(), либо вообще не расширять базовый класс и придумать собственную конфигурацию на основе Java.

Я использовал другую БД с помощью 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());    
}

Поэтому, когда мне нужно изменить свою базу данных, я только выбираю, какую конфигурацию использовать.

Насколько я понимаю, вы хотите больше гибкости в изменении текущей базы данных на лету.

Я связал проект, который реализует мультитенантность простым способом.

Его можно использовать в качестве отправной точки для приложения.

Он реализует SimpleMongoDbFactory и предоставляет пользовательский метод getDB для разрешения правильной базы данных для использования в определенный момент. Его можно улучшить разными способами, например, путем получения сведений о базе данных из HttpSession из объекта SpringSession, который, например, может кэшироваться Redis.

Чтобы иметь разные mongoTemplates, использующие разные базы данных одновременно, возможно, измените область действия вашего mongoDbFactory на сеанс.

Использованная литература:

multi-tenant-spring-mongodb