21
21
22
22
import org .elasticsearch .index .query .QueryBuilder ;
23
23
import org .elasticsearch .script .mustache .SearchTemplateRequestBuilder ;
24
+ import org .elasticsearch .search .SearchExtBuilder ;
24
25
import org .elasticsearch .search .aggregations .AbstractAggregationBuilder ;
25
26
import org .elasticsearch .search .aggregations .PipelineAggregationBuilder ;
26
27
import org .elasticsearch .search .collapse .CollapseBuilder ;
41
42
* @author Jean-Baptiste Nizet
42
43
* @author Martin Choraine
43
44
* @author Peter-Josef Meisch
45
+ * @author Sijia Liu
44
46
*/
45
47
public class NativeSearchQuery extends BaseQuery {
46
48
@@ -56,6 +58,7 @@ public class NativeSearchQuery extends BaseQuery {
56
58
@ Nullable private List <IndexBoost > indicesBoost ;
57
59
@ Nullable private SearchTemplateRequestBuilder searchTemplate ;
58
60
@ Nullable private SuggestBuilder suggestBuilder ;
61
+ @ Nullable private List <SearchExtBuilder > searchExtBuilders ;
59
62
60
63
public NativeSearchQuery (@ Nullable QueryBuilder query ) {
61
64
@@ -201,4 +204,20 @@ public void setSuggestBuilder(SuggestBuilder suggestBuilder) {
201
204
public SuggestBuilder getSuggestBuilder () {
202
205
return suggestBuilder ;
203
206
}
207
+
208
+ public void setSearchExtBuilders (List <SearchExtBuilder > searchExtBuilders ) {
209
+ this .searchExtBuilders = searchExtBuilders ;
210
+ }
211
+
212
+ public void addSearchExtBuilder (SearchExtBuilder searchExtBuilder ) {
213
+ if (searchExtBuilders == null ) {
214
+ searchExtBuilders = new ArrayList <>();
215
+ }
216
+ searchExtBuilders .add (searchExtBuilder );
217
+ }
218
+
219
+ @ Nullable
220
+ public List <SearchExtBuilder > getSearchExtBuilders () {
221
+ return searchExtBuilders ;
222
+ }
204
223
}
0 commit comments