reuse DocValues iterators in SolrDocumentFetcher/udvas (#144)

(cherry picked from commit 9bb42b8)
cowpaths · Sep 15, 2023 · b4b637b · b4b637b
1 parent b7af3d0
commit b4b637b
Show file tree

Hide file tree

Showing 2 changed files with 397 additions and 36 deletions.
diff --git a/solr/core/src/java/org/apache/solr/handler/component/RealTimeGetComponent.java b/solr/core/src/java/org/apache/solr/handler/component/RealTimeGetComponent.java
@@ -239,6 +239,7 @@ public void process(ResponseBuilder rb) throws IOException {
 
       boolean opennedRealtimeSearcher = false;
       BytesRefBuilder idBytes = new BytesRefBuilder();
+      Map<String, SolrDocumentFetcher.DVIterEntry> reuseDvIters = new HashMap<>();
       for (String idStr : reqIds.allIds) {
         fieldType.readableToIndexed(idStr, idBytes);
         // if _route_ is passed, id is a child doc.  TODO remove in SOLR-15064
@@ -349,7 +350,8 @@ public void process(ResponseBuilder rb) throws IOException {
             searcherInfo.getSearcher().doc(docid, rsp.getReturnFields().getLuceneFieldNames());
         SolrDocument doc = toSolrDoc(luceneDocument, core.getLatestSchema());
         SolrDocumentFetcher docFetcher = searcherInfo.getSearcher().getDocFetcher();
-        docFetcher.decorateDocValueFields(doc, docid, docFetcher.getNonStoredDVs(true));
+        docFetcher.decorateDocValueFields(
+            doc, docid, docFetcher.getNonStoredDVs(true), reuseDvIters);
         if (null != transformer) {
           if (null == resultContext) {
             // either first pass, or we've re-opened searcher - either way now we setContext
@@ -576,7 +578,7 @@ private static SolrDocument mergePartialDocWithFullDocFromIndex(
       if (!doc.containsKey(VERSION_FIELD)) {
         searcher
             .getDocFetcher()
-            .decorateDocValueFields(doc, docid, Collections.singleton(VERSION_FIELD));
+            .decorateDocValueFields(doc, docid, Collections.singleton(VERSION_FIELD), null);
       }
 
       long docVersion = (long) doc.getFirstValue(VERSION_FIELD);