Question

我正在为项目使用spring-data-elasticsearch为其提供全文搜索功能。我们将真实数据保存在关系数据库和相关元数据中以及弹性搜索中的相应ID。因此，对于搜索结果，只需要id字段，因为将从关系数据库中检索实际数据。

我正在构建基于搜索条件的搜索查询，然后执行queryForIds（）：

    SearchQuery searchQuery = new NativeSearchQueryBuilder()
            .withIndices(indexName)
            .withTypes(typeName)
            .withQuery(getQueryBuilder(searchParams))
            .withPageable(pageable)
            .build();

    return elasticsearchTemplate.queryForIds(searchQuery);

如果我还需要该特定searchQuery的总计数，我可以进行另一次elasticsearchTemplate.count(searchQuery)调用，但根据我的理解，这将是多余的。我认为有一种方法可以通过在一次调用中使用类似elasticsearchTemplate.queryForPage()的内容来获取ID和总计数列表。

另外，我可以在queryForPage(SearchQuery query, Class<T> clazz, SearchResultMapper mapper)中使用未注明@Document的自定义类吗？实际的文档类非常大，如果我不确定传递大类是否会导致引擎上的任何额外负载，因为有超过100个字段要进行json映射，但我需要的只是id字段。无论如何，我将在查询构建器中有一个.withFields("id")。

Answer 1

如果你想阻止对elasticsearch的两次调用，我建议写一个自定义的ResultsExtractor：

SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices(indexName)
        .withTypes(typeName)
        .withQuery(queryBuilder)
        .withPageable(pageable)
        .build();
SearchResult result = template.query(searchQuery, new ResultsExtractor<SearchResult>() {

    @Override
    public SearchResult extract(SearchResponse response) {
        long totalHits = response.getHits()
                .totalHits();
        List<String> ids = new ArrayList<String>();
        for (SearchHit hit : response.getHits()) {
            if (hit != null) {
                ids.add(hit.getId());
            }
        }
        return new SearchResult(ids, totalHits);
    }
});
System.out.println(result.getIds());
System.out.println(result.getCount());

其中SearchResult是自定义类：

public class SearchResult {

    List<String> ids;
    long count;

    //getter and setter
}

通过这种方式，您可以从elasticsearch SearchResponse获取所需信息。

关于你的第二个问题：据我所知，在致电queryForPage(SearchQuery query, Class<T> clazz, SearchResultMapper mapper)时未检查传递的类的@Document注释。试试吧！

Answer 2

也可以考虑使用AggregatedPage<T>。您可以像Pageable<T>中一样获得记录总数，页面总数，当前页面记录等。

SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices(indexName)
        .withTypes(typeName)
        .withQuery(queryBuilder)
        .withPageable(pageable)
        .build();

AggregatedPage<ElasticDTO> queryResult = elasticsearchTemplate.queryForPage(searchQuery , ElasticDTO.class);

在spring-data-elasticsearch中使用ElasticsearchTemplate获取计数和id列表

2 个答案: