Added getAll convenience method

outr · Aug 10, 2024 · 06f0718 · 06f0718
1 parent e83a99d
commit 06f0718
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/async/src/main/scala/lightdb/async/AsyncCollection.scala b/async/src/main/scala/lightdb/async/AsyncCollection.scala
@@ -36,6 +36,9 @@ case class AsyncCollection[Doc <: Document[Doc], Model <: DocumentModel[Doc]](un
   def get(id: Id[Doc])(implicit transaction: Transaction[Doc]): IO[Option[Doc]] =
     IO.blocking(underlying.get(id))
 
+  def getAll(ids: Seq[Id[Doc]])(implicit transaction: Transaction[Doc]): fs2.Stream[IO, Doc] =
+    fs2.Stream.fromBlockingIterator[IO](underlying.getAll(ids), 512)
+
   def apply(id: Id[Doc])(implicit transaction: Transaction[Doc]): IO[Doc] =
     IO.blocking(underlying(id))
 

diff --git a/core/src/main/scala/lightdb/collection/Collection.scala b/core/src/main/scala/lightdb/collection/Collection.scala
@@ -147,6 +147,10 @@ case class Collection[Doc <: Document[Doc], Model <: DocumentModel[Doc]](name: S
     store.get(model._id, id)
   }
 
+  def getAll(ids: Seq[Id[Doc]])(implicit transaction: Transaction[Doc]): Iterator[Doc] = ids
+    .iterator
+    .flatMap(get)
+
   def apply(id: Id[Doc])(implicit transaction: Transaction[Doc]): Doc =
     store.get(model._id, id).getOrElse {
       throw DocNotFoundException(name, "_id", id)