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

Example 6 with QueryContext

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

the class MongoTemplate method doFind.

/**
 * Map the results of an ad-hoc query on the default MongoDB collection to a List of the specified targetClass while
 * using sourceClass for mapping the query.
 *
 * @since 2.0
 */
<S, T> List<T> doFind(String collectionName, Document query, Document fields, Class<S> sourceClass, Class<T> targetClass, CursorPreparer preparer) {
    MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(sourceClass);
    EntityProjection<T, S> projection = operations.introspectProjection(targetClass, sourceClass);
    QueryContext queryContext = queryOperations.createQueryContext(new BasicQuery(query, fields));
    Document mappedFields = queryContext.getMappedFields(entity, projection);
    Document mappedQuery = queryContext.getMappedQuery(entity);
    if (LOGGER.isDebugEnabled()) {
        LOGGER.debug(String.format("find using query: %s fields: %s for class: %s in collection: %s", serializeToJsonSafely(mappedQuery), mappedFields, sourceClass, collectionName));
    }
    return executeFindMultiInternal(new FindCallback(mappedQuery, mappedFields, null), preparer, new ProjectingReadCallback<>(mongoConverter, projection, collectionName), collectionName);
}

Example 7 with QueryContext

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

the class MongoTemplate method findAndReplace.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.MongoOperations#findAndReplace(org.springframework.data.mongodb.core.query.Query, java.lang.Object, org.springframework.data.mongodb.core.FindAndReplaceOptions, java.lang.Class, java.lang.String, java.lang.Class)
	 */
@Override
public <S, T> T findAndReplace(Query query, S replacement, FindAndReplaceOptions options, Class<S> entityType, String collectionName, Class<T> resultType) {
    Assert.notNull(query, "Query must not be null!");
    Assert.notNull(replacement, "Replacement must not be null!");
    Assert.notNull(options, "Options must not be null! Use FindAndReplaceOptions#empty() instead.");
    Assert.notNull(entityType, "EntityType must not be null!");
    Assert.notNull(collectionName, "CollectionName must not be null!");
    Assert.notNull(resultType, "ResultType must not be null! Use Object.class instead.");
    Assert.isTrue(query.getLimit() <= 1, "Query must not define a limit other than 1 ore none!");
    Assert.isTrue(query.getSkip() <= 0, "Query must not define skip.");
    MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(entityType);
    QueryContext queryContext = queryOperations.createQueryContext(query);
    EntityProjection<T, S> projection = operations.introspectProjection(resultType, entityType);
    Document mappedQuery = queryContext.getMappedQuery(entity);
    Document mappedFields = queryContext.getMappedFields(entity, projection);
    Document mappedSort = queryContext.getMappedSort(entity);
    replacement = maybeCallBeforeConvert(replacement, collectionName);
    Document mappedReplacement = operations.forEntity(replacement).toMappedDocument(this.mongoConverter).getDocument();
    maybeEmitEvent(new BeforeSaveEvent<>(replacement, mappedReplacement, collectionName));
    maybeCallBeforeSave(replacement, mappedReplacement, collectionName);
    T saved = doFindAndReplace(collectionName, mappedQuery, mappedFields, mappedSort, queryContext.getCollation(entityType).orElse(null), entityType, mappedReplacement, options, projection);
    if (saved != null) {
        maybeEmitEvent(new AfterSaveEvent<>(saved, mappedReplacement, collectionName));
        return maybeCallAfterSave(saved, mappedReplacement, collectionName);
    }
    return saved;
}

Example 8 with QueryContext

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

the class ReactiveMongoTemplate method findAndReplace.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.ReactiveMongoOperations#findAndReplace(org.springframework.data.mongodb.core.query.Query, java.lang.Object, org.springframework.data.mongodb.core.FindAndReplaceOptions, java.lang.Class, java.lang.String, java.lang.Class)
	 */
@Override
public <S, T> Mono<T> findAndReplace(Query query, S replacement, FindAndReplaceOptions options, Class<S> entityType, String collectionName, Class<T> resultType) {
    Assert.notNull(query, "Query must not be null!");
    Assert.notNull(replacement, "Replacement must not be null!");
    Assert.notNull(options, "Options must not be null! Use FindAndReplaceOptions#empty() instead.");
    Assert.notNull(entityType, "Entity class must not be null!");
    Assert.notNull(collectionName, "CollectionName must not be null!");
    Assert.notNull(resultType, "ResultType must not be null! Use Object.class instead.");
    Assert.isTrue(query.getLimit() <= 1, "Query must not define a limit other than 1 ore none!");
    Assert.isTrue(query.getSkip() <= 0, "Query must not define skip.");
    MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(entityType);
    QueryContext queryContext = queryOperations.createQueryContext(query);
    EntityProjection<T, S> projection = operations.introspectProjection(resultType, entityType);
    Document mappedQuery = queryContext.getMappedQuery(entity);
    Document mappedFields = queryContext.getMappedFields(entity, projection);
    Document mappedSort = queryContext.getMappedSort(entity);
    return Mono.defer(() -> {
        PersistableEntityModel<S> pem = PersistableEntityModel.of(replacement, collectionName);
        maybeEmitEvent(new BeforeConvertEvent<>(pem.getSource(), pem.getCollection()));
        return maybeCallBeforeConvert(pem.getSource(), pem.getCollection()).map(pem::mutate).flatMap(it -> {
            PersistableEntityModel<S> mapped = it.addTargetDocument(operations.forEntity(it.getSource()).toMappedDocument(mongoConverter).getDocument());
            maybeEmitEvent(new BeforeSaveEvent(mapped.getSource(), mapped.getTarget(), mapped.getCollection()));
            return maybeCallBeforeSave(it.getSource(), mapped.getTarget(), mapped.getCollection()).map(potentiallyModified -> PersistableEntityModel.of(potentiallyModified, mapped.getTarget(), mapped.getCollection()));
        }).flatMap(it -> {
            Mono<T> afterFindAndReplace = doFindAndReplace(it.getCollection(), mappedQuery, mappedFields, mappedSort, queryContext.getCollation(entityType).orElse(null), entityType, it.getTarget(), options, projection);
            return afterFindAndReplace.flatMap(saved -> {
                maybeEmitEvent(new AfterSaveEvent<>(saved, it.getTarget(), it.getCollection()));
                return maybeCallAfterSave(saved, it.getTarget(), it.getCollection());
            });
        });
    });
}

Example 9 with QueryContext

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

the class MongoTemplate method doFind.

protected <S, T> List<T> doFind(String collectionName, Document query, Document fields, Class<S> entityClass, @Nullable CursorPreparer preparer, DocumentCallback<T> objectCallback) {
    MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(entityClass);
    QueryContext queryContext = queryOperations.createQueryContext(new BasicQuery(query, fields));
    Document mappedFields = queryContext.getMappedFields(entity, EntityProjection.nonProjecting(entityClass));
    Document mappedQuery = queryContext.getMappedQuery(entity);
    if (LOGGER.isDebugEnabled()) {
        LOGGER.debug(String.format("find using query: %s fields: %s for class: %s in collection: %s", serializeToJsonSafely(mappedQuery), mappedFields, entityClass, collectionName));
    }
    return executeFindMultiInternal(new FindCallback(mappedQuery, mappedFields, null), preparer != null ? preparer : CursorPreparer.NO_OP_PREPARER, objectCallback, collectionName);
}