Fixed AsyncLightDB to allow customizing collections for re-indexing a…

…s well
outr · Dec 21, 2024 · b241beb · b241beb
1 parent 2feef79
commit b241beb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/async/src/main/scala/lightdb/async/AsyncLightDB.scala b/async/src/main/scala/lightdb/async/AsyncLightDB.scala
@@ -71,7 +71,7 @@ trait AsyncLightDB extends FeatureSupport[DBFeatureKey] { db =>
                                                                     storeManager: Option[StoreManager] = None): AsyncCollection[Doc, Model] =
     AsyncCollection(underlying.collection[Doc, Model](model, name, storeManager))
 
-  def reIndex(): Task[Int] = rapid.Stream.emits(underlying.collections)
+  def reIndex(collections: List[Collection[_, _]] = underlying.collections): Task[Int] = rapid.Stream.emits(collections)
     .par(maxThreads = 32) { collection =>
       AsyncCollection[KeyValue, KeyValue.type](collection.asInstanceOf[Collection[KeyValue, KeyValue.type]]).reIndex()
     }
@@ -106,6 +106,9 @@ trait AsyncLightDB extends FeatureSupport[DBFeatureKey] { db =>
 
   protected def initialize(): Task[Unit] = Task.unit
 
+  def collectionsByNames(collectionNames: String*): Task[List[Collection[_, _]]] =
+    Task(underlying.collectionsByNames(collectionNames: _*))
+
   def dispose(): Task[Boolean] = Task {
     if (underlying.disposed) {
       false