Address FT.SPELLCHECK changes

sazzad16 · Aug 6, 2023 · 3682860 · 3682860
1 parent 6292b06
commit 3682860
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/src/main/java/redis/clients/jedis/search/SearchBuilderFactory.java b/src/main/java/redis/clients/jedis/search/SearchBuilderFactory.java
@@ -138,25 +138,32 @@ public Map<String, List<String>> build(Object data) {
       = new Builder<Map<String, Map<String, Double>>>() {
 
     private static final String TERM = "TERM";
+    private static final String RESULTS = "results";
 
     @Override
     public Map<String, Map<String, Double>> build(Object data) {
-      List rawTerms = (List) data;
-      if (rawTerms.isEmpty()) return Collections.emptyMap();
+      List rawDataList = (List) data;
+      if (rawDataList.isEmpty()) return Collections.emptyMap();
 
-      if (rawTerms.get(0) instanceof KeyValue) {
-        return ((List<KeyValue>) rawTerms).stream().collect(Collectors.toMap(
+      if (rawDataList.get(0) instanceof KeyValue) {
+        KeyValue rawData = (KeyValue) rawDataList.get(0);
+        String header = STRING.build(rawData.getKey());
+        if (!RESULTS.equals(header)) {
+          throw new IllegalStateException("Unrecognized header: " + header);
+        }
+
+        return ((List<KeyValue>) rawData.getValue()).stream().collect(Collectors.toMap(
             rawTerm -> STRING.build(rawTerm.getKey()),
             rawTerm -> ((List<List<KeyValue>>) rawTerm.getValue()).stream()
                 .collect(Collectors.toMap(entry -> STRING.build(entry.get(0).getKey()),
                       entry -> BuilderFactory.DOUBLE.build(entry.get(0).getValue()))),
             (x, y) -> x, LinkedHashMap::new));
       }
 
-      Map<String, Map<String, Double>> returnTerms = new LinkedHashMap<>(rawTerms.size());
+      Map<String, Map<String, Double>> returnTerms = new LinkedHashMap<>(rawDataList.size());
 
-      for (Object rawTerm : rawTerms) {
-        List<Object> rawElements = (List<Object>) rawTerm;
+      for (Object rawData : rawDataList) {
+        List<Object> rawElements = (List<Object>) rawData;
 
         String header = STRING.build(rawElements.get(0));
         if (!TERM.equals(header)) {