Implement offline search for facet values in OfflineIndex

Author: Clément Le Provost

algoliasearch/src/offline/java/com/algolia/search/saas/OfflineIndex.java

@@ -390,6 +390,28 @@ private JSONObject browseFromSync(@NonNull String cursor) throws AlgoliaExceptio
         return OfflineClient.parseSearchResults(localIndex.browse(query.build()));
     }
 
+    /**
+     * Search for facet values (asynchronously).
+     * Same parameters as {@link Index#searchForFacetValues(String, String, Query, CompletionHandler)}.
+     */
+    public Request searchForFacetValuesAsync(final @NonNull String facetName, final @NonNull String facetQuery, @Nullable Query query, @NonNull CompletionHandler completionHandler) {
+        final Query queryCopy = query != null ? new Query(query) : null;
+        return getClient().new AsyncTaskRequest(completionHandler) {
+            @NonNull
+            @Override
+            protected JSONObject run() throws AlgoliaException {
+                return searchForFacetValuesSync(facetName, facetQuery, queryCopy);
+            }
+        }.start();
+    }
+
+    /**
+     * Search for facet values (synchronously).
+     */
+    private JSONObject searchForFacetValuesSync(@NonNull String facetName, @NonNull String facetQuery, @Nullable Query query) throws AlgoliaException {
+        return OfflineClient.parseSearchResults(localIndex.searchForFacetValues(facetName, facetQuery, query != null ? query.build() : null));
+    }
+
     // ----------------------------------------------------------------------
     // Transaction management
     // ----------------------------------------------------------------------

algoliasearch/src/testOffline/java/com/algolia/search/saas/OfflineIndexTest.java

@@ -563,4 +563,40 @@ public void doRequestCompleted(JSONObject content, AlgoliaException error) {
             }
         });
     }
+
+    @Test
+    public void testSearchForFacetValues() throws Exception {
+        final CountDownLatch signal = new CountDownLatch(1);
+        final OfflineIndex index = client.getOfflineIndex(Helpers.getMethodName());
+        final OfflineIndex.WriteTransaction transaction = index.newTransaction();
+        transaction.setSettingsAsync(settings, new AssertCompletionHandler() {
+            @Override
+            public void doRequestCompleted(JSONObject content, AlgoliaException error) {
+                transaction.saveObjectsAsync(new JSONArray(objects.values()), new AssertCompletionHandler() {
+                    @Override
+                    public void doRequestCompleted(JSONObject content, AlgoliaException error) {
+                        assertNull(error);
+                        transaction.commitAsync(new AssertCompletionHandler() {
+                            @Override
+                            public void doRequestCompleted(JSONObject content, AlgoliaException error) {
+                                final Query query = new Query("snoopy");
+                                index.searchForFacetValuesAsync("series", "pea", new Query().setQuery("snoopy"), new AssertCompletionHandler() {
+                                    @Override
+                                    public void doRequestCompleted(JSONObject content, AlgoliaException error) {
+                                        assertNotNull(content);
+                                        JSONArray facetHits = content.optJSONArray("facetHits");
+                                        assertNotNull(facetHits);
+                                        assertEquals(1, facetHits.length());
+                                        assertEquals("Peanuts", facetHits.optJSONObject(0).optString("value"));
+                                        assertEquals(1, facetHits.optJSONObject(0).optInt("count"));
+                                        signal.countDown();
+                                    }
+                                });
+                            }
+                        });
+                    }
+                });
+            }
+        });
+    }
 }