Add typoTolerance + allowTyposOnNumericTokens query parameters handling

algolia · Jun 9, 2014 · 3414a07 · 3414a07
1 parent 2754c3a
commit 3414a07
Showing 1 changed file with 30 additions and 2 deletions.
diff --git a/src/main/java/com/algolia/search/saas/Query.java b/src/main/java/com/algolia/search/saas/Query.java
@@ -63,6 +63,8 @@ public enum QueryType
     protected boolean analytics;
     protected boolean synonyms;
     protected boolean replaceSynonyms;
+    protected boolean typoTolerance;
+    protected boolean allowTyposOnNumericTokens;
 
     public Query(String query) {
         minWordSizeForApprox1 = 3;
@@ -75,7 +77,7 @@ public Query(String query) {
         queryType = QueryType.PREFIX_LAST;
         maxNumberOfFacets = -1;
         advancedSyntax = false;
-        analytics = synonyms = replaceSynonyms = true;
+        analytics = synonyms = replaceSynonyms = typoTolerance = allowTyposOnNumericTokens = true;
     }
 
     public Query() {
@@ -88,7 +90,7 @@ public Query() {
         queryType = QueryType.PREFIX_ALL;
         maxNumberOfFacets = -1;
         advancedSyntax = false;
-        analytics = synonyms = replaceSynonyms = true;
+        analytics = synonyms = replaceSynonyms = typoTolerance = allowTyposOnNumericTokens = true;
     }
 
     /**
@@ -173,6 +175,14 @@ public Query enableReplaceSynonymsInHighlight(boolean enabled) {
     	return this;
     }
 
+    /**
+     * @param If set to false, disable typo-tolerance. Default to true.
+     */
+    public Query enableTypoTolerance(boolean enabled) {
+        this.typoTolerance = enabled;
+        return this;
+    }
+
     /**
      * Specify the minimum number of characters in a query word to accept one typo in this word. 
      * Defaults to 3.
@@ -191,6 +201,14 @@ public Query setMinWordSizeToAllowTwoTypos(int nbChars) {
         return this;
     }
 
+    /**
+     * @param If set to false, disable typo-tolerance on numeric tokens. Default to true.
+     */
+    public Query enableTyposOnNumericTokens(boolean enabled) {
+        this.allowTyposOnNumericTokens = enabled;
+        return this;
+    }
+
     /**
      * if set, the result hits will contain ranking information in _rankingInfo attribute.
      */
@@ -405,6 +423,16 @@ protected String getQueryString() {
                     first = false;
                 }
             }
+            if (!typoTolerance) {
+                if (stringBuilder.length() > 0)
+                    stringBuilder.append('&');
+                stringBuilder.append("typoTolerance=false");
+            }
+            if (!allowTyposOnNumericTokens) {
+                if (stringBuilder.length() > 0)
+                    stringBuilder.append('&');
+                stringBuilder.append("allowTyposOnNumericTokens=false");
+            }
             if (minWordSizeForApprox1 != 3) {
                 if (stringBuilder.length() > 0)
                     stringBuilder.append('&');