ElasticSearch6.x 基于SpringBoot 实现ElasticSearch的多条件查询

SpringBoot 功能封装涉及ElasticSearch的多条件查询方法约定如下：

public SearchResponse searchConstantScoreQuery(String[] indexs, String[] types, String name, Object value, float boost)

public SearchResponse searchBoolQuery(String[] indexs, String[] types, List musts, List mustNots, List shoulds, List filters)

public SearchResponse searchDisMaxQuery(String[] indexs, String[] types, List query, float boost, float breaker)

public SearchResponse searchFunctionScoreQuery(String[] indexs, String[] types, QueryBuilder matchQuery, String fieldName, Object origin, Object scale)

public SearchResponse searchBoostiQuery(String[] indexs, String[] types, QueryBuilder positiveQuery,  QueryBuilder negativeQuery, float boost)

在上一篇文中说到：ElasticSearch6.x 基于SpringBoot 实现ElasticSearch连接功能封装,将约定的方法填充到ElasticSearchIndexUtil.java 工具类中。

@Component
public class ElasticSearchIndexUtil {
	// 引入 Ela 连接实列化对象
	@Autowired
	private TransportClient client;

/**
	 * 功能描述:多条件检索之constant score query
	 * @param indexs 索引数组
	 * @param types  类型数组
	 * @param name   文档属性名称
	 * @param value  文档属性值
	 * @param boost  权重值
	 * @return
	 * @throws InterruptedException
	 * @throws ExecutionException
	 */
	public SearchResponse searchConstantScoreQuery(String[] indexs, String[] types, String name, Object value, float boost) throws InterruptedException, ExecutionException{
		TermQueryBuilder termQuery = QueryBuilders.termQuery(name, value);
		
		ConstantScoreQueryBuilder constantScoreQuery = QueryBuilders.constantScoreQuery(termQuery).boost(boost);
		SearchResponse response = client.prepareSearch(indexs).setTypes(types).setQuery(constantScoreQuery).execute().get();
		return response;
	}
	
	
	/**
	 * 功能描述:多条件查询之bool query
	 * @param indexs 索引数组
	 * @param types  类型数组
	 * @param musts  必须查询
	 * @param mustNots 不能查询
	 * @param shoulds  应查询
	 * @param filters  拦截条件
	 * @return
	 * @throws InterruptedException
	 * @throws ExecutionException
	 */
	public SearchResponse searchBoolQuery(String[] indexs, String[] types, List musts, List mustNots, List shoulds, List filters) throws InterruptedException, ExecutionException{
		BoolQueryBuilder boolQuery =  QueryBuilders.boolQuery();
		
		musts.stream().forEach(item->{
			boolQuery.must(item);
		});
		mustNots.stream().forEach(item->{
			boolQuery.mustNot(item);
		});
		shoulds.stream().forEach(item ->{
			boolQuery.should(item);
		});
		filters.stream().forEach(item->{
			boolQuery.filter(item);
		});
		
		SearchResponse response = client.prepareSearch(indexs).setTypes(types).setQuery(boolQuery).execute().get();
		return response;
	}
	
	/**
	 * 功能描述:多条件查询之dis max query
	 * @param indexs   索引数组
	 * @param types    类型数组
	 * @param query    查询条件对列
	 * @param boost    权重值
	 * @param breaker  判断值
	 * @return
	 * @throws ExecutionException 
	 * @throws InterruptedException 
	 */
	public SearchResponse searchDisMaxQuery(String[] indexs, String[] types, List query, float boost, float breaker) throws InterruptedException, ExecutionException{
		DisMaxQueryBuilder disMaxQuery = QueryBuilders.disMaxQuery();
		query.stream().forEach(item ->{
			disMaxQuery.add(item);
		});
		disMaxQuery.boost(boost).tieBreaker(breaker);
		SearchResponse response = client.prepareSearch(indexs).setTypes(types).setQuery(disMaxQuery).execute().get();
		return response;
	}
	
	/**
	 *  功能描述:多条件查询之function score query
	 * @param indexs 索引数组
	 * @param types  类型数组
	 * @param matchQuery  条件
	 * @param fieldName   文档属性名称
	 * @param origin  
	 * @param scale
	 * @return
	 * @throws InterruptedException
	 * @throws ExecutionException
	 */
	public SearchResponse searchFunctionScoreQuery(String[] indexs, String[] types, QueryBuilder matchQuery, String fieldName, Object origin, Object scale) throws InterruptedException, ExecutionException{
		FilterFunctionBuilder[] functions = {
				new FunctionScoreQueryBuilder.FilterFunctionBuilder(matchQuery,  ScoreFunctionBuilders.randomFunction().seed(Math.round(Math.random() * 100))),
				new FunctionScoreQueryBuilder.FilterFunctionBuilder(ScoreFunctionBuilders.exponentialDecayFunction(fieldName, origin, scale))
		};
		SearchResponse response = client.prepareSearch(indexs).setTypes(types).setQuery(QueryBuilders.functionScoreQuery(functions)).execute().get();
		return response;
	}
	
	/**
	 * 功能描述:多条件查询之boost query
	 * @param indexs 索引数组
	 * @param types  类型数组
	 * @param positiveQuery   条件
	 * @param negativeQuery   条件
	 * @param boost  权重值
	 * @return
	 * @throws ExecutionException 
	 * @throws InterruptedException 
	 */
	public SearchResponse searchBoostiQuery(String[] indexs, String[] types, QueryBuilder positiveQuery,  QueryBuilder negativeQuery, float boost) throws InterruptedException, ExecutionException{
		BoostingQueryBuilder boostingQuery = QueryBuilders.boostingQuery(positiveQuery, negativeQuery);
		boostingQuery.boost(boost);
		
		SearchResponse response = client.prepareSearch(indexs).setTypes(types).setQuery(boostingQuery).execute().get();
		return response;
	}
}