Fixing termFrequencyStatistics helper.

afoucret · Apr 30, 2024 · 7771036 · 7771036
1 parent 002d21d
commit 7771036
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/server/src/main/java/org/elasticsearch/script/TermsStatsReaderProvider.java b/server/src/main/java/org/elasticsearch/script/TermsStatsReaderProvider.java
@@ -16,6 +16,7 @@
 import org.apache.lucene.index.PostingsEnum;
 import org.apache.lucene.index.ReaderUtil;
 import org.apache.lucene.index.Term;
+import org.apache.lucene.index.TermState;
 import org.apache.lucene.index.TermStates;
 import org.apache.lucene.index.TermsEnum;
 import org.elasticsearch.search.lookup.SearchLookup;
@@ -65,8 +66,14 @@ private TermsStatsReader buildTermStatsReader(String fieldName, String query) th
                 TermStates termContext = searchLookup.getTermStates(term);
                 termStates.put(term, termContext);
 
+                TermState state = termContext.get(leafReaderContext);
+
+                if (state == null || termContext.docFreq() == 0) {
+                    continue;
+                }
+
                 TermsEnum termsEnum = leafReaderContext.reader().terms(term.field()).iterator();
-                termsEnum.seekExact(term.bytes(), termContext.get(leafReaderContext));
+                termsEnum.seekExact(term.bytes(), state);
                 postingsEnums.put(term, termsEnum.postings(null, PostingsEnum.ALL));
             }
         }