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

Example 1 with DistinctQueryContext

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

the class ReactiveMongoTemplate method findDistinct.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.ReactiveMongoOperations#findDistinct(org.springframework.data.mongodb.core.query.Query, java.lang.String, java.lang.String, java.lang.Class, java.lang.Class)
	 */
@SuppressWarnings("unchecked")
public <T> Flux<T> findDistinct(Query query, String field, String collectionName, Class<?> entityClass, Class<T> resultClass) {
    Assert.notNull(query, "Query must not be null!");
    Assert.notNull(field, "Field must not be null!");
    Assert.notNull(collectionName, "CollectionName must not be null!");
    Assert.notNull(entityClass, "EntityClass must not be null!");
    Assert.notNull(resultClass, "ResultClass must not be null!");
    MongoPersistentEntity<?> entity = getPersistentEntity(entityClass);
    DistinctQueryContext distinctQueryContext = queryOperations.distinctQueryContext(query, field);
    Document mappedQuery = distinctQueryContext.getMappedQuery(entity);
    String mappedFieldName = distinctQueryContext.getMappedFieldName(entity);
    Class<T> mongoDriverCompatibleType = distinctQueryContext.getDriverCompatibleClass(resultClass);
    Flux<?> result = execute(collectionName, collection -> {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug(String.format("Executing findDistinct using query %s for field: %s in collection: %s", serializeToJsonSafely(mappedQuery), field, collectionName));
        }
        FindPublisherPreparer preparer = new QueryFindPublisherPreparer(query, entityClass);
        if (preparer.hasReadPreference()) {
            collection = collection.withReadPreference(preparer.getReadPreference());
        }
        DistinctPublisher<T> publisher = collection.distinct(mappedFieldName, mappedQuery, mongoDriverCompatibleType);
        distinctQueryContext.applyCollation(entityClass, publisher::collation);
        return publisher;
    });
    if (resultClass == Object.class || mongoDriverCompatibleType != resultClass) {
        Class<?> targetType = distinctQueryContext.getMostSpecificConversionTargetType(resultClass, entityClass);
        MongoConverter converter = getConverter();
        result = result.map(it -> converter.mapValueToTargetType(it, targetType, NO_OP_REF_RESOLVER));
    }
    return (Flux<T>) result;
}

Example 2 with DistinctQueryContext

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

the class MongoTemplate method findDistinct.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.MongoOperations#findDistinct(org.springframework.data.mongodb.core.query.Query, java.lang.String, java.lang.String, java.lang.Class, java.lang.Class)
	 */
@Override
@SuppressWarnings("unchecked")
public <T> List<T> findDistinct(Query query, String field, String collectionName, Class<?> entityClass, Class<T> resultClass) {
    Assert.notNull(query, "Query must not be null!");
    Assert.notNull(field, "Field must not be null!");
    Assert.notNull(collectionName, "CollectionName must not be null!");
    Assert.notNull(entityClass, "EntityClass must not be null!");
    Assert.notNull(resultClass, "ResultClass must not be null!");
    MongoPersistentEntity<?> entity = entityClass != Object.class ? getPersistentEntity(entityClass) : null;
    DistinctQueryContext distinctQueryContext = queryOperations.distinctQueryContext(query, field);
    Document mappedQuery = distinctQueryContext.getMappedQuery(entity);
    String mappedFieldName = distinctQueryContext.getMappedFieldName(entity);
    Class<T> mongoDriverCompatibleType = distinctQueryContext.getDriverCompatibleClass(resultClass);
    MongoIterable<?> result = execute(collectionName, (collection) -> {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug(String.format("Executing findDistinct using query %s for field: %s in collection: %s", serializeToJsonSafely(mappedQuery), field, collectionName));
        }
        QueryCursorPreparer preparer = new QueryCursorPreparer(query, entityClass);
        if (preparer.hasReadPreference()) {
            collection = collection.withReadPreference(preparer.getReadPreference());
        }
        DistinctIterable<T> iterable = collection.distinct(mappedFieldName, mappedQuery, mongoDriverCompatibleType);
        distinctQueryContext.applyCollation(entityClass, iterable::collation);
        return iterable;
    });
    if (resultClass == Object.class || mongoDriverCompatibleType != resultClass) {
        MongoConverter converter = getConverter();
        DefaultDbRefResolver dbRefResolver = new DefaultDbRefResolver(mongoDbFactory);
        result = result.map((source) -> converter.mapValueToTargetType(source, distinctQueryContext.getMostSpecificConversionTargetType(resultClass, entityClass), dbRefResolver));
    }
    try {
        return (List<T>) result.into(new ArrayList<>());
    } catch (RuntimeException e) {
        throw potentiallyConvertRuntimeException(e, exceptionTranslator);
    }
}