com.sksamuel.elastic4s.analysis.Analysis.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elastic4s-domain_2.12 Show documentation

elastic4s-domain

The newest version!

package com.sksamuel.elastic4s.analysis

import com.sksamuel.elastic4s.json.{XContentBuilder, XContentFactory}

case class Analysis(analyzers: List[Analyzer],
                    tokenizers: List[Tokenizer] = Nil,
                    tokenFilters: List[TokenFilter] = Nil,
                    charFilters: List[CharFilter] = Nil,
                    normalizers: List[Normalizer] = Nil)

object Analysis {
  def apply(analyzer: Analyzer): Analysis = Analysis(analyzers = List(analyzer))
}

object AnalysisBuilder extends Builder[Analysis] {
  override def build(a: Analysis): XContentBuilder = {

    val b = XContentFactory.jsonBuilder()

    if (a.analyzers.nonEmpty) {
      b.startObject("analyzer")
      a.analyzers.foreach { analyzer => b.rawField(analyzer.name, analyzer.build) }
      b.endObject()
    }

    if (a.normalizers.nonEmpty) {
      b.startObject("normalizer")
      a.normalizers.foreach { normalizer => b.rawField(normalizer.name, normalizer.build) }
      b.endObject()
    }

    if (a.tokenizers.nonEmpty) {
      b.startObject("tokenizer")
      a.tokenizers.foreach { tokenizer => b.rawField(tokenizer.name, tokenizer.build) }
      b.endObject()
    }

    if (a.charFilters.nonEmpty) {
      b.startObject("char_filter")
      a.charFilters.foreach { filter => b.rawField(filter.name, filter.build) }
      b.endObject()
    }

    if (a.tokenFilters.nonEmpty) {
      b.startObject("filter")
      a.tokenFilters.foreach { filter => b.rawField(filter.name, filter.build) }
      b.endObject()
    }

    b.endObject()
  }
}