34 lines
1.1 KiB
Scala
34 lines
1.1 KiB
Scala
package lila.db
|
|
package api
|
|
|
|
import Implicits._
|
|
import play.api.libs.json._
|
|
import reactivemongo.bson._
|
|
|
|
object $projection {
|
|
import play.modules.reactivemongo.json._
|
|
|
|
def apply[A: InColl, B](
|
|
q: JsObject,
|
|
fields: Seq[String],
|
|
modifier: QueryBuilder => QueryBuilder = identity,
|
|
max: Option[Int] = None)(extract: JsObject => Option[B]): Fu[List[B]] =
|
|
modifier {
|
|
implicitly[InColl[A]].coll.genericQueryBuilder query q projection projector(fields)
|
|
} toList[BSONDocument] max map (list => list map { obj =>
|
|
extract(JsObjectReader read obj)
|
|
} flatten)
|
|
|
|
def one[A: InColl, B](
|
|
q: JsObject,
|
|
fields: Seq[String],
|
|
modifier: QueryBuilder => QueryBuilder = identity)(extract: JsObject => Option[B]): Fu[Option[B]] =
|
|
modifier(implicitly[InColl[A]].coll.genericQueryBuilder query q projection projector(fields)).one[BSONDocument] map (opt => opt map { obj =>
|
|
extract(JsObjectReader read obj)
|
|
} flatten)
|
|
|
|
private def projector(fields: Seq[String]): JsObject = Json obj {
|
|
(fields map (_ -> Json.toJsFieldJsValueWrapper(1))): _*
|
|
}
|
|
}
|