jp.co.bizreach.elasticsearch4s.BulkAction.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elastic-scala-httpclient_2.12 Show documentation

elastic-scala-httpclient

The newest version!

package jp.co.bizreach.elasticsearch4s

import JsonUtils._

sealed trait BulkAction {
  def jsonString: String
}

object BulkAction {

  case class Index(config: ESConfig, doc: AnyRef) extends BulkAction {
    def jsonString: String = {
      s"""{ "index" : { "_index" : "${config.indexName}", "_type" : "${config.typeName.getOrElse("_doc")}" } }
         |${singleLine(serialize(doc))}""".stripMargin
    }
  }

  case class Create(config: ESConfig, doc: AnyRef, id: String) extends BulkAction {
    def jsonString: String = {
      s"""{ "create" : { "_index" : "${config.indexName}", "_type" : "${config.typeName.getOrElse("_doc")}", "_id": "${id}"} }
         |${singleLine(serialize(doc))}""".stripMargin
    }
  }

  case class Update(config: ESConfig, doc: AnyRef, id: String) extends BulkAction {
    def jsonString: String = {
      s"""{ "update" : { "_index" : "${config.indexName}", "_type" : "${config.typeName.getOrElse("_doc")}", "_id": "${id}"} }
         |{ "doc": ${singleLine(serialize(doc))} }""".stripMargin
    }
  }

  case class Script(config: ESConfig, script: String, id: String) extends BulkAction {
    def jsonString: String = {
      s"""{ "update" : { "_index" : "${config.indexName}", "_type" : "${config.typeName.getOrElse("_doc")}", "_id": "${id}"} }
         |{ "script": ${script} }""".stripMargin
    }
  }

  case class Delete(config: ESConfig, id: String) extends BulkAction {
    def jsonString: String = {
      s"""{ "delete" : { "_index" : "${config.indexName}", "_type" : "${config.typeName.getOrElse("_doc")}", "_id": "${id}"} }"""
    }
  }

  private def singleLine(str: String): String = str.replace("\n", "").replace("\r", "")

}