lightdb.async.AsyncSearchResults.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lightdb-async_3 Show documentation

lightdb-async

The newest version!

package lightdb.async

import cats.effect.IO
import lightdb.doc.{Document, DocumentModel}
import lightdb.facet.FacetResult
import lightdb.field.Field.FacetField
import lightdb.transaction.Transaction

case class AsyncSearchResults[Doc <: Document[Doc], Model <: DocumentModel[Doc], V](model: Model,
                                                                                    offset: Int,
                                                                                    limit: Option[Int],
                                                                                    total: Option[Int],
                                                                                    scoredStream: fs2.Stream[IO, (V, Double)],
                                                                                    facetResults: Map[FacetField[Doc], FacetResult],
                                                                                    transaction: Transaction[Doc]) {
  def stream: fs2.Stream[IO, V] = scoredStream.map(_._1)

  def first: IO[Option[V]] = stream.take(1).compile.toList.map(_.headOption)

  def one: IO[V] = first.map {
    case Some(v) => v
    case None => throw new NullPointerException("No results for search")
  }

  def facet(f: Model => FacetField[Doc]): FacetResult = facetResults(f(model))

  def getFacet(f: Model => FacetField[Doc]): Option[FacetResult] = facetResults.get(f(model))
}