Skip to content

Commit 5d1eac5

Browse files
jimczidavidkyle
authored andcommitted
Fix standard retriever rewrite (elastic#114480)
Closes elastic#114466
1 parent 297337f commit 5d1eac5

File tree

2 files changed

+10
-13
lines changed

2 files changed

+10
-13
lines changed

muted-tests.yml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -357,9 +357,6 @@ tests:
357357
- class: org.elasticsearch.xpack.inference.InferenceRestIT
358358
method: test {p0=inference/40_semantic_text_query/Query a field that uses the default ELSER 2 endpoint}
359359
issue: https://github.com/elastic/elasticsearch/issues/114376
360-
- class: org.elasticsearch.search.retriever.StandardRetrieverBuilderParsingTests
361-
method: testRewrite
362-
issue: https://github.com/elastic/elasticsearch/issues/114466
363360
- class: org.elasticsearch.search.retriever.RankDocsRetrieverBuilderTests
364361
method: testRewrite
365362
issue: https://github.com/elastic/elasticsearch/issues/114467

server/src/main/java/org/elasticsearch/search/retriever/StandardRetrieverBuilder.java

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -113,29 +113,29 @@ private StandardRetrieverBuilder(StandardRetrieverBuilder clone) {
113113
@Override
114114
public RetrieverBuilder rewrite(QueryRewriteContext ctx) throws IOException {
115115
boolean changed = false;
116-
List<SortBuilder<?>> newSortBuilders = null;
116+
List<SortBuilder<?>> rewrittenSortBuilders = null;
117117
if (sortBuilders != null) {
118-
newSortBuilders = new ArrayList<>(sortBuilders.size());
118+
rewrittenSortBuilders = new ArrayList<>(sortBuilders.size());
119119
for (var sort : sortBuilders) {
120120
var newSort = sort.rewrite(ctx);
121-
newSortBuilders.add(newSort);
122-
changed = newSort != sort;
121+
rewrittenSortBuilders.add(newSort);
122+
changed |= newSort != sort;
123123
}
124124
}
125125
var rewrittenFilters = rewritePreFilters(ctx);
126126
changed |= rewrittenFilters != preFilterQueryBuilders;
127127

128-
QueryBuilder queryBuilderRewrite = null;
128+
QueryBuilder rewrittenQuery = null;
129129
if (queryBuilder != null) {
130-
queryBuilderRewrite = queryBuilder.rewrite(ctx);
131-
changed |= queryBuilderRewrite != queryBuilder;
130+
rewrittenQuery = queryBuilder.rewrite(ctx);
131+
changed |= rewrittenQuery != queryBuilder;
132132
}
133133

134134
if (changed) {
135135
var rewritten = new StandardRetrieverBuilder(this);
136-
rewritten.sortBuilders = newSortBuilders;
137-
rewritten.preFilterQueryBuilders = preFilterQueryBuilders;
138-
rewritten.queryBuilder = queryBuilderRewrite;
136+
rewritten.sortBuilders = rewrittenSortBuilders;
137+
rewritten.preFilterQueryBuilders = rewrittenFilters;
138+
rewritten.queryBuilder = rewrittenQuery;
139139
return rewritten;
140140
}
141141
return this;

0 commit comments

Comments
 (0)