fix: Add support for numbers on TermsQuery (quickwit-oss#6097)

* fix: Add support for numbers on TermsQuery

Signed-off-by: Darkheir <raphael.cohen@sekoia.io>

Author: Darkheir

quickwit/quickwit-query/src/elastic_query_dsl/terms_query.rs

@@ -37,17 +37,36 @@ struct TermsQueryForSerialization {
     capture_other: serde_json::Value,
 }
 
+#[derive(Deserialize)]
+#[serde(untagged)]
+enum TermValue {
+    I64(i64),
+    U64(u64),
+    Str(String),
+}
+
+impl From<TermValue> for String {
+    fn from(term_value: TermValue) -> String {
+        match term_value {
+            TermValue::I64(val) => val.to_string(),
+            TermValue::U64(val) => val.to_string(),
+            TermValue::Str(val) => val,
+        }
+    }
+}
+
 #[derive(Deserialize)]
 #[serde(untagged)]
 enum OneOrMany {
-    One(String),
-    Many(Vec<String>),
+    One(TermValue),
+    Many(Vec<TermValue>),
 }
+
 impl From<OneOrMany> for Vec<String> {
     fn from(one_or_many: OneOrMany) -> Vec<String> {
         match one_or_many {
-            OneOrMany::One(one_value) => vec![one_value],
-            OneOrMany::Many(values) => values,
+            OneOrMany::One(one_value) => vec![String::from(one_value)],
+            OneOrMany::Many(values) => values.into_iter().map(String::from).collect(),
         }
     }
 }
@@ -109,6 +128,14 @@ mod tests {
         assert_eq!(&terms_query.values[..], &["hello".to_string()]);
     }
 
+    #[test]
+    fn test_terms_query_not_string() {
+        let terms_query_json = r#"{ "user.id": [1, 2] }"#;
+        let terms_query: TermsQuery = serde_json::from_str(terms_query_json).unwrap();
+        assert_eq!(&terms_query.field, "user.id");
+        assert_eq!(&terms_query.values[..], &["1".to_string(), "2".to_string()]);
+    }
+
     #[test]
     fn test_terms_query_single_term_boost() {
         let terms_query_json = r#"{ "user.id": ["hello", "happy"], "boost": 2 }"#;