Improvements

outr · Mar 31, 2024 · 6704029 · 6704029
1 parent 63d220a
commit 6704029
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/shared/src/main/scala/lightdb/collection/Collection.scala b/core/shared/src/main/scala/lightdb/collection/Collection.scala
@@ -20,6 +20,12 @@ case class Collection[D <: Document[D]](db: LightDB, mapping: ObjectMapping[D],
     data.get(id)
   }
 
+  def getAll(ids: fs2.Stream[IO, Id[D]]): fs2.Stream[IO, D] = ids.evalMap(get).collect {
+    case Some(d) => d
+  }
+
+  def getAll(ids: List[Id[D]]): IO[List[D]] = getAll(fs2.Stream(ids: _*)).compile.toList
+
   def fromArray(array: Array[Byte]): D = data.fromArray(array)
 
   def apply(id: Id[D]): IO[D] = data(id)