如何在Elasticsearch中按字段特定值排序并结合评分排序？

需求说明 #

搜索包含obama的文档时，需按照以下规则排序：

• 优先展示countryCode为US的记录
• 其次展示countryCode为AR的记录
• 最后展示countryCode为ES的记录
• 同一国家内的文档，按rating字段降序排列

原方案问题 #

你尝试使用带boost的filter查询，但filter上下文仅负责过滤文档，不会参与评分计算，其中的boost参数无法影响文档的_score值，因此无法通过_score排序实现自定义的国家优先级。原查询如下：

{
    "query": {
        "bool": {
            "must": [
                {
                    "bool": {
                        "must": [
                            {
                                "match_phrase_prefix": {
                                    "name": {
                                        "query": "obama"
                                    }
                                }
                            }
                        ],
                        "boost": 2.0
                    }
                }
            ],
            "filter": {
                "bool": {
                    "should": [
                        {
                            "term": {
                                "countryCode": {
                                    "value": "US",
                                    "boost": 4
                                }
                            }
                        },
                        {
                            "term": {
                                "countryCode": {
                                    "value": "AR",
                                    "boost": 3
                                }
                            }
                        },
                        {
                            "term": {
                                "countryCode": {
                                    "value": "ES",
                                    "boost": 2
                                }
                            }
                        }
                    ]
                }
            }
        }
    },
    "size": 50,
    "sort": [
        {
            "rating": {
                "order": "desc"
            }
        },
        {
            "_score": {
                "order": "desc"
            }
        }
    ]
}

正确实现方案 #

要实现自定义的国家优先级排序，最直接的方式是在sort中使用脚本排序，通过脚本为不同countryCode赋值权重，先按权重降序排列，再按rating降序排列。调整后的完整查询如下：

{
    "query": {
        "bool": {
            "must": [
                {
                    "match_phrase_prefix": {
                        "name": {
                            "query": "obama"
                        }
                    }
                }
            ],
            "filter": {
                "terms": {
                    "countryCode": ["US", "AR", "ES"]
                }
            }
        }
    },
    "size": 50,
    "sort": [
        {
            "_script": {
                "type": "number",
                "script": {
                    "source": """
                        if (doc['countryCode'].value == 'US') {
                            return 3;
                        } else if (doc['countryCode'].value == 'AR') {
                            return 2;
                        } else if (doc['countryCode'].value == 'ES') {
                            return 1;
                        } else {
                            return 0;
                        }
                    """
                },
                "order": "desc"
            }
        },
        {
            "rating": {
                "order": "desc"
            }
        }
    ]
}

方案说明 #

1. 过滤逻辑简化：把原filter中的bool should替换为terms查询，直接指定允许的countryCode值，逻辑更简洁
2. 脚本排序核心：通过脚本为每个countryCode分配不同的权重值（US=3、AR=2、ES=1），权重越高排序越靠前
3. 二级排序：同一权重组内的文档，按rating字段降序排列，满足同国家内的排序要求

预期结果 #

执行上述查询后，返回结果将符合预期顺序：

[
    {"name":"obama a", "countryCode":"US", "rating":5},
    {"name":"obama b", "countryCode":"US", "rating":4},
    {"name":"obama ac", "countryCode":"AR", "rating":3},
    {"name":"obama ess", "countryCode":"ES", "rating":3.5}
]

内容的提问来源于stack exchange，提问作者Rohit Kumar Mishra