com.sksamuel.elastic4s.searches.aggs.GeoDistanceAggregationBuilder.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elastic4s-tcp_2.13 Show documentation

elastic4s-tcp

The newest version!

package com.sksamuel.elastic4s.searches.aggs

import com.sksamuel.elastic4s.{EnumConversions, ScriptBuilder}
import org.elasticsearch.search.aggregations.AggregationBuilders
import org.elasticsearch.search.aggregations.bucket.range.GeoDistanceAggregationBuilder

import scala.collection.JavaConverters._

object GeoDistanceAggregationBuilder {

  import com.sksamuel.elastic4s.EnumConversions._

  def apply(agg: GeoDistanceAggregationDefinition): GeoDistanceAggregationBuilder = {

    val builder = AggregationBuilders.geoDistance(agg.name, agg.origin)

    agg.field.foreach(builder.field)
    agg.missing.foreach(builder.missing)
    agg.format.foreach(builder.format)
    agg.keyed.foreach(builder.keyed)
    agg.distanceType.map(EnumConversions.geoDistance).foreach(builder.distanceType)
    agg.unit.map(EnumConversions.distanceUnit).foreach(builder.unit)
    agg.script.map(ScriptBuilder.apply).foreach(builder.script)

    agg.ranges.foreach {
      case (Some(key), from, to) => builder.addRange(key, from, to)
      case (None, from, to)      => builder.addRange(from, to)
    }

    agg.unboundedFrom.foreach {
      case (Some(key), from) => builder.addUnboundedFrom(key, from)
      case (None, from)      => builder.addUnboundedFrom(from)
    }

    agg.unboundedTo.foreach {
      case (Some(key), to) => builder.addUnboundedTo(key, to)
      case (None, to)      => builder.addUnboundedTo(to)
    }

    SubAggsFn(builder, agg.subaggs)
    if (agg.metadata.nonEmpty) builder.setMetaData(agg.metadata.asJava)
    builder
  }
}