1
0
Эх сурвалжийг харах

Override bulk visit methods of exitable point visitor (#82120)

zhouhui 3 жил өмнө
parent
commit
8f08c7b55b

+ 5 - 0
docs/changelog/82120.yaml

@@ -0,0 +1,5 @@
+pr: 82120
+summary: Override bulk visit methods of exitable point visitor
+area: Search
+type: bug
+issues: []

+ 13 - 0
server/src/main/java/org/elasticsearch/search/internal/ExitableDirectoryReader.java

@@ -18,6 +18,7 @@ import org.apache.lucene.index.PointValues;
 import org.apache.lucene.index.QueryTimeout;
 import org.apache.lucene.index.Terms;
 import org.apache.lucene.index.TermsEnum;
+import org.apache.lucene.search.DocIdSetIterator;
 import org.apache.lucene.search.suggest.document.CompletionTerms;
 import org.apache.lucene.util.BytesRef;
 import org.apache.lucene.util.automaton.CompiledAutomaton;
@@ -346,12 +347,24 @@ class ExitableDirectoryReader extends FilterDirectoryReader {
             in.visit(docID);
         }
 
+        @Override
+        public void visit(DocIdSetIterator iterator) throws IOException {
+            checkAndThrowWithSampling();
+            in.visit(iterator);
+        }
+
         @Override
         public void visit(int docID, byte[] packedValue) throws IOException {
             checkAndThrowWithSampling();
             in.visit(docID, packedValue);
         }
 
+        @Override
+        public void visit(DocIdSetIterator iterator, byte[] packedValue) throws IOException {
+            checkAndThrowWithSampling();
+            in.visit(iterator, packedValue);
+        }
+
         @Override
         public PointValues.Relation compare(byte[] minPackedValue, byte[] maxPackedValue) {
             queryCancellation.checkCancelled();