Fields/highlighting would only apply to "size" number of docs

blevesearch · Feb 8, 2021 · 36861dd · 36861dd
1 parent c870b73
commit 36861dd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/search.go b/search.go
@@ -603,9 +603,11 @@ func MemoryNeededForSearchResult(req *SearchRequest) uint64 {
 	// overhead from fields, highlighting
 	var d document.Document
 	if len(req.Fields) > 0 || req.Highlight != nil {
-		for i := 0; i < numDocMatches; i++ {
-			estimate += d.Size()
+		numDocsApplicable := req.Size
+		if numDocsApplicable > collector.PreAllocSizeSkipCap {
+			numDocsApplicable = collector.PreAllocSizeSkipCap
 		}
+		estimate += numDocsApplicable * d.Size()
 	}
 
 	return uint64(estimate)