Examples with WriteConcern com.mongodb.WriteConcern used on opensource projects

Example 46 with WriteConcern

use of com.mongodb.WriteConcern in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplate method doRemove.

protected <T> Mono<DeleteResult> doRemove(String collectionName, Query query, @Nullable Class<T> entityClass) {
    if (query == null) {
        throw new InvalidDataAccessApiUsageException("Query passed in to remove can't be null!");
    }
    Assert.hasText(collectionName, "Collection name must not be null or empty!");
    MongoPersistentEntity<?> entity = getPersistentEntity(entityClass);
    DeleteContext deleteContext = queryOperations.deleteQueryContext(query);
    Document queryObject = deleteContext.getMappedQuery(entity);
    DeleteOptions deleteOptions = deleteContext.getDeleteOptions(entityClass);
    Document removeQuery = deleteContext.getMappedQuery(entity);
    MongoAction mongoAction = new MongoAction(writeConcern, MongoActionOperation.REMOVE, collectionName, entityClass, null, removeQuery);
    WriteConcern writeConcernToUse = prepareWriteConcern(mongoAction);
    return execute(collectionName, collection -> {
        maybeEmitEvent(new BeforeDeleteEvent<>(removeQuery, entityClass, collectionName));
        MongoCollection<Document> collectionToUse = prepareCollection(collection, writeConcernToUse);
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug(String.format("Remove using query: %s in collection: %s.", serializeToJsonSafely(removeQuery), collectionName));
        }
        if (query.getLimit() > 0 || query.getSkip() > 0) {
            FindPublisher<Document> cursor = new QueryFindPublisherPreparer(query, entityClass).prepare(// 
            collection.find(removeQuery)).projection(MappedDocument.getIdOnlyProjection());
            return // 
            Flux.from(cursor).map(// 
            MappedDocument::of).map(// 
            MappedDocument::getId).collectList().flatMapMany(val -> {
                return collectionToUse.deleteMany(MappedDocument.getIdIn(val), deleteOptions);
            });
        } else {
            return collectionToUse.deleteMany(removeQuery, deleteOptions);
        }
    }).doOnNext(// 
    it -> maybeEmitEvent(new AfterDeleteEvent<>(queryObject, entityClass, collectionName))).next();
}

Example 47 with WriteConcern

use of com.mongodb.WriteConcern in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplate method saveDocument.

protected Mono<Object> saveDocument(String collectionName, Document document, Class<?> entityClass) {
    if (LOGGER.isDebugEnabled()) {
        LOGGER.debug(String.format("Saving Document containing fields: %s", document.keySet()));
    }
    return createMono(collectionName, collection -> {
        MongoAction mongoAction = new MongoAction(writeConcern, MongoActionOperation.SAVE, collectionName, entityClass, document, null);
        WriteConcern writeConcernToUse = prepareWriteConcern(mongoAction);
        MappedDocument mapped = MappedDocument.of(document);
        MongoCollection<Document> collectionToUse = // 
        writeConcernToUse == null ? // 
        collection : collection.withWriteConcern(writeConcernToUse);
        Publisher<?> publisher;
        if (!mapped.hasId()) {
            publisher = collectionToUse.insertOne(document);
        } else {
            MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(entityClass);
            UpdateContext updateContext = queryOperations.replaceSingleContext(mapped, true);
            Document filter = updateContext.getMappedQuery(entity);
            Document replacement = updateContext.getMappedUpdate(entity);
            Mono<Document> deferredFilter;
            if (updateContext.requiresShardKey(filter, entity)) {
                if (entity.getShardKey().isImmutable()) {
                    deferredFilter = Mono.just(updateContext.applyShardKey(entity, filter, null));
                } else {
                    deferredFilter = Mono.from(collection.find(filter, Document.class).projection(updateContext.getMappedShardKey(entity)).first()).defaultIfEmpty(replacement).map(it -> updateContext.applyShardKey(entity, filter, it));
                }
            } else {
                deferredFilter = Mono.just(filter);
            }
            publisher = deferredFilter.flatMapMany(it -> collectionToUse.replaceOne(it, replacement, updateContext.getReplaceOptions(entityClass)));
        }
        return Mono.from(publisher).map(o -> mapped.getId());
    });
}

Example 48 with WriteConcern

use of com.mongodb.WriteConcern in project spring-data-mongodb by spring-projects.

the class MongoTemplate method insertDocumentList.

protected List<Object> insertDocumentList(String collectionName, List<Document> documents) {
    if (documents.isEmpty()) {
        return Collections.emptyList();
    }
    if (LOGGER.isDebugEnabled()) {
        LOGGER.debug(String.format("Inserting list of Documents containing %s items", documents.size()));
    }
    execute(collectionName, collection -> {
        MongoAction mongoAction = new MongoAction(writeConcern, MongoActionOperation.INSERT_LIST, collectionName, null, null, null);
        WriteConcern writeConcernToUse = prepareWriteConcern(mongoAction);
        if (writeConcernToUse == null) {
            collection.insertMany(documents);
        } else {
            collection.withWriteConcern(writeConcernToUse).insertMany(documents);
        }
        return null;
    });
    return MappedDocument.toIds(documents);
}

Example 49 with WriteConcern

use of com.mongodb.WriteConcern in project spring-data-mongodb by spring-projects.

the class MongoDbFactoryParserIntegrationTests method assertWriteConcern.

private static void assertWriteConcern(ClassPathXmlApplicationContext ctx, WriteConcern expectedWriteConcern) {
    SimpleMongoClientDatabaseFactory dbFactory = ctx.getBean("first", SimpleMongoClientDatabaseFactory.class);
    MongoDatabase db = dbFactory.getMongoDatabase();
    assertThat(db.getName()).isEqualTo("db");
    WriteConcern configuredConcern = (WriteConcern) ReflectionTestUtils.getField(dbFactory, "writeConcern");
    assertThat(configuredConcern).isEqualTo(expectedWriteConcern);
    assertThat(db.getWriteConcern()).isEqualTo(expectedWriteConcern);
    assertThat(db.getWriteConcern()).isEqualTo(expectedWriteConcern);
}

Example 50 with WriteConcern

use of com.mongodb.WriteConcern in project spring-data-mongodb by spring-projects.

the class MongoDbFactoryParserIntegrationTests method testWriteConcernEquality.

// This test will fail since equals in WriteConcern uses == for _w and not .equals
public void testWriteConcernEquality() {
    String s1 = new String("rack1");
    String s2 = new String("rack1");
    WriteConcern wc1 = new WriteConcern(s1);
    WriteConcern wc2 = new WriteConcern(s2);
    assertThat(wc1).isEqualTo(wc2);
}