Как избежать дублирования записей при вставке в MongoDB

1

Я просто хочу быть shure при вводе нового DBObject в DB, что он действительно уникален, и Collection не содержит дубликатов ключевых полей.

Вот как это выглядит сейчас:

public abstract class AbstractMongoDAO<ID, MODEL> implements GenericDAO<ID, MODEL> {

    protected Mongo client;

    protected Class<MODEL> model;

    protected DBCollection dbCollection;

    /**
     * Contains model data : unique key name and name of get method
     */
    protected KeyField keyField;

    @SuppressWarnings("unchecked")
    protected AbstractMongoDAO() {
        ParameterizedType genericSuperclass = (ParameterizedType) this.getClass().getGenericSuperclass();
        model = (Class<MODEL>) genericSuperclass.getActualTypeArguments()[1];
        getKeyField();
    }

    public void connect() throws UnknownHostException {
        client = new MongoClient(Config.getMongoHost(), Integer.parseInt(Config.getMongoPort()));
        DB clientDB = client.getDB(Config.getMongoDb());
        clientDB.authenticate(Config.getMongoDbUser(), Config.getMongoDbPass().toCharArray());
        dbCollection = clientDB.getCollection(getCollectionName(model));
    }

    public void disconnect() {
        if (client != null) {
            client.close();
        }
    }

    @Override
    public void create(MODEL model) {
        Object keyValue = get(model);
        try {
            ObjectMapper mapper = new ObjectMapper();
            String requestAsString = mapper.writerWithDefaultPrettyPrinter().writeValueAsString(model);
            // check if not presented
            BasicDBObject dbObject = new BasicDBObject((String) keyValue, requestAsString);
            dbCollection.ensureIndex(dbObject, new BasicDBObject("unique", true));
            dbCollection.insert(new BasicDBObject((String) keyValue, requestAsString));
        } catch (Throwable e) {
            throw new RuntimeException(String.format("Duplicate parameters '%s' : '%s'", keyField.id(), keyValue));
        }
    }

private Object get(MODEL model) {
    Object result = null;
    try {
        Method m = this.model.getMethod(this.keyField.get());
        result = m.invoke(model);
    } catch (Exception e) {
        throw new RuntimeException(String.format("Couldn't find method by name '%s' at class '%s'", this.keyField.get(), this.model.getName()));
    }
    return result;
}

/**
 * Extract the name of collection that is specified at '@Entity' annotation.
 *
 * @param clazz is model class object.
 * @return the name of collection that is specified.
 */
private String getCollectionName(Class<MODEL> clazz) {
    Entity entity = clazz.getAnnotation(Entity.class);
    String tableName = entity.value();
    if (tableName.equals(Mapper.IGNORED_FIELDNAME)) {
        // think about usual logger
        tableName = clazz.getName();
    }
    return tableName;
}

private void getKeyField() {
    for (Field field : this.model.getDeclaredFields()) {
        if (field.isAnnotationPresent(KeyField.class)) {
            keyField = field.getAnnotation(KeyField.class);
            break;
        }
    }
    if (keyField == null) {
        throw new RuntimeException(String.format("Couldn't find key field at class : '%s'", model.getName()));
    }
}

KeyFeld - это специальная аннотация:

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface KeyField {

    String id();

    String get();

    String statusProp() default "ALL";

Но я не уверен, что это решение действительно доказывает это. Я недавно в Монго.

Какие-либо предложения?

  • 0
    Вы смотрели на Spring Data MongoDB? На вопрос, потому что вы пометили вопрос Spring-данными, а затем закончили кодировать, что библиотека предоставляет вам OOTB.
  • 0
    @OliverGierke Хороший выстрел. Я исправил вопрос.
Теги:
mongodb-query

2 ответа

3
Лучший ответ

Единственности может поддерживаться в MonboDb, используя _id поле. Если мы не укажем значение этого поля, MongoDB автоматически создаст уникальный идентификатор для этой коллекции.

Итак, в вашем случае просто создайте свойство с именем _id в java и присвойте свое уникальное значение поля здесь. Если он будет дублирован, он выдает исключение.

1

С Spring Data MongoDB (вопрос был помечен spring-data, поэтому я предлагаю его), все, что вам нужно, это то, что:

 // Your types

 class YourType {

    BigInteger id;
    @Indexed(unique = true) String emailAddress;
    …
 }

 interface YourTypeRepository extends CrudRepository<YourType, BigInteger> { }

 // Infrastructure setup, if you use Spring as container prefer @EnableMongoRepositories

 MongoOperations operations = new MongoTemplate(new MongoClient(), "myDatabase");
 MongoRepositoryFactory factory = new MongoRepositoryFactory(operations);
 YourTypeRepository repository = factory.getRepository(YourTypeRepository.class);

 // Now use it…

 YourType first = …; // set email address
 YourType second = …; // set same email address

 repository.save(first);
 repository.save(second); // will throw an exception

@Indexed частью, которая больше всего связана с вашим исходным вопросом, является @Indexed как это приведет к созданию необходимого уникального индекса, созданного при создании репозитория.

Что вы получаете за это:

  • нет необходимости вручную реализовывать какой-либо репозиторий (удаленный код не содержит ошибок \o/)
  • автоматическое преобразование объектов в документ
  • автоматическое создание индекса
  • мощная абстракция репозитория для простого запроса данных путем объявления методов запросов

Для получения дополнительной информации ознакомьтесь с справочной документацией.

Ещё вопросы

Сообщество Overcoder
Наверх
Меню