Elasticsearch权威指南：使用脚本评分（script_score）实现自定义相关性计算

Elasticsearch权威指南：使用脚本评分（script_score）实现自定义相关性计算

elasticsearch-definitive-guide The Definitive Guide to Elasticsearch 项目地址: https://gitcode.com/gh_mirrors/el/elasticsearch-definitive-guide

理解脚本评分的基本概念

在Elasticsearch的搜索功能中，script_score是一个强大的工具，它允许我们通过自定义脚本实现复杂的相关性评分逻辑。当内置的评分函数无法满足特定业务需求时，脚本评分提供了极大的灵活性。

典型应用场景：利润计算示例

假设我们经营一个度假屋租赁平台，需要将利润因素纳入搜索结果的排序中。利润计算涉及三个关键因素：

1. 每晚价格（price）
2. 用户会员等级对应的折扣（discount）
3. 扣除折扣后的利润率（margin）

利润计算算法如下：

if (price < threshold) {
  profit = price * margin
} else {
  profit = price * (1 - discount) * margin
}

为了不影响其他评分因素（如位置、受欢迎程度等），我们将利润转换为相对于目标利润的百分比：

return profit / target

实现脚本评分的具体步骤

1. 编写Groovy脚本

Elasticsearch默认使用Groovy作为脚本语言，其语法与JavaScript相似。实现上述逻辑的脚本如下：

price = doc['price'].value
margin = doc['margin'].value

if (price < threshold) {
  return price * margin / target
}
return price * (1 - discount) * margin / target

2. 参数传递技巧

通过params传递变量值，可以避免每次修改参数都重新编译脚本：

"params": {
  "threshold": 80,
  "discount": 0.1,
  "target": 10
}

3. 完整查询示例

将脚本评分与其他评分函数结合使用：

{
  "function_score": {
    "functions": [
      { ...位置评分... },
      { ...价格评分... },
      {
        "script_score": {
          "params": {
            "threshold": 80,
            "discount": 0.1,
            "target": 10
          },
          "script": "price = doc['price'].value; margin = doc['margin'].value; if (price < threshold) { return price * margin / target }; return price * (1 - discount) * margin / target;"
        }
      }
    ]
  }
}

性能优化建议

虽然脚本评分功能强大，但需要注意性能问题：

1. 预计算数据：尽可能将可预先计算的信息存入文档
2. 语言选择：Groovy速度较快，但原生Java脚本性能更佳
3. 结果重评分：使用rescore功能只对高分文档应用脚本计算

高级功能

在脚本中，开发者可以访问：

• 文档字段值
• 当前_score
• 词频、逆文档频率
• 字段长度规范等文本评分信息

实际应用建议

1. 对于复杂的业务逻辑，脚本评分提供了极大的灵活性
2. 在实现前应评估是否真的需要自定义脚本，有时通过合理的文档设计和查询组合也能达到类似效果
3. 生产环境中应对脚本性能进行充分测试

通过合理使用脚本评分功能，开发者可以构建出既符合业务需求又保持良好性能的搜索系统。

elasticsearch-definitive-guide The Definitive Guide to Elasticsearch 项目地址: https://gitcode.com/gh_mirrors/el/elasticsearch-definitive-guide