80
80
import java .util .stream .Collectors ;
81
81
82
82
import static org .opensearch .search .query .QueryCollectorContext .createEarlyTerminationCollectorContext ;
83
+ import static org .opensearch .search .query .QueryCollectorContext .createFakeContext ;
83
84
import static org .opensearch .search .query .QueryCollectorContext .createFilteredCollectorContext ;
84
85
import static org .opensearch .search .query .QueryCollectorContext .createMinScoreCollectorContext ;
85
86
import static org .opensearch .search .query .QueryCollectorContext .createMultiCollectorContext ;
@@ -100,6 +101,7 @@ public class QueryPhase {
100
101
private final QueryPhaseSearcher queryPhaseSearcher ;
101
102
private final SuggestProcessor suggestProcessor ;
102
103
private final RescoreProcessor rescoreProcessor ;
104
+ // private final QueryContextProvider contextProvider;
103
105
104
106
public QueryPhase () {
105
107
this (DEFAULT_QUERY_PHASE_SEARCHER );
@@ -109,6 +111,7 @@ public QueryPhase(QueryPhaseSearcher queryPhaseSearcher) {
109
111
this .queryPhaseSearcher = Objects .requireNonNull (queryPhaseSearcher , "QueryPhaseSearcher is required" );
110
112
this .suggestProcessor = new SuggestProcessor ();
111
113
this .rescoreProcessor = new RescoreProcessor ();
114
+ // this.contextProvider = new QueryContextProvider();
112
115
}
113
116
114
117
public void preProcess (SearchContext context ) {
@@ -192,6 +195,7 @@ static boolean executeInternal(SearchContext searchContext, QueryPhaseSearcher q
192
195
final ContextIndexSearcher searcher = searchContext .searcher ();
193
196
final IndexReader reader = searcher .getIndexReader ();
194
197
QuerySearchResult queryResult = searchContext .queryResult ();
198
+ QueryContextProvider contextProvider = new QueryContextProvider ();
195
199
queryResult .searchTimedOut (false );
196
200
try {
197
201
queryResult .from (searchContext .from ());
@@ -248,7 +252,10 @@ static boolean executeInternal(SearchContext searchContext, QueryPhaseSearcher q
248
252
.map (Map .Entry ::getValue )
249
253
.collect (Collectors .toList ());
250
254
if (managersExceptGlobalAgg .isEmpty () == false ) {
251
- collectors .add (createMultiCollectorContext (managersExceptGlobalAgg ));
255
+ contextProvider .addQueryContext (createFakeContext (), 100 );
256
+ contextProvider .addQueryContext (createMultiCollectorContext (managersExceptGlobalAgg ), 50 );
257
+ collectors .add (contextProvider .getComposedContext ());
258
+ // collectors.add(createMultiCollectorContext(managersExceptGlobalAgg));
252
259
}
253
260
254
261
if (searchContext .minimumScore () != null ) {
0 commit comments