Handle IndexedDISI in exitableIterator

Author: ChrisHegarty

server/src/main/java/org/elasticsearch/search/internal/ExitableDirectoryReader.java

@@ -10,6 +10,7 @@
 package org.elasticsearch.search.internal;
 
 import org.apache.lucene.codecs.StoredFieldsReader;
+import org.apache.lucene.codecs.lucene90.IndexedDISI;
 import org.apache.lucene.index.ByteVectorValues;
 import org.apache.lucene.index.DirectoryReader;
 import org.apache.lucene.index.FilterDirectoryReader;
@@ -669,6 +670,8 @@ public FloatVectorValues copy() throws IOException {
     static DocIdSetIterator exitableIterator(DocIdSetIterator iterator, QueryCancellation queryCancellation) {
         if (iterator instanceof KnnVectorValues.DocIndexIterator docIndexIterator) {
             return createExitableIterator(docIndexIterator, queryCancellation);
+        } else if (iterator instanceof IndexedDISI indexedDISI) {
+            return createExitableIterator(IndexedDISI.asDocIndexIterator(indexedDISI), queryCancellation);
         } else {
             return new ExitableDocSetIterator(iterator, queryCancellation);
         }