From 615e6393542ef602b17f2b02a6c8ae117e97b66f Mon Sep 17 00:00:00 2001
From: Guillaume Naimi <guillaume.naimi@saagie.com>
Date: Wed, 12 Feb 2020 16:00:16 +0100
Subject: [PATCH] :sparkles: Add method to iterate on collections

---
 .../io/saagie/updatarium/dsl/action/MongoScriptAction.kt    | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/engine-mongodb/src/main/kotlin/io/saagie/updatarium/dsl/action/MongoScriptAction.kt b/engine-mongodb/src/main/kotlin/io/saagie/updatarium/dsl/action/MongoScriptAction.kt
index ed92af7..b2a26b3 100644
--- a/engine-mongodb/src/main/kotlin/io/saagie/updatarium/dsl/action/MongoScriptAction.kt
+++ b/engine-mongodb/src/main/kotlin/io/saagie/updatarium/dsl/action/MongoScriptAction.kt
@@ -37,4 +37,10 @@ class MongoScriptAction(
 
     fun onCollection(databaseName: String, collectionName: String) =
         mongoClient.getDatabase(databaseName).getCollection(collectionName)
+
+
+    fun onCollections(databaseNameRegex: Regex, collectionName: String) =
+        mongoClient.listDatabaseNames().filter { it.matches(databaseNameRegex) }.map {
+            mongoClient.getDatabase(it).getCollection(collectionName)
+        }
 }