Skip to content

Commit f4ce957

Browse files
committed
Simplify phrase query generation
1 parent 0861dcc commit f4ce957

File tree

1 file changed

+9
-28
lines changed

1 file changed

+9
-28
lines changed

test/framework/src/main/java/org/elasticsearch/datageneration/queries/LeafQueryGenerator.java

Lines changed: 9 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -82,46 +82,27 @@ public List<QueryBuilder> generate(Map<String, Object> fieldMapping, String path
8282
}
8383
}
8484

85-
var results = new ArrayList<QueryBuilder>();
86-
results.add(QueryBuilders.matchQuery(path, value));
87-
var phraseQuery = buildPhraseQuery(path, (String) value);
88-
if (phraseQuery != null) {
89-
results.add(phraseQuery);
90-
}
91-
return results;
85+
return List.of(
86+
QueryBuilders.matchQuery(path, value),
87+
QueryBuilders.matchPhraseQuery(path, value)
88+
);
9289
}
9390
}
9491

9592
record MatchOnlyTextQueryGenerator(MappingContextHelper mappingContextHelper) implements LeafQueryGenerator {
9693

9794
public List<QueryBuilder> generate(Map<String, Object> fieldMapping, String path, Object value) {
98-
var results = new ArrayList<QueryBuilder>();
99-
results.add(QueryBuilders.matchQuery(path, value));
100-
10195
// TODO remove when fixed
10296
// match_only_text in nested context fails for synthetic source https://github.com/elastic/elasticsearch/issues/132352
10397
if (mappingContextHelper.inNestedContext(path)) {
104-
return results;
98+
return List.of(QueryBuilders.matchQuery(path, value));
10599
}
106100

107-
var phraseQuery = buildPhraseQuery(path, (String) value);
108-
if (phraseQuery != null) {
109-
results.add(phraseQuery);
110-
}
111-
return results;
112-
}
113-
}
114-
115-
private static QueryBuilder buildPhraseQuery(String path, String value) {
116-
var tokens = Arrays.asList(value.split("[^a-zA-Z0-9]"));
117-
if (tokens.isEmpty()) {
118-
return null;
101+
return List.of(
102+
QueryBuilders.matchQuery(path, value),
103+
QueryBuilders.matchPhraseQuery(path, value)
104+
);
119105
}
120-
121-
int low = ESTestCase.randomIntBetween(0, tokens.size() - 1);
122-
int hi = ESTestCase.randomIntBetween(low + 1, tokens.size());
123-
var phrase = String.join(" ", tokens.subList(low, hi));
124-
return QueryBuilders.matchPhraseQuery(path, phrase);
125106
}
126107

127108
static boolean containsHighSurrogates(String s) {

0 commit comments

Comments
 (0)