33 lines
923 B
Scala
33 lines
923 B
Scala
package lila.db
|
|
|
|
import scala.collection.Factory
|
|
|
|
import reactivemongo.api._
|
|
import reactivemongo.api.bson._
|
|
|
|
trait CursorExt { self: dsl =>
|
|
|
|
// Can be refactor as CursorProducer
|
|
implicit final class ExtendCursor[A: BSONDocumentReader](val c: Cursor[A])(implicit
|
|
ec: scala.concurrent.ExecutionContext
|
|
) {
|
|
|
|
// like collect, but with stopOnError defaulting to false
|
|
def gather[M[_]](upTo: Int = Int.MaxValue)(implicit cbf: Factory[A, M[A]]): Fu[M[A]] =
|
|
c.collect[M](upTo, Cursor.ContOnError[M[A]]())
|
|
|
|
private def list(limit: Option[Int]): Fu[List[A]] = gather[List](limit | Int.MaxValue)
|
|
|
|
def list(limit: Int): Fu[List[A]] = list(limit.some)
|
|
|
|
def list(): Fu[List[A]] = list(none)
|
|
|
|
// like headOption, but with stopOnError defaulting to false
|
|
def uno: Fu[Option[A]] =
|
|
c.collect[Iterable](
|
|
1,
|
|
Cursor.ContOnError[Iterable[A]]()
|
|
).map(_.headOption)
|
|
}
|
|
}
|