com.sksamuel.elastic4s.handlers.searches.queries.geo.GeoBoundingBoxQueryBodyFn.scala Maven / Gradle / Ivy

Go to download
package com.sksamuel.elastic4s.handlers.searches.queries.geo

import com.sksamuel.elastic4s.EnumConversions
import com.sksamuel.elastic4s.json.{XContentBuilder, XContentFactory}
import com.sksamuel.elastic4s.requests.searches.queries.geo.GeoBoundingBoxQuery

object GeoBoundingBoxQueryBodyFn {

  def apply(q: GeoBoundingBoxQuery): XContentBuilder = {
    val builder = XContentFactory.jsonBuilder().startObject("geo_bounding_box")

    q.geoExecType.map(EnumConversions.geoExecType).foreach(builder.field("type", _))
    q.ignoreUnmapped.foreach(builder.field("ignore_unmapped", _))
    q.validationMethod.map(EnumConversions.geoValidationMethod).foreach(builder.field("validation_method", _))
    q.queryName.foreach(builder.field("_name", _))

    builder.startObject(q.field)

    q.corners.foreach { corners =>
      builder.startObject("top_left")
      builder.field("lat", corners.top)
      builder.field("lon", corners.left)
      builder.endObject()
      builder.startObject("bottom_right")
      builder.field("lat", corners.bottom)
      builder.field("lon", corners.right)
      builder.endObject()
    }

    q.geohash.foreach {
      case (topleft, bottomright) =>
        builder.field("top_left", topleft)
        builder.field("bottom_right", bottomright)
    }

    builder.endObject().endObject().endObject()
  }
}