如何基于Spring Data Elasticsearch通过Swagger API获取ES数据聚合结果

你当前的实现存在3个核心问题，导致聚合结果无法正常返回：

• Repository层@Query注解的JSON结构不符合Elasticsearch查询规范，存在语法错误
• Repository层直接返回DTO类型，Spring Data Elasticsearch无法正确封装聚合结果
• Controller/Service层仅提取了分页文档列表，未提取聚合数据，也没有设计承载聚合结果的响应结构

1. 修复Repository层查询逻辑 #

首先修正@Query的JSON结构，补全外层query包裹节点，修正JSON语法，同时将返回值改为实体类Member，避免聚合封装失败：

/**
 * Spring Data Elasticsearch repository for the {@link Member} entity.
 */
public interface MemberSearchRepository extends ElasticsearchRepository<Member, Long> {

    @Query("{\"query\": {\"query_string\": {\"query\": \"?0\"}}, \"aggs\": {\"sdoh_aggs\": {\"terms\": {\"field\": \"populationStreams.adultsWithChronicConditionIndicator\"}}}}")
    AggregatedPage<Member> searchWithAggs(String query, Pageable pageable);

}

注意：如果你的populationStreams是nested嵌套类型，需要将聚合改为nested聚合，否则无法正确统计；如果该字段是text类型，需要用.keyword子字段做聚合，或者将字段类型改为boolean/keyword，否则会抛出fielddata禁用的错误。

2. 定义统一的搜索响应DTO #

新建响应类，同时承载分页数据和聚合统计结果，方便前端解析：

import lombok.Data;
import java.util.List;
import java.util.Map;

@Data
public class MemberSearchResultDTO {
    // 分页匹配的成员列表
    private List<MemberDTO> members;
    // 聚合统计结果，key为字段值(true/false)，value为对应记录数
    private Map<String, Long> conditionIndicatorStats;
    // 分页相关字段
    private long totalElements;
    private int totalPages;
}

3. 修正Service层逻辑，提取聚合结果 #

在Service层完成实体转DTO，同时从AggregatedPage中提取terms聚合的桶数据，组装成最终响应：

import org.elasticsearch.search.aggregations.bucket.terms.StringTerms;
import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;

@Transactional(readOnly = true)
public MemberSearchResultDTO search(String query, Pageable pageable) {
    log.debug("Request to search for a page of Members for query {}", query);
    AggregatedPage<Member> page = memberSearchRepository.searchWithAggs(query, pageable);
    
    // 转换实体为DTO
    List<MemberDTO> memberDTOS = page.getContent().stream()
            .map(memberMapper::toDto)
            .collect(Collectors.toList());
    
    // 提取聚合结果
    Map<String, Long> stats = new HashMap<>();
    if (page.hasAggregations()) {
        StringTerms aggs = page.getAggregations().get("sdoh_aggs");
        for (StringTerms.Bucket bucket : aggs.getBuckets()) {
            // key为字段值(true/false)，value为对应文档数
            stats.put(bucket.getKeyAsString(), bucket.getDocCount());
        }
    }
    
    // 组装返回结果
    MemberSearchResultDTO result = new MemberSearchResultDTO();
    result.setMembers(memberDTOS);
    result.setConditionIndicatorStats(stats);
    result.setTotalElements(page.getTotalElements());
    result.setTotalPages(page.getTotalPages());
    return result;
}

优化提示：如果你只需要统计值为true的记录总数，不需要统计false的数量，可以把terms聚合换成filter聚合，查询性能更高，示例聚合结构：

"aggs": {
  "true_count": {
    "filter": {"term": {"populationStreams.adultsWithChronicConditionIndicator": true}}
  }
}

提取时直接取( (Filter) page.getAggregations().get("true_count") ).getDocCount()即可得到true的总数量。

4. 修正Controller层返回逻辑 #

修改Controller的返回值类型，直接返回组装好的MemberSearchResultDTO：

@GetMapping("/_search/members")
public ResponseEntity<MemberSearchResultDTO> searchMembers(@RequestParam String query, Pageable pageable) {
    log.debug("REST request to search for a page of Members for query {}", query);
    MemberSearchResultDTO result = memberService.search(query, pageable);
    // 分页头可根据需要保留
    HttpHeaders headers = PaginationUtil.generatePaginationHttpHeaders(
            ServletUriComponentsBuilder.fromCurrentRequest(), 
            new PageImpl<>(result.getMembers(), pageable, result.getTotalElements())
    );
    return ResponseEntity.ok().headers(headers).body(result);
}

重启服务后通过Swagger调用接口，响应体中会同时返回成员列表和conditionIndicatorStats字段，其中key为true对应的value就是你需要统计的记录总数，前端直接取值即可。

内容的提问来源于stack exchange，提问作者cluis92