deepcrayon
/
lila
1
0
Fork 0
Code Releases Activity

skip DB and lift throttling when querying online titled players 

closes #4856
nvuiRefactor
Thibault Duplessis 2019-01-23 07:57:26 +08:00
parent cf3f27f259
commit ca5ef3a338
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/api/src/main/UserApi.scala
View File

@ -36,18 +36,21 @@ private[api] final class UserApi(
def exportTitled(config: UserApi.Titled): Enumerator[JsObject] =
if (config.titles.isEmpty) Enumerator.empty[JsObject]
else if (config.online) Enumerator.enumerate(recentTitledUserIds()) &>
Enumeratee.filter(isOnline) &>
Enumeratee.mapM(lightUserApi.async) &>
Enumeratee.collect {
case Some(lu) if lu.title.exists(ut => config.titles.exists(_.value == ut)) =>
addPlayingStreaming(LightUser.lightUserWrites.writes(lu), lu.id)
}
else UserRepo.idCursor(
selector = $doc(
"title" $in config.titles, "enabled" -> true
) ++ config.online.??($doc("_id" $in recentTitledUserIds())),
selector = $doc("title" $in config.titles, "enabled" -> true),
batchSize = config.perSecond.value
).bulkEnumerator() &>
lila.common.Iteratee.delay(1 second) &>
Enumeratee.mapM { docs =>
lightUserApi.asyncMany {
docs.flatMap { _.getAs[User.ID]("_id") } filter { id =>
!config.online || isOnline(id)
} toList
docs.flatMap { _.getAs[User.ID]("_id") } toList
}
} &>
Enumeratee.mapConcat { users =>