quix.athena.AthenaAutocomplete.scala Maven / Gradle / Ivy

Go to download

package quix.athena

import monix.eval.Task
import quix.api.v1.db.{Autocomplete, Catalog, Catalogs}

class AthenaAutocomplete(val catalogs: Catalogs) extends Autocomplete {

  override def fast: Task[Map[String, List[String]]] = catalogs.fast.map(extractAutoCompleteItems)

  override def full: Task[Map[String, List[String]]] = catalogs.full.map(extractAutoCompleteItems)

  def extractAutoCompleteItems(catalogList: List[Catalog]): Map[String, List[String]] = {
    Map("schemas" -> catalogList.flatMap(_.children.map(_.name)),
      "tables" -> catalogList.flatMap(_.children.flatMap(_.children.map(_.name))))
  }
}