使用客户端 API 解析深度嵌套的 elasticsearch 聚合

use*_*049 5 java scala elasticsearch

我正在使用 Java 客户端 API 来恢复聚合。以下是我正在处理的结构。

aggregations
top_models
   buckets
       key : "BMW"
      doc_count : 3
      top_models
           buckets
               key : "X5"
               doc_count : 2
              top_hits
                   source
                      model : X5
                      color : Red
                  source
                     model:X5
                    color : White
           key : "X3"
               doc_count : 1
              top_hits
                   source
                      model : X3
                      color : Red
      key : "Mercedes"
      doc_count : 2
      top_models
           buckets
               key : "Benz"
               doc_count : 1
               top_hits
                   source
                      model : Benz
                      color : Red

              key : "ML"
              doc_count : 1
              top_hits
                   source
                      model : ML
                      color : Black
Run Code Online (Sandbox Code Playgroud)

我正在尝试按照(玩具)代码来检索所有结果。

def getAggregations(aggres: Option[Aggregations]): Option[Iterable[Any]] = {


aggres map { agg =>


  val aggS = agg.asMap().asScala


  aggS map {


    case (name, termAgg: Terms) => getBuckets(Option(termAgg.getBuckets()))


    case (name, topHits: TopHits) =>


      val tHits = Option(topHits.getHits())


      tHits map { th => getTopHits(th.asScala)
    }


    case (h, a: InternalAvg) => println(h + "=>" + a.getValue());


  }





}


}





def getBuckets(buckets: Option[java.util.Collection[Bucket]]) = {


buckets map { bks =>


  val bksS = bks.asScala


  bksS map { b =>


    println("Bucket Key =>" + b.getKey())


    println("Doc count =>" + b.getDocCount())


    getAggregations(Option(b.getAggregations())


  }


  }

 }
Run Code Online (Sandbox Code Playgroud)

需要将最终结果填充到此类中

case class FinalResponse(bucketName: String, count: Long, children: List[FinalResponse])
Run Code Online (Sandbox Code Playgroud)

由于聚合和存储桶之间存在嵌套关系,检索所有聚合结果变得非常复杂。你如何处理这个问题?

小智 2

在我之前的项目中,我们使用这种方式从弹性搜索中反序列化复杂对象:

Elasticsearch 允许您获取结果集每个元素的纯 json 视图(作为字符串)。我们只是简单地使用 Jacson json 库和 scala 模块,并创建类似 pojo 的类来反序列化数据。

Elastic search java api 是可怕的多层嵌套映射集。把它给忘了。

val hits = qb.execute().actionGet().getHits().getHits().asScala
hits.map { hit =>
  (hit.getId, hit.getSourceAsString, hit.getVersion)
}
Run Code Online (Sandbox Code Playgroud)

对于聚合它也应该可用