Add missing TextQueryBuilder and FuzzyQueryBuilder properties

radu-gheorghe · Apr 24, 2012 · 22b2aa6 · 22b2aa6
1 parent 783649a
commit 22b2aa6
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/main/java/org/elasticsearch/index/query/FuzzyQueryBuilder.java b/src/main/java/org/elasticsearch/index/query/FuzzyQueryBuilder.java
@@ -40,6 +40,8 @@ public class FuzzyQueryBuilder extends BaseQueryBuilder {
 
     private Integer prefixLength;
 
+    private Integer maxExpansions;
+
     /**
      * Constructs a new term query.
      *
@@ -75,6 +77,11 @@ public FuzzyQueryBuilder prefixLength(int prefixLength) {
         return this;
     }
 
+    public FuzzyQueryBuilder maxExpansions(int maxExpansions) {
+        this.maxExpansions = maxExpansions;
+        return this;
+    }
+
     @Override
     public void doXContent(XContentBuilder builder, Params params) throws IOException {
         builder.startObject(FuzzyQueryParser.NAME);
@@ -92,6 +99,9 @@ public void doXContent(XContentBuilder builder, Params params) throws IOExceptio
             if (prefixLength != null) {
                 builder.field("prefix_length", prefixLength);
             }
+            if (maxExpansions != null) {
+                builder.field("max_expansions", maxExpansions);
+            }
             builder.endObject();
         }
         builder.endObject();

diff --git a/src/main/java/org/elasticsearch/index/query/TextQueryBuilder.java b/src/main/java/org/elasticsearch/index/query/TextQueryBuilder.java
@@ -126,6 +126,11 @@ public TextQueryBuilder fuzziness(Object fuzziness) {
         return this;
     }
 
+    public TextQueryBuilder prefixLength(int prefixLength) {
+        this.prefixLength = prefixLength;
+        return this;
+    }
+
     /**
      * When using fuzzy or prefix type query, the number of term expansions to use. Defaults to unbounded
      * so its recommended to set it to a reasonable value for faster execution.