Examples with UpdateContext org.springframework.data.mongodb.core.QueryOperations.UpdateContext used on opensource projects

Example 6 with UpdateContext

use of org.springframework.data.mongodb.core.QueryOperations.UpdateContext in project spring-data-mongodb by spring-projects.

the class MongoTemplate method doFindAndModify.

@SuppressWarnings("ConstantConditions")
protected <T> T doFindAndModify(String collectionName, Document query, Document fields, Document sort, Class<T> entityClass, UpdateDefinition update, @Nullable FindAndModifyOptions options) {
    EntityReader<? super T, Bson> readerToUse = this.mongoConverter;
    if (options == null) {
        options = new FindAndModifyOptions();
    }
    MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(entityClass);
    UpdateContext updateContext = queryOperations.updateSingleContext(update, query, false);
    updateContext.increaseVersionForUpdateIfNecessary(entity);
    Document mappedQuery = updateContext.getMappedQuery(entity);
    Object mappedUpdate = updateContext.isAggregationUpdate() ? updateContext.getUpdatePipeline(entityClass) : updateContext.getMappedUpdate(entity);
    if (LOGGER.isDebugEnabled()) {
        LOGGER.debug(String.format("findAndModify using query: %s fields: %s sort: %s for class: %s and update: %s in collection: %s", serializeToJsonSafely(mappedQuery), fields, sort, entityClass, serializeToJsonSafely(mappedUpdate), collectionName));
    }
    return executeFindOneInternal(new FindAndModifyCallback(mappedQuery, fields, sort, mappedUpdate, update.getArrayFilters().stream().map(ArrayFilter::asDocument).collect(Collectors.toList()), options), new ReadDocumentCallback<>(readerToUse, entityClass, collectionName), collectionName);
}