如何在Elasticsearch Painless脚本中基于传入HashMap计算日期差平均值聚合？

问题背景 #

之前在Elasticsearch中使用expression脚本计算同一索引内两个日期字段的差值平均值聚合，脚本如下：

"avg": {
  "script": {
    "source": "(doc['lastFixed'] - doc['firstFound'])/ (1000*60*60)",
    "lang": "expression"
  }
}

现在第二个日期存储在Java HashMap中（键为文档ID，值为对应日期，共约100条数据），需要将该HashMap作为参数传入Painless脚本，匹配文档ID后计算符合条件的日期差平均值聚合。尝试了以下脚本但不确定是否可行（循环内未返回或保存计算结果）：

"for(entry in param.dataset.entrySet())"+
  "{if (item.getKey() == doc['id'])"+
  "{ (doc['lastFixed'] - item.getValue())/ (1000*60*60) }"+
"}"

解决方案 #

原脚本的问题 #

1. 没必要遍历HashMap：遍历整个entrySet效率低下，直接通过文档ID作为键取值即可。
2. 缺少返回值：avg聚合要求脚本返回单个数值，原脚本仅在循环内计算但未返回结果，聚合无法获取有效数据。

正确的Painless脚本实现 #

不需要遍历，直接通过文档ID从传入的HashMap中匹配日期，计算差值后返回，找不到匹配ID时返回null（该文档会被聚合忽略）。完整的聚合请求示例如下：

{
  "aggs": {
    "avg_fix_duration": {
      "avg": {
        "script": {
          "source": """
            // 获取当前文档的ID
            def docId = doc['id'].value;
            // 从传入的参数中获取对应日期
            def externalDate = params.dataset[docId];
            // 存在匹配日期则计算差值（转换为小时），否则返回null
            if (externalDate != null) {
              return (doc['lastFixed'].value - externalDate) / (1000 * 60 * 60);
            }
            return null;
          """,
          "lang": "painless",
          "params": {
            "dataset": {
              "doc_1": 1620000000000,
              "doc_2": 1620100000000
              // 这里传入你的HashMap对应的键值对，日期需为毫秒时间戳
            }
          }
        }
      }
    }
  }
}

关键注意事项 #

• 日期格式一致性：确保HashMap中的日期值和ES文档里的lastFixed字段类型一致，建议统一使用毫秒时间戳，避免日期格式转换错误。
• 参数传递：在Java代码中构造请求时，直接将HashMap作为params中的dataset参数传入即可，Elasticsearch会自动解析为Painless可识别的Map结构。
• 性能优化：100条数据的Map规模很小，直接通过键取值的时间复杂度为O(1)，远优于遍历的O(n)，不会带来性能问题。

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