Add client multi search method meilisearch#591

src/main/java/com/meilisearch/sdk/Client.java

@@ -20,6 +20,7 @@
 import com.meilisearch.sdk.model.TaskInfo;
 import com.meilisearch.sdk.model.TasksQuery;
 import com.meilisearch.sdk.model.TasksResults;
+import com.meilisearch.sdk.model.MultiSearchResult;
 import java.util.Date;
 import java.util.Map;
 import java.util.TimeZone;
@@ -379,6 +380,10 @@ public void deleteKey(String key) throws MeilisearchException {
         this.keysHandler.deleteKey(key);
     }
 
+    public Results<MultiSearchResult> multiSearch(MultiSearchRequest search) throws MeilisearchException {
+        return this.config.httpClient.post("/multi-search", search, Results.class, MultiSearchResult.class);
+    }
+
     public String generateTenantToken(String apiKeyUid, Map<String, Object> searchRules)
             throws MeilisearchException {
         return this.generateTenantToken(apiKeyUid, searchRules, new TenantTokenOptions());

src/main/java/com/meilisearch/sdk/HttpClient.java

@@ -90,10 +90,10 @@ <T> T get(String api, String param, Class<T> targetClass, Class<?>... parameters
      * @return results of the search
      * @throws MeilisearchException if the response is an error
      */
-    <S, T> T post(String api, S body, Class<T> targetClass) throws MeilisearchException {
+    <S, T> T post(String api, S body, Class<T> targetClass, Class<?>... parameters) throws MeilisearchException {
         HttpRequest requestConfig = request.create(HttpMethod.POST, api, this.headers, body);
         HttpResponse<T> httpRequest = this.client.post(requestConfig);
-        HttpResponse<T> httpResponse = response.create(httpRequest, targetClass);
+        HttpResponse<T> httpResponse = response.create(httpRequest, targetClass, parameters);
 
         if (httpResponse.getStatusCode() >= 400) {
             throw new MeilisearchApiException(

src/main/java/com/meilisearch/sdk/IndexSearchRequest.java

@@ -0,0 +1,94 @@
+package com.meilisearch.sdk;
+
+import com.meilisearch.sdk.model.MatchingStrategy;
+import lombok.*;
+import lombok.experimental.Accessors;
+import org.json.JSONObject;
+
+@Builder
+@AllArgsConstructor(access = AccessLevel.PACKAGE)
+@NoArgsConstructor(access = AccessLevel.PACKAGE)
+@Getter
+@Setter
+@Accessors(chain = true)
+public class IndexSearchRequest {
+    private String indexUid;
+    private String q;
+    private Integer offset;
+    private Integer limit;
+    private String[] attributesToRetrieve;
+    private String[] attributesToCrop;
+    private Integer cropLength;
+    private String cropMarker;
+    private String highlightPreTag;
+    private String highlightPostTag;
+    private MatchingStrategy matchingStrategy;
+    private String[] attributesToHighlight;
+    private String[] filter;
+    private String[][] filterArray;
+    private Boolean showMatchesPosition;
+    private String[] facets;
+    private String[] sort;
+    protected Integer page;
+    protected Integer hitsPerPage;
+
+    /**
+     * Constructor for MultiSearchRequest for building search queries with the default values: offset: 0,
+     * limit: 20, attributesToRetrieve: ["*"], attributesToCrop: null, cropLength: 200,
+     * attributesToHighlight: null, filter: null, showMatchesPosition: false, facets: null, sort:
+     * null
+     *
+     * @param indexUid uid of the requested index String
+     */
+    public IndexSearchRequest(String indexUid) {
+        this();
+        this.indexUid = indexUid;
+    }
+
+    /**
+     * Method to set the Query String
+     *
+     * <p>This method is an alias of setQ()
+     *
+     * @param q Query String
+     * @return SearchRequest
+     */
+    public IndexSearchRequest setQuery(String q) {
+        return setQ(q);
+    }
+
+    /**
+     * Method that returns the JSON String of the SearchRequest
+     *
+     * @return JSON String of the SearchRequest query
+     */
+    @Override
+    public String toString() {
+        JSONObject jsonObject =
+            new JSONObject()
+                .put("q", this.q)
+                .put("offset", this.offset)
+                .put("limit", this.limit)
+                .put("attributesToRetrieve", this.attributesToRetrieve)
+                .put("cropLength", this.cropLength)
+                .put("cropMarker", this.cropMarker)
+                .put("highlightPreTag", this.highlightPreTag)
+                .put("highlightPostTag", this.highlightPostTag)
+                .put(
+                    "matchingStrategy",
+                    this.matchingStrategy == null
+                        ? null
+                        : this.matchingStrategy.toString())
+                .put("showMatchesPosition", this.showMatchesPosition)
+                .put("facets", this.facets)
+                .put("sort", this.sort)
+                .put("page", this.page)
+                .put("hitsPerPage", this.hitsPerPage)
+                .putOpt("attributesToCrop", this.attributesToCrop)
+                .putOpt("attributesToHighlight", this.attributesToHighlight)
+                .putOpt("filter", this.filter)
+                .putOpt("filter", this.filterArray);
+
+        return jsonObject.toString();
+    }
+}

src/main/java/com/meilisearch/sdk/MultiSearchRequest.java

@@ -0,0 +1,25 @@
+package com.meilisearch.sdk;
+
+import lombok.*;
+
+import java.util.ArrayList;
+
+public class MultiSearchRequest {
+    private ArrayList<IndexSearchRequest> queries;
+
+    public MultiSearchRequest() {
+        this.queries = new ArrayList();
+    }
+
+    /**
+     * Method to add new Query
+     *
+     * @param search Query IndexSearchRequest
+     * @return MultiSearchRequest
+     */
+    public MultiSearchRequest addQuery(IndexSearchRequest search) {
+        this.queries.add(search);
+        return this;
+    }
+
+}

src/main/java/com/meilisearch/sdk/model/MultiSearchResult.java

@@ -0,0 +1,27 @@
+package com.meilisearch.sdk.model;
+
+import lombok.Getter;
+import lombok.ToString;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+
+/**
+ * Multi search response
+ *
+ * <p>https://www.meilisearch.com/docs/reference/api/multi_search#response
+ */
+@Getter
+@ToString
+public class MultiSearchResult implements Searchable {
+    String indexUid;
+    ArrayList<HashMap<String, Object>> hits;
+    Object facetDistribution;
+    int processingTimeMs;
+    String query;
+    int offset;
+    int limit;
+    int estimatedTotalHits;
+
+    public MultiSearchResult() {}
+}

src/test/java/com/meilisearch/integration/SearchTest.java

@@ -13,6 +13,8 @@
 import com.meilisearch.integration.classes.AbstractIT;
 import com.meilisearch.integration.classes.TestData;
 import com.meilisearch.sdk.Index;
+import com.meilisearch.sdk.IndexSearchRequest;
+import com.meilisearch.sdk.MultiSearchRequest;
 import com.meilisearch.sdk.SearchRequest;
 import com.meilisearch.sdk.json.GsonJsonHandler;
 import com.meilisearch.sdk.model.MatchingStrategy;
@@ -21,12 +23,15 @@
 import com.meilisearch.sdk.model.Searchable;
 import com.meilisearch.sdk.model.Settings;
 import com.meilisearch.sdk.model.TaskInfo;
+import com.meilisearch.sdk.model.MultiSearchResult;
 import com.meilisearch.sdk.utils.Movie;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
 
+import java.util.HashSet;
+
 @Tag("integration")
 public class SearchTest extends AbstractIT {
 
@@ -598,4 +603,45 @@ public void testPlaceHolderWithLimit() throws Exception {
 
         assertThat(searchResult.getHits(), hasSize(10));
     }
+
+    @Test
+    public void testMultiSearch() throws Exception {
+        HashSet<String> indexUids = new HashSet();
+        indexUids.add("MultiSearch1");
+        indexUids.add("MultiSearch2");
+
+
+        for(String indexUid : indexUids) {
+            Index index = client.index(indexUid);
+
+            TestData<Movie> testData = this.getTestData(MOVIES_INDEX, Movie.class);
+            TaskInfo task = index.addDocuments(testData.getRaw());
+
+            index.waitForTask(task.getTaskUid());
+        }
+
+
+        MultiSearchRequest search = new MultiSearchRequest();
+
+        for(String indexUid : indexUids) {
+            search.addQuery(
+                new IndexSearchRequest(indexUid)
+                    .setQuery("batman")
+            );
+        }
+
+        MultiSearchResult[] results = client.multiSearch(search).getResults();
+
+        assertThat(results.length, is(2));
+
+        for(MultiSearchResult searchResult : results) {
+            assertThat(indexUids.contains(searchResult.getIndexUid()), is(true));
+            assertThat(searchResult.getFacetDistribution(), is(nullValue()));
+            assertThat(searchResult.getHits(), hasSize(1));
+            assertThat(searchResult.getOffset(), is(equalTo(0)));
+            assertThat(searchResult.getLimit(), is(equalTo(20)));
+            assertThat(searchResult.getEstimatedTotalHits(), is(equalTo(1)));
+        }
+
+    }
 }