Add autoGenerateSynonymsPhraseQuery to MatchQuery (#3058)

Philippus · May 5, 2024 · 0a5c0af · 0a5c0af
1 parent ab34744
commit 0a5c0af
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 1 deletion.
diff --git a/.../src/main/scala/com/sksamuel/elastic4s/requests/searches/queries/matches/MatchQuery.scala b/.../src/main/scala/com/sksamuel/elastic4s/requests/searches/queries/matches/MatchQuery.scala
@@ -19,12 +19,16 @@ case class MatchQuery(field: String,
                       operator: Option[Operator] = None,
                       prefixLength: Option[Int] = None,
                       queryName: Option[String] = None,
-                      zeroTerms: Option[String] = None)
+                      zeroTerms: Option[String] = None,
+                      autoGenerateSynonymsPhraseQuery: Option[Boolean] = None)
   extends Query {
 
   def analyzer(an: String): MatchQuery = copy(analyzer = an.some)
   def analyzer(an: Analyzer): MatchQuery = copy(analyzer = an.name.some)
 
+  def autoGenerateSynonymsPhraseQuery(autoGenerate: Boolean): MatchQuery =
+    copy(autoGenerateSynonymsPhraseQuery = autoGenerate.some)
+
   def boost(boost: Double): MatchQuery = copy(boost = boost.some)
   def cutoffFrequency(f: Double): MatchQuery = copy(cutoffFrequency = f.some)
   def lenient(lenient: Boolean): MatchQuery = copy(lenient = lenient.some)

diff --git a/...ain/scala/com/sksamuel/elastic4s/handlers/searches/queries/text/MatchQueryBuilderFn.scala b/...ain/scala/com/sksamuel/elastic4s/handlers/searches/queries/text/MatchQueryBuilderFn.scala
@@ -14,6 +14,7 @@ object MatchQueryBuilderFn {
 
     q.queryName.foreach(builder.field("_name", _))
     q.analyzer.foreach(builder.field("analyzer", _))
+    q.autoGenerateSynonymsPhraseQuery.foreach(builder.field("auto_generate_synonyms_phrase_query", _))
     q.boost.foreach(builder.field("boost", _))
     q.cutoffFrequency.map(_.toString).foreach(builder.field("cutoff_frequency", _))
     q.fuzziness.foreach(builder.field("fuzziness", _))

diff --git a/elastic4s-tests/src/test/resources/json/search/search_match.json b/elastic4s-tests/src/test/resources/json/search/search_match.json
@@ -4,6 +4,7 @@
             "name": {
                 "query": "coldplay",
                 "analyzer": "french",
+                "auto_generate_synonyms_phrase_query": false,
                 "cutoff_frequency": "3.4",
                 "fuzziness": "2",
                 "fuzzy_transpositions": true,

diff --git a/elastic4s-tests/src/test/scala/com/sksamuel/elastic4s/search/SearchDslTest.scala b/elastic4s-tests/src/test/scala/com/sksamuel/elastic4s/search/SearchDslTest.scala
@@ -172,6 +172,7 @@ class SearchDslTest extends AnyFlatSpec with MockitoSugar with JsonSugar with On
         .fuzziness("2")
         .prefixLength(4)
         .analyzer(FrenchLanguageAnalyzer)
+        .autoGenerateSynonymsPhraseQuery(false)
     }
     req.request.entity.get.get should matchJsonResource("/json/search/search_match.json")
   }