如何在Elasticsearch中按指定整数列表优先排序复合聚合结果

你需要通过painless脚本自定义排序权重，搭配bucket_sort子聚合对composite聚合返回的结果重排序，即可实现要求的优先级规则：

• 第一优先级：id匹配指定列表[2,3,6]的排在最前，且严格遵循列表顺序
• 第二优先级：非指定id按id升序排列
• 第三优先级：按name升序排列

以下是可直接使用的DSL示例：

{
  "size": 0,
  "aggs": {
    "your_composite_agg": { // 替换为你的复合聚合名称
      "composite": {
        "sources": [
          {"id": {"terms": {"field": "id"}}},
          {"name": {"terms": {"field": "name.keyword"}}} // 字符串字段需使用keyword类型排序
        ]
      },
      "aggs": {
        "custom_sort": {
          "bucket_sort": {
            "sort": [
              // 第一排序规则：自定义优先级脚本
              {
                "_script": {
                  "type": "number",
                  "script": {
                    "lang": "painless",
                    "params": {
                      "priority_ids": [2,3,6] // 可按需修改优先级ID列表
                    },
                    "source": """
                      int idx = params.priority_ids.indexOf(doc['id'].value);
                      return idx == -1 ? params.priority_ids.length : idx;
                    """
                  },
                  "order": "asc"
                }
              },
              // 第二排序规则：id升序
              {"id": {"order": "asc"}},
              // 第三排序规则：name升序
              {"name": {"order": "asc"}}
            ],
            "size": 1000 // 按需调整返回的bucket总数量
          }
        }
      }
    }
  }
}

关键逻辑说明 #

• 脚本会先取当前id在priority_ids列表中的下标，匹配到的返回对应下标（2→0、3→1、6→2），未匹配到的返回列表长度3，按升序排序后匹配的结果自然排在最前，且顺序和指定列表完全一致
• 如果不需要分页返回全量结果，把bucket_sort的size参数调整为大于你的总bucket数量即可

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