13
13
import org .elasticsearch .action .search .SearchRequestBuilder ;
14
14
import org .elasticsearch .action .search .SearchResponse ;
15
15
import org .elasticsearch .action .search .SearchType ;
16
- import org . elasticsearch .common .collect .Lists ;
17
- import org . elasticsearch .common .collect .Maps ;
18
- import org . elasticsearch .common .collect .Sets ;
16
+ import com . google .common .collect .Lists ;
17
+ import com . google .common .collect .Maps ;
18
+ import com . google .common .collect .Sets ;
19
19
import org .elasticsearch .common .logging .ESLogger ;
20
20
import org .elasticsearch .common .logging .Loggers ;
21
- import org .elasticsearch .index .query .FilterBuilder ;
22
- import org .elasticsearch .index .query .FilterBuilders ;
21
+ // import org.elasticsearch.index.query.FilterBuilder;
22
+ // import org.elasticsearch.index.query.FilterBuilders;
23
23
import org .elasticsearch .index .query .QueryBuilder ;
24
24
import org .elasticsearch .index .query .QueryBuilders ;
25
25
import org .elasticsearch .index .query .functionscore .ScoreFunctionBuilders ;
@@ -135,7 +135,7 @@ public interface IFilterableQueryBuilderHelper<T extends IFilterableQueryBuilder
135
135
* @param customFilter user provided filters.
136
136
* @return current instance.
137
137
*/
138
- T filters (FilterBuilder ... customFilter );
138
+ T filters (QueryBuilder ... customFilter );
139
139
140
140
/**
141
141
* Add filters to the current query.
@@ -144,7 +144,7 @@ public interface IFilterableQueryBuilderHelper<T extends IFilterableQueryBuilder
144
144
* @param customFilters user provided filters to add (using and clause) to the annotation based filters.
145
145
* @return current instance.
146
146
*/
147
- T filters (Map <String , String []> filters , FilterBuilder ... customFilters );
147
+ T filters (Map <String , String []> filters , QueryBuilder ... customFilters );
148
148
149
149
/**
150
150
* Add filters to the current query.
@@ -154,7 +154,7 @@ public interface IFilterableQueryBuilderHelper<T extends IFilterableQueryBuilder
154
154
* @param customFilters user provided filters to add (using and clause) to the annotation based filters.
155
155
* @return current instance.
156
156
*/
157
- T filters (Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies , FilterBuilder ... customFilters );
157
+ T filters (Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies , QueryBuilder ... customFilters );
158
158
159
159
/**
160
160
*
@@ -333,18 +333,18 @@ public QueryBuilderHelper prepareSearch(String... indices) {
333
333
}
334
334
335
335
@ Override
336
- public QueryBuilderHelper filters (FilterBuilder ... customFilter ) {
336
+ public QueryBuilderHelper filters (QueryBuilder ... customFilter ) {
337
337
this .queryBuilder = addFilters (queryBuilder , Lists .newArrayList (customFilter ));
338
338
return this ;
339
339
}
340
340
341
341
@ Override
342
- public QueryBuilderHelper filters (Map <String , String []> filters , FilterBuilder ... customFilters ) {
342
+ public QueryBuilderHelper filters (Map <String , String []> filters , QueryBuilder ... customFilters ) {
343
343
return this .filters (filters , null , customFilters );
344
344
}
345
345
346
346
@ Override
347
- public QueryBuilderHelper filters (Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies , FilterBuilder ... customFilters ) {
347
+ public QueryBuilderHelper filters (Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies , QueryBuilder ... customFilters ) {
348
348
this .filters = filters ;
349
349
if (classes != null && classes .length > 0 ) {
350
350
QueryBuilder filteredQueryBuilder = addFilters (this .queryBuilder , classes [0 ], filters , filterStrategies , customFilters );
@@ -359,13 +359,13 @@ public QueryBuilderHelper filters(Map<String, String[]> filters, Map<String, Fil
359
359
}
360
360
361
361
private QueryBuilder addFilters (QueryBuilder query , Class <?> clazz , Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies ,
362
- FilterBuilder ... customFilters ) {
362
+ QueryBuilder ... customFilters ) {
363
363
if (clazz == null ) {
364
364
return query ;
365
365
}
366
- final List <FilterBuilder > esFilters = buildFilters (clazz .getName (), filters , filterStrategies );
366
+ final List <QueryBuilder > esFilters = buildFilters (clazz .getName (), filters , filterStrategies );
367
367
if (customFilters != null ) {
368
- for (FilterBuilder customFilter : customFilters ) {
368
+ for (QueryBuilder customFilter : customFilters ) {
369
369
if (customFilter != null ) {
370
370
esFilters .add (customFilter );
371
371
}
@@ -375,8 +375,8 @@ private QueryBuilder addFilters(QueryBuilder query, Class<?> clazz, Map<String,
375
375
return addFilters (query , esFilters );
376
376
}
377
377
378
- private QueryBuilder addFilters (QueryBuilder query , final List <FilterBuilder > esFilters ) {
379
- FilterBuilder filter = null ;
378
+ private QueryBuilder addFilters (QueryBuilder query , final List <QueryBuilder > esFilters ) {
379
+ QueryBuilder filter = null ;
380
380
if (esFilters .size () > 0 ) {
381
381
filter = getAndFilter (esFilters );
382
382
if (filter != null ) {
@@ -386,8 +386,8 @@ private QueryBuilder addFilters(QueryBuilder query, final List<FilterBuilder> es
386
386
return query ;
387
387
}
388
388
389
- private List <FilterBuilder > buildFilters (String className , Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies ) {
390
- List <FilterBuilder > filterBuilders = new ArrayList <FilterBuilder >();
389
+ private List <QueryBuilder > buildFilters (String className , Map <String , String []> filters , Map <String , FilterValuesStrategy > filterStrategies ) {
390
+ List <QueryBuilder > filterBuilders = new ArrayList <QueryBuilder >();
391
391
392
392
if (filters == null ) {
393
393
return filterBuilders ;
@@ -397,7 +397,7 @@ private List<FilterBuilder> buildFilters(String className, Map<String, String[]>
397
397
filterStrategies = Maps .newHashMap ();
398
398
}
399
399
400
- Map <String , List <FilterBuilder >> nestedFilterBuilders = new HashMap <String , List <FilterBuilder >>();
400
+ Map <String , List <QueryBuilder >> nestedFilterBuilders = new HashMap <String , List <QueryBuilder >>();
401
401
List <IFilterBuilderHelper > filterBuilderHelpers = mappingBuilder .getFilters (className );
402
402
if (filterBuilderHelpers == null ) {
403
403
return filterBuilders ;
@@ -407,9 +407,9 @@ private List<FilterBuilder> buildFilters(String className, Map<String, String[]>
407
407
String esFieldName = filterBuilderHelper .getEsFieldName ();
408
408
if (filters .containsKey (esFieldName )) {
409
409
if (filterBuilderHelper .isNested ()) {
410
- List <FilterBuilder > nestedFilters = nestedFilterBuilders .get (filterBuilderHelper .getNestedPath ());
410
+ List <QueryBuilder > nestedFilters = nestedFilterBuilders .get (filterBuilderHelper .getNestedPath ());
411
411
if (nestedFilters == null ) {
412
- nestedFilters = new ArrayList <FilterBuilder >(3 );
412
+ nestedFilters = new ArrayList <QueryBuilder >(3 );
413
413
nestedFilterBuilders .put (filterBuilderHelper .getNestedPath (), nestedFilters );
414
414
}
415
415
nestedFilters .addAll (buildFilters (filterBuilderHelper , esFieldName , filters .get (esFieldName ), filterStrategies .get (esFieldName )));
@@ -419,29 +419,29 @@ private List<FilterBuilder> buildFilters(String className, Map<String, String[]>
419
419
}
420
420
}
421
421
422
- for (Entry <String , List <FilterBuilder >> nestedFilters : nestedFilterBuilders .entrySet ()) {
423
- filterBuilders .add (FilterBuilders . nestedFilter (nestedFilters .getKey (), getAndFilter (nestedFilters .getValue ())));
422
+ for (Entry <String , List <QueryBuilder >> nestedFilters : nestedFilterBuilders .entrySet ()) {
423
+ filterBuilders .add (QueryBuilders . nestedQuery (nestedFilters .getKey (), getAndFilter (nestedFilters .getValue ())));
424
424
}
425
425
426
426
return filterBuilders ;
427
427
}
428
428
429
- private List <FilterBuilder > buildFilters (IFilterBuilderHelper filterBuilderHelper , String esFieldName , String [] values , FilterValuesStrategy strategy ) {
429
+ private List <QueryBuilder > buildFilters (IFilterBuilderHelper filterBuilderHelper , String esFieldName , String [] values , FilterValuesStrategy strategy ) {
430
430
if (strategy == null || FilterValuesStrategy .OR .equals (strategy )) {
431
431
return Lists .newArrayList (filterBuilderHelper .buildFilter (esFieldName , values ));
432
432
}
433
- List <FilterBuilder > valuesFilters = Lists .newArrayList ();
433
+ List <QueryBuilder > valuesFilters = Lists .newArrayList ();
434
434
for (String value : values ) {
435
435
valuesFilters .add (filterBuilderHelper .buildFilter (esFieldName , value ));
436
436
}
437
437
return valuesFilters ;
438
438
}
439
439
440
- private FilterBuilder getAndFilter (List <FilterBuilder > filters ) {
440
+ private QueryBuilder getAndFilter (List <QueryBuilder > filters ) {
441
441
if (filters .size () == 1 ) {
442
442
return filters .get (0 );
443
443
}
444
- return FilterBuilders . andFilter (filters .toArray (new FilterBuilder [filters .size ()]));
444
+ return QueryBuilders . andQuery (filters .toArray (new QueryBuilder [filters .size ()]));
445
445
}
446
446
447
447
@ Override
0 commit comments