Made internalRun visible on MongoSession for accessing Java driver fe…

…atures
uberto · Aug 22, 2024 · 55e1fe1 · 55e1fe1
1 parent d596f7c
commit 55e1fe1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoDbSession.kt b/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoDbSession.kt
@@ -25,7 +25,7 @@ class MongoDbSession(
             }
         }
 
-    fun <T> MongoTable<*>.internalRun(block: (MongoCollection<BsonDocument>) -> T): T =
+    override fun <T> MongoTable<*>.internalRun(block: (MongoCollection<BsonDocument>) -> T): T =
         block(
             withCollection(this)
         )

diff --git a/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoOperation.kt b/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoOperation.kt
@@ -7,8 +7,6 @@ typealias MongoOutcome<T> = Outcome<MongoError, T>
 
 
 
-
-
 fun <U, T> mongoCalculation(calculation: MongoSession.(U) -> T): (U) -> MongoOperation<T> = //unit
     { input: U -> MongoOperation { session -> calculation(session, input) } }
 

diff --git a/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoSession.kt b/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoSession.kt
@@ -57,6 +57,10 @@ interface MongoSession {
     fun <T : Any> MongoTable<T>.drop()
     fun <T : Any> MongoTable<T>.listIndexes(): Sequence<BsonDocument>
 
+
+    //Useful for getting access to all drivers methods
+    fun <T> MongoTable<*>.internalRun(block: (MongoCollection<BsonDocument>) -> T): T
+
     companion object {
         val RETURN_UPDATED = FindOneAndUpdateOptions().returnDocument(ReturnDocument.AFTER)
         val RETURN_BEFORE = FindOneAndUpdateOptions().returnDocument(ReturnDocument.BEFORE)