Spring Elasticsearch能否按首条结果分数的80%范围限制返回结果？

完全可行，以下是两种实用的实现方案：

方案一：分两次查询（推荐，性能更优） #

先获取最高分数的文档，再基于该分数的80%作为阈值过滤结果：

1. 执行一次仅返回第一条结果的查询，提取其分数值
2. 计算阈值：最高分数 * 0.8
3. 构造新查询，添加对_score字段的范围过滤，只保留分数在[阈值, 最高分数]区间内的文档

代码示例 #

// 注入Spring Elasticsearch操作模板
@Autowired
private ElasticsearchRestTemplate elasticsearchRestTemplate;

public List<YourEntity> searchWithScoreFilter(QueryBuilder originalQuery) {
    // 第一步：获取首条文档的最高分数
    SearchQuery topScoreQuery = new NativeSearchQueryBuilder()
            .withQuery(originalQuery)
            .withPageable(PageRequest.of(0, 1)) // 只取第一条
            .build();
    SearchHits<YourEntity> topHits = elasticsearchRestTemplate.search(topScoreQuery, YourEntity.class);
    if (topHits.getTotalHits() == 0) {
        return Collections.emptyList();
    }
    float topScore = topHits.getSearchHit(0).getScore();
    float scoreThreshold = topScore * 0.8f;

    // 第二步：构造带分数过滤的查询
    SearchQuery filteredQuery = new NativeSearchQueryBuilder()
            .withQuery(QueryBuilders.boolQuery()
                    .must(originalQuery) // 保留原始查询条件
                    .filter(QueryBuilders.rangeQuery("_score")
                            .gte(scoreThreshold)
                            .lte(topScore)) // 过滤分数区间
            )
            .build();
    SearchHits<YourEntity> filteredHits = elasticsearchRestTemplate.search(filteredQuery, YourEntity.class);
    return filteredHits.getSearchHits().stream()
            .map(SearchHit::getContent)
            .collect(Collectors.toList());
}

方案二：脚本过滤（不推荐，性能较差） #

通过Painless脚本动态比较文档分数与阈值，但需要先获取最高分数并传入脚本参数：

代码示例 #

public List<YourEntity> searchWithScriptFilter(QueryBuilder originalQuery) {
    // 先获取最高分数（同方案一的第一步）
    SearchQuery topScoreQuery = new NativeSearchQueryBuilder()
            .withQuery(originalQuery)
            .withPageable(PageRequest.of(0, 1))
            .build();
    SearchHits<YourEntity> topHits = elasticsearchRestTemplate.search(topScoreQuery, YourEntity.class);
    if (topHits.getTotalHits() == 0) {
        return Collections.emptyList();
    }
    float topScore = topHits.getSearchHit(0).getScore();
    float scoreThreshold = topScore * 0.8f;

    // 构造脚本过滤查询
    Script scoreFilterScript = new Script(
            ScriptType.INLINE,
            "painless",
            "_score >= params.threshold && _score <= params.topScore",
            Map.of("threshold", scoreThreshold, "topScore", topScore)
    );
    SearchQuery scriptFilterQuery = new NativeSearchQueryBuilder()
            .withQuery(QueryBuilders.boolQuery()
                    .must(originalQuery)
                    .filter(QueryBuilders.scriptQuery(scoreFilterScript))
            )
            .build();
    SearchHits<YourEntity> filteredHits = elasticsearchRestTemplate.search(scriptFilterQuery, YourEntity.class);
    return filteredHits.getSearchHits().stream()
            .map(SearchHit::getContent)
            .collect(Collectors.toList());
}

注意事项 #

• 方案一的范围过滤依赖Elasticsearch对_score的内置处理，性能远优于脚本过滤，数据量较大时优先选择
• 若自定义了排序规则，需确保首条文档的分数是全局最高的（默认排序为_score降序，满足要求）
• 浮点数计算可能存在精度误差，可根据业务需求适当调整阈值的精度（如保留两位小数）

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