com.sksamuel.elastic4s.index.IndexExecutables.scala Maven / Gradle / Ivy

Go to download
package com.sksamuel.elastic4s.index

import com.sksamuel.elastic4s.indexes.IndexDefinition
import com.sksamuel.elastic4s.json.XContentFactory
import com.sksamuel.elastic4s.{EnumConversions, Executable, XContentFieldValueWriter}
import org.elasticsearch.action.index.{IndexRequestBuilder, IndexResponse}
import org.elasticsearch.client.Client
import org.elasticsearch.common.xcontent.XContentType

import scala.concurrent.Future

trait IndexExecutables extends IndexShowImplicitsTcp {

  implicit object IndexDefinitionExecutable extends Executable[IndexDefinition, IndexResponse, RichIndexResponse] {

    def builder(c: Client, t: IndexDefinition): IndexRequestBuilder = {
      val builder = c.prepareIndex(t.indexAndType.index, t.indexAndType.`type`)
      t.id.map(_.toString).foreach(builder.setId)
      t.source match {
        case Some(json) => builder.setSource(json, XContentType.JSON)
        case _ =>
          val source = XContentFactory.obj()
          t.fields.foreach(XContentFieldValueWriter(source, _))
          builder.setSource(source.string, XContentType.JSON)
      }
      t.parent.foreach(builder.setParent)
      t.refresh.map(EnumConversions.refreshPolicy).foreach(builder.setRefreshPolicy)
      t.version.foreach(builder.setVersion)
      t.versionType.map(EnumConversions.versionType).foreach(builder.setVersionType)
      t.routing.foreach(builder.setRouting)
      t.pipeline.foreach(builder.setPipeline)
      t.source.foreach(builder.setSource(_, XContentType.JSON))
      t.createOnly.foreach(builder.setCreate)
      builder
    }

    override def apply(c: Client, t: IndexDefinition): Future[RichIndexResponse] = {
      val req = builder(c, t)
      injectFutureAndMap(req.execute)(RichIndexResponse.apply)
    }
  }

  implicit class IndexDefinitionShowOps(f: IndexDefinition) {
    def show: String = IndexShowTcp.show(f)
  }
}