Examples with SearchHits org.springframework.data.elasticsearch.core.SearchHits used on opensource projects

Example 6 with SearchHits

use of org.springframework.data.elasticsearch.core.SearchHits in project spring-data-elasticsearch by spring-projects.

the class ElasticsearchStringQuery method execute.

@Override
public Object execute(Object[] parameters) {
    Class<?> clazz = queryMethod.getResultProcessor().getReturnedType().getDomainType();
    ParametersParameterAccessor accessor = new ParametersParameterAccessor(queryMethod.getParameters(), parameters);
    StringQuery stringQuery = createQuery(accessor);
    Assert.notNull(stringQuery, "unsupported query");
    if (queryMethod.hasAnnotatedHighlight()) {
        stringQuery.setHighlightQuery(queryMethod.getAnnotatedHighlightQuery());
    }
    IndexCoordinates index = elasticsearchOperations.getIndexCoordinatesFor(clazz);
    Object result = null;
    if (isCountQuery()) {
        result = elasticsearchOperations.count(stringQuery, clazz, index);
    } else if (queryMethod.isPageQuery()) {
        stringQuery.setPageable(accessor.getPageable());
        SearchHits<?> searchHits = elasticsearchOperations.search(stringQuery, clazz, index);
        if (queryMethod.isSearchPageMethod()) {
            result = SearchHitSupport.searchPageFor(searchHits, stringQuery.getPageable());
        } else {
            result = SearchHitSupport.unwrapSearchHits(SearchHitSupport.searchPageFor(searchHits, stringQuery.getPageable()));
        }
    } else if (queryMethod.isStreamQuery()) {
        stringQuery.setPageable(accessor.getPageable().isPaged() ? accessor.getPageable() : PageRequest.of(0, DEFAULT_STREAM_BATCH_SIZE));
        result = StreamUtils.createStreamFromIterator(elasticsearchOperations.searchForStream(stringQuery, clazz, index));
    } else if (queryMethod.isCollectionQuery()) {
        stringQuery.setPageable(accessor.getPageable().isPaged() ? accessor.getPageable() : Pageable.unpaged());
        result = elasticsearchOperations.search(stringQuery, clazz, index);
    } else {
        result = elasticsearchOperations.searchOne(stringQuery, clazz, index);
    }
    return (queryMethod.isNotSearchHitMethod() && queryMethod.isNotSearchPageMethod()) ? SearchHitSupport.unwrapSearchHits(result) : result;
}