Improved Mongo typed filters

Signed-off-by: Uberto Barbini <[email protected]>
uberto · Aug 25, 2023 · 2290ba7 · 2290ba7
1 parent 8cec7be
commit 2290ba7
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 5 deletions.
diff --git a/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoFilters.kt b/kondor-mongo/src/main/kotlin/com/ubertob/kondor/mongo/core/MongoFilters.kt
@@ -4,19 +4,25 @@ import com.mongodb.client.model.Filters
 import com.ubertob.kondor.json.JsonProperty
 import org.bson.conversions.Bson
 
-infix fun <T> JsonProperty<T>.eq(value: T): Bson =
+infix fun <T : CharSequence> JsonProperty<T>.eq(value: T): Bson =
     Filters.eq(propName, value)
 
-infix fun <T : Any> JsonProperty<T>.lt(value: T): Bson =
+infix fun <T : Number> JsonProperty<T>.eq(value: T): Bson =
+    Filters.eq(propName, value)
+
+infix fun JsonProperty<Boolean>.eq(value: Boolean): Bson =
+    Filters.eq(propName, value)
+
+infix fun <T : Comparable<T>> JsonProperty<T>.lt(value: T): Bson =
     Filters.lt(propName, value)
 
-infix fun <T : Any> JsonProperty<T>.lte(value: T): Bson =
+infix fun <T : Comparable<T>> JsonProperty<T>.lte(value: T): Bson =
     Filters.lte(propName, value)
 
-infix fun <T : Any> JsonProperty<T>.gt(value: T): Bson =
+infix fun <T : Comparable<T>> JsonProperty<T>.gt(value: T): Bson =
     Filters.gt(propName, value)
 
-infix fun <T : Any> JsonProperty<T>.gte(value: T): Bson =
+infix fun <T : Comparable<T>> JsonProperty<T>.gte(value: T): Bson =
     Filters.gte(propName, value)
 
 

diff --git a/kondor-mongo/src/test/kotlin/com/ubertob/kondor/mongo/core/MongoTableTest.kt b/kondor-mongo/src/test/kotlin/com/ubertob/kondor/mongo/core/MongoTableTest.kt
@@ -242,5 +242,22 @@ class MongoTableTest {
 
     }
 
+    @Test
+    fun `use typed filters`() {
+
+        val query = mongoOperation {
+            val a = complexDocTable.find(JSmallClass.double eq 12.0).firstOrNull()
+            val b = complexDocTable.find(JSmallClass.string eq "SmallClass3").firstOrNull()
+            val c = complexDocTable.find(JSmallClass.int eq 9).firstOrNull()
+            val d = complexDocTable.find(JSmallClass.boolean eq true).firstOrNull()
+
+            "a=${a}, b=${b}, c=${c}, d=${d}"
+        }
+        val res = write100Doc + query exec localMongo
+
+        expectThat(res.expectSuccess()).isEqualTo("a=SmallClass(string=SmallClass12, int=12, double=12.0, boolean=true), b=SmallClass(string=SmallClass3, int=3, double=3.0, boolean=false), c=SmallClass(string=SmallClass9, int=9, double=9.0, boolean=false), d=SmallClass(string=SmallClass6, int=6, double=6.0, boolean=true)")
+    }
+
+
 }