com.sumologic.elasticsearch.restlastic.dsl.CompletionDsl.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch-core Show documentation
The newest version!
/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package com.sumologic.elasticsearch.restlastic.dsl

trait CompletionDsl extends DslCommons with QueryDsl {

  case class SuggestRoot(textOpt: Option[String], suggestions: List[Suggestion]) extends Query with RootObject {
    val _suggest = "suggest"
    val _text = "text"

    override def toJson(version: EsVersion): Map[String, Any] = {
      val globalTextJson: Map[String, Any] = Map.empty[String, Any] ++ textOpt.map(text => _text -> text)
      val suggestionsMap: Map[String, Any] = suggestions.map(p => p.name -> p.toJson(version)).toMap
      Map(_suggest -> (globalTextJson ++ suggestionsMap))
    }
  }

  case class Suggestion(name: String, textOpt: Option[String], termOpt: Option[SuggestionTerm], completionOpt: Option[Completion]) extends Query {
    val _text = "text"
    val _completion = "completion"

    override def toJson(version: EsVersion): Map[String, Any] = {
      val textJson = Map.empty[String, Any] ++ textOpt.map(text => _text -> text)

      // TODO: Merge both the termOpt and completionOpt
      require(!(termOpt.isDefined && completionOpt.isDefined), "Both the completionOpt and termOpt can not be defined.")

      val termJson = Map.empty[String, Any] ++ termOpt.map(term => _completion -> term.toJson(version))
      val completionJson =
        Map.empty[String, Any] ++ completionOpt.map(completion => _completion -> completion.toJson(version))
      termJson ++ textJson ++ completionJson
    }
  }

  case class SuggestionTerm(fieldName: String) extends Query {
    val _field = "field"

    override def toJson(version: EsVersion): Map[String, Any] = {
      Map(_field -> fieldName)
    }
  }

  case class Completion(field: String, size: Int, name: String, contexts: List[Context]) extends Query {
    val _field = "field"
    val _contexts = "contexts"
    val _context = "context"
    val _size = "size"

    override def toJson(version: EsVersion): Map[String, Any] = {
      val contextJson = version match {
        case V6 => Map(_contexts -> Map(name -> contexts.map(_.toJson(version))))
        case V2 => Map(_context -> Map(name -> contexts.flatMap(_.values)))
      }
      Map(
        _field -> field,
        _size -> size
      ) ++ contextJson
    }
  }

  case class Context(values: List[String]) extends Query {
    val _context = "context"

    override def toJson(version: EsVersion): Map[String, Any] = {
      version match {
        case V6 => values.flatMap(value => Map(_context -> value)).toMap
        case V2 => throw new UnsupportedOperationException
      }
    }
  }

}