add number and boolean support to term query

fixes #129
jillesvangurp · May 11, 2024 · 58d4d53 · 58d4d53
1 parent 9449768
commit 58d4d53
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 2 deletions.
diff --git a/docs/src/test/kotlin/documentation/manual/search/term-level-queries.kt b/docs/src/test/kotlin/documentation/manual/search/term-level-queries.kt
@@ -20,6 +20,16 @@ val termLevelQueriesMd = sourceGitRepository.md {
             }.pretty("Term Query.").let { println(it) }
         }.printStdOut()
 
+        +"""
+            You can also do terms queries using numbers or booleans.
+        """.trimIndent()
+        example {
+            client.search(indexName) {
+                query = term(TestDoc::price, 0.80)
+            }.pretty("Term Query.").let { println(it) }
+        }
+
+
     }
     section("Terms query") {
         example {

diff --git a/...ch-dsls/src/commonMain/kotlin/com/jillesvangurp/searchdsls/querydsl/term-level-queries.kt b/...ch-dsls/src/commonMain/kotlin/com/jillesvangurp/searchdsls/querydsl/term-level-queries.kt
@@ -171,14 +171,14 @@ fun QueryClauses.regExp(
 
 // BEGIN term-query
 class TermQueryConfig : JsonDsl() {
-    var value by property<String>()
+    var value by property<Any>()
     var boost by property<Double>()
 }
 
 @SearchDSLMarker
 class TermQuery(
     field: String,
-    value: String,
+    value: Any,
     termQueryConfig: TermQueryConfig = TermQueryConfig(),
     block: (TermQueryConfig.() -> Unit)? = null
 ) : ESQuery("term") {
@@ -203,6 +203,35 @@ fun QueryClauses.term(
     block: (TermQueryConfig.() -> Unit)? = null
 ) =
     TermQuery(field,value, block = block)
+
+fun QueryClauses.term(
+    field: KProperty<*>,
+    value: Number,
+    block: (TermQueryConfig.() -> Unit)? = null
+) =
+    TermQuery(field.name,value, block = block)
+
+fun QueryClauses.term(
+    field: String,
+    value: Number,
+    block: (TermQueryConfig.() -> Unit)? = null
+) =
+    TermQuery(field,value, block = block)
+
+fun QueryClauses.term(
+    field: KProperty<*>,
+    value: Boolean,
+    block: (TermQueryConfig.() -> Unit)? = null
+) =
+    TermQuery(field.name,value, block = block)
+
+fun QueryClauses.term(
+    field: String,
+    value: Boolean,
+    block: (TermQueryConfig.() -> Unit)? = null
+) =
+    TermQuery(field,value, block = block)
+
 // END term-query
 
 @SearchDSLMarker