simplify query when counting own studies

lichess-org · Feb 18, 2025 · 5ecebd8 · 5ecebd8
1 parent 36c128c
commit 5ecebd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/study/src/main/StudyRepo.scala b/modules/study/src/main/StudyRepo.scala
@@ -97,8 +97,8 @@ final class StudyRepo(private[study] val coll: AsyncColl)(using
   private def hasMemberOrIsPublic(using as: Option[MyId]) = as.fold(selectPublic): me =>
     $or($doc(s"members.$me".$exists(true)), selectPublic)
 
-  def countByOwner(ownerId: UserId)(using Option[MyId]) = coll:
-    _.countSel(selectOwnerId(ownerId) ++ hasMemberOrIsPublic)
+  def countByOwner(ownerId: UserId)(using as: Option[MyId]) = coll:
+    _.countSel(selectOwnerId(ownerId) ++ as.forall(_.isnt(ownerId)).so(hasMemberOrIsPublic))
 
   def sourceByOwner(ownerId: UserId, isMe: Boolean): Source[Study, ?] =
     Source.futureSource: