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

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
}

这样你就可以从elasticsearchSearchResponse获取你需要的信息了。

关于您的第二个问题：据我所知，在致电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);