deepcrayon/lila
deepcrayon/lila
1
0
Fork 0
Code Releases Activity

use mongodb aggregation to fetch a user friends

Browse source
This commit is contained in:
Thibault Duplessis 2016-01-24 10:31:35 +07:00
parent 5024c64505
commit e98f3674dd
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
modules/relation/src/main/RelationApi.scala
View file

@ -12,6 +12,7 @@ import lila.user.{ User => UserModel, UserRepo }
import tube.relationTube
import BSONHandlers._
import reactivemongo.api.collections.bson.BSONBatchCommands.AggregationFramework._
import reactivemongo.bson._
final class RelationApi(
@ -39,8 +40,17 @@ final class RelationApi(
def fetchBlocking = RelationRepo blocking _
def fetchFriends(userId: ID) = fetchFollowing(userId) zip fetchFollowers(userId) map {
case (f1, f2) => f1 intersect f2
def fetchFriends(userId: ID) = coll.aggregate(Match(BSONDocument(
"$or" -> BSONArray(BSONDocument("u1" -> userId), BSONDocument("u2" -> userId))
)), List(
Group(BSONNull)(
"u1" -> AddToSet("u1"),
"u2" -> AddToSet("u2")),
Project(BSONDocument(
"_id" -> BSONDocument("$setIntersection" -> BSONArray("$u1", "$u2"))
))
)).map {
~_.documents.headOption.flatMap(_.getAs[Set[String]]("_id"))
}
def fetchFollows(u1: ID, u2: ID) =