com.sksamuel.elastic4s.handlers.bulk.BulkBuilderFn.scala Maven / Gradle / Ivy

Go to download
package com.sksamuel.elastic4s.handlers.bulk

import com.sksamuel.elastic4s.handlers
import com.sksamuel.elastic4s.handlers.VersionTypeHttpString
import com.sksamuel.elastic4s.handlers.index.IndexContentBuilder
import com.sksamuel.elastic4s.handlers.update.UpdateBuilderFn
import com.sksamuel.elastic4s.json.XContentFactory
import com.sksamuel.elastic4s.requests.bulk.BulkRequest
import com.sksamuel.elastic4s.requests.common.FetchSourceContext
import com.sksamuel.elastic4s.requests.delete.DeleteByIdRequest
import com.sksamuel.elastic4s.requests.indexes.IndexRequest
import com.sksamuel.elastic4s.requests.update.UpdateRequest

object BulkBuilderFn {

  def apply(bulk: BulkRequest): Iterator[String] = {
    bulk.requests.iterator.flatMap {
      case index: IndexRequest =>
        val builder = XContentFactory.jsonBuilder()
        val createOrIndex = if (index.createOnly.getOrElse(false)) "create" else "index"
        builder.startObject(createOrIndex)
        builder.field("_index", index.index.name)
        index.id.foreach(id => builder.field("_id", id))
        index.parent.foreach(builder.field("_parent", _))
        index.routing.foreach(builder.field("routing", _))
        index.version.foreach(builder.field("version", _))
        index.ifPrimaryTerm.foreach(builder.field("if_primary_term", _))
        index.ifSeqNo.foreach(builder.field("if_seq_no", _))
        index.versionType.foreach(versionType => builder.field("version_type", VersionTypeHttpString(versionType)))
        index.pipeline.foreach(builder.field("pipeline", _))
        builder.endObject()
        builder.endObject()

        Iterator(builder.string, IndexContentBuilder(index))

      case delete: DeleteByIdRequest =>
        val builder = XContentFactory.jsonBuilder()
        builder.startObject("delete")
        builder.field("_index", delete.index.index)
        builder.field("_id", delete.id)
        delete.parent.foreach(builder.field("_parent", _))
        delete.routing.foreach(builder.field("routing", _))
        delete.version.foreach(builder.field("version", _))
        delete.ifPrimaryTerm.foreach(builder.field("if_primary_term", _))
        delete.ifSeqNo.foreach(builder.field("if_seq_no", _))
        delete.versionType.foreach(versionType => builder.field("version_type", handlers.VersionTypeHttpString(versionType)))
        builder.endObject()
        builder.endObject()

        Iterator.single(builder.string)

      case update: UpdateRequest =>
        val builder = XContentFactory.jsonBuilder()
        builder.startObject("update")
        builder.field("_index", update.index.index)
        builder.field("_id", update.id)
        update.parent.foreach(builder.field("_parent", _))
        update.routing.foreach(builder.field("routing", _))
        update.version.foreach(builder.field("version", _))
        update.ifPrimaryTerm.foreach(builder.field("if_primary_term", _))
        update.ifSeqNo.foreach(builder.field("if_seq_no", _))
        update.versionType.foreach(builder.field("version_type", _))
        update.retryOnConflict.foreach(builder.field("retry_on_conflict", _))
        update.fetchSource.foreach {
          case FetchSourceContext(fetchSource, includes, excludes) =>
            builder.field("_source", fetchSource)
            if (includes.nonEmpty) builder.field("_source_includes", includes.mkString(","))
            if (excludes.nonEmpty) builder.field("_source_excludes", excludes.mkString(","))
        }
        builder.endObject()
        builder.endObject()

        Iterator(builder.string, UpdateBuilderFn(update).string)
    }
  }
}