Examples with AggregationOperationContext org.springframework.data.mongodb.core.aggregation.AggregationOperationContext used on opensource projects

Example 1 with AggregationOperationContext

use of org.springframework.data.mongodb.core.aggregation.AggregationOperationContext in project spring-data-mongodb by spring-projects.

the class MongoTemplate method aggregate.

protected <O> AggregationResults<O> aggregate(Aggregation aggregation, String collectionName, Class<O> outputType, @Nullable AggregationOperationContext context) {
    Assert.hasText(collectionName, "Collection name must not be null or empty!");
    Assert.notNull(aggregation, "Aggregation pipeline must not be null!");
    Assert.notNull(outputType, "Output type must not be null!");
    AggregationOperationContext rootContext = context == null ? Aggregation.DEFAULT_CONTEXT : context;
    return doAggregate(aggregation, collectionName, outputType, rootContext);
}

Example 2 with AggregationOperationContext

use of org.springframework.data.mongodb.core.aggregation.AggregationOperationContext in project spring-data-mongodb by spring-projects.

the class MongoTemplate method aggregate.

/* (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.MongoOperations#aggregate(org.springframework.data.mongodb.core.aggregation.TypedAggregation, java.lang.String, java.lang.Class)
	 */
@Override
public <O> AggregationResults<O> aggregate(TypedAggregation<?> aggregation, String inputCollectionName, Class<O> outputType) {
    Assert.notNull(aggregation, "Aggregation pipeline must not be null!");
    AggregationOperationContext context = new TypeBasedAggregationOperationContext(aggregation.getInputType(), mappingContext, queryMapper);
    return aggregate(aggregation, inputCollectionName, outputType, context);
}

Example 3 with AggregationOperationContext

use of org.springframework.data.mongodb.core.aggregation.AggregationOperationContext in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplate method aggregate.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.ReactiveMongoOperations#aggregate(org.springframework.data.mongodb.core.aggregation.TypedAggregation, java.lang.String, java.lang.Class)
	 */
@Override
public <O> Flux<O> aggregate(TypedAggregation<?> aggregation, String inputCollectionName, Class<O> outputType) {
    Assert.notNull(aggregation, "Aggregation pipeline must not be null!");
    AggregationOperationContext context = new TypeBasedAggregationOperationContext(aggregation.getInputType(), mappingContext, queryMapper);
    return aggregate(aggregation, inputCollectionName, outputType, context);
}

Example 4 with AggregationOperationContext

use of org.springframework.data.mongodb.core.aggregation.AggregationOperationContext in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplate method aggregate.

/**
 * @param aggregation must not be {@literal null}.
 * @param collectionName must not be {@literal null}.
 * @param outputType must not be {@literal null}.
 * @param context can be {@literal null} and will be defaulted to {@link Aggregation#DEFAULT_CONTEXT}.
 * @return never {@literal null}.
 */
protected <O> Flux<O> aggregate(Aggregation aggregation, String collectionName, Class<O> outputType, @Nullable AggregationOperationContext context) {
    Assert.notNull(aggregation, "Aggregation pipeline must not be null!");
    Assert.hasText(collectionName, "Collection name must not be null or empty!");
    Assert.notNull(outputType, "Output type must not be null!");
    AggregationOperationContext rootContext = context == null ? Aggregation.DEFAULT_CONTEXT : context;
    AggregationOptions options = aggregation.getOptions();
    List<Document> pipeline = aggregation.toPipeline(rootContext);
    Assert.isTrue(!options.isExplain(), "Cannot use explain option with streaming!");
    Assert.isNull(options.getCursorBatchSize(), "Cannot use batchSize cursor option with streaming!");
    if (LOGGER.isDebugEnabled()) {
        LOGGER.debug("Streaming aggregation: {} in collection {}", serializeToJsonSafely(pipeline), collectionName);
    }
    ReadDocumentCallback<O> readCallback = new ReadDocumentCallback<>(mongoConverter, outputType, collectionName);
    return execute(collectionName, collection -> aggregateAndMap(collection, pipeline, options, readCallback));
}

Example 5 with AggregationOperationContext

use of org.springframework.data.mongodb.core.aggregation.AggregationOperationContext in project spring-data-mongodb by spring-projects.

the class ChangeStreamTask method prepareFilter.

List<Document> prepareFilter(MongoTemplate template, ChangeStreamOptions options) {
    if (!options.getFilter().isPresent()) {
        return Collections.emptyList();
    }
    Object filter = options.getFilter().get();
    if (filter instanceof Aggregation) {
        Aggregation agg = (Aggregation) filter;
        AggregationOperationContext context = agg instanceof TypedAggregation ? new TypeBasedAggregationOperationContext(((TypedAggregation<?>) agg).getInputType(), template.getConverter().getMappingContext(), queryMapper) : Aggregation.DEFAULT_CONTEXT;
        return agg.toPipeline(new PrefixingDelegatingAggregationOperationContext(context, "fullDocument", blacklist));
    } else if (filter instanceof List) {
        return (List<Document>) filter;
    } else {
        throw new IllegalArgumentException("ChangeStreamRequestOptions.filter mut be either an Aggregation or a plain list of Documents");
    }
}