com.sksamuel.elastic4s.ValidateDsl.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elastic4s-core_2.10 Show documentation

elastic4s-core

There is a newer version: 6.0.0-rc1

package com.sksamuel.elastic4s

import org.elasticsearch.action.admin.indices.validate.query.{ValidateQueryAction, ValidateQueryRequestBuilder, ValidateQueryResponse}
import org.elasticsearch.client.Client
import org.elasticsearch.common.xcontent.XContentHelper

import scala.concurrent.Future

trait ValidateDsl {

  implicit object ValidateDefinitionExecutable
    extends Executable[ValidateDefinition, ValidateQueryResponse, ValidateQueryResponse] {
    override def apply(c: Client, t: ValidateDefinition): Future[ValidateQueryResponse] = {
      injectFuture(c.admin.indices.validateQuery(t.build, _))
    }
  }

  implicit object ValidateDefinitionShow extends Show[ValidateDefinition] {
    override def show(f: ValidateDefinition): String = {
      Option(f.q).fold("")(q => XContentHelper.convertToJson(q.builder.buildAsBytes, true, true))
    }
  }

  implicit class ValidateDefinitionShowOps(f: ValidateDefinition) {
    def show: String = ValidateDefinitionShow.show(f)
  }
}

case class ValidateDefinition(index: String, `type`: String) {

  private[elastic4s] var q: QueryDefinition = _

  val _builder = {
    new ValidateQueryRequestBuilder(ProxyClients.indices, ValidateQueryAction.INSTANCE)
      .setIndices(index)
      .setTypes(`type`)
  }

  def build = _builder.request

  /**
   * Adds a single string query to this search
   *
   * @param string the query string
   *
   * @return this
   */
  def query(string: String): this.type = {
    this.q = new QueryStringQueryDefinition(string)
    _builder.setQuery(q.builder)
    this
  }

  def rewrite(rewrite: Boolean): this.type = {
    _builder.setRewrite(rewrite)
    this
  }

  def query(block: => QueryDefinition): this.type = {
    this.q = block
    _builder.setQuery(q.builder)
    this
  }

  def explain(ex: Boolean): this.type = {
    _builder.setExplain(ex)
    this
  }
}