首页 > 代码库 > Elasticsearch 5.4.3实战--Java API调用:搜索建议
Elasticsearch 5.4.3实战--Java API调用:搜索建议
通常的搜索引擎,都会根据用户的输入,实时给予匹配的提示。 那么这个功能在elasticsearch中如何实现呢?
Elasticsearch里设计了4种类别的Suggester,分别是:
- Term Suggester
- Phrase Suggester
- Completion Suggester
- Context Suggester
我是采用Completion Suggester来实现的。
原理的文章网上有很多,这里不多说。我们直接上代码:
1 package com.cs99lzzs.elasticsearch.service.imp; 2 3 import java.text.DecimalFormat; 4 import java.util.ArrayList; 5 import java.util.Iterator; 6 import java.util.List; 7 import java.util.Map; 8 9 import javax.annotation.Resource; 10 11 import org.apache.commons.lang.StringUtils; 12 import org.apache.log4j.Logger; 13 import org.apache.lucene.queryparser.classic.QueryParser; 14 import org.elasticsearch.action.search.SearchRequestBuilder; 15 import org.elasticsearch.action.search.SearchResponse; 16 import org.elasticsearch.action.search.SearchType; 17 import org.elasticsearch.client.Client; 18 import org.elasticsearch.index.query.BoolQueryBuilder; 19 import org.elasticsearch.index.query.QueryBuilder; 20 import org.elasticsearch.index.query.QueryBuilders; 21 import org.elasticsearch.search.SearchHit; 22 import org.elasticsearch.search.aggregations.Aggregation; 23 import org.elasticsearch.search.aggregations.AggregationBuilders; 24 import org.elasticsearch.search.aggregations.bucket.terms.StringTerms; 25 import org.elasticsearch.search.aggregations.bucket.terms.Terms.Bucket; 26 import org.elasticsearch.search.aggregations.bucket.terms.TermsAggregationBuilder; 27 import org.elasticsearch.search.sort.SortBuilder; 28 import org.elasticsearch.search.sort.SortBuilders; 29 import org.elasticsearch.search.sort.SortOrder; 30 import org.elasticsearch.search.suggest.Suggest; 31 import org.elasticsearch.search.suggest.SuggestBuilder; 32 import org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder; 33 import org.springframework.beans.factory.annotation.Value; 34 import org.springframework.stereotype.Service; 35 36 37 @Service("productSearchService") 38 public class ProductSearchServiceImp implements ProductSearchService { 39 40 private static Logger logger = Logger.getLogger(ProductSearchService.class); 41 42 private static String aggrationBrandName = "brand_count"; 43 44 private static String suggestZhName = "suggestName"; 45 46 @Resource(name="esClient") 47 Client esClient; 48 49 @Value("${elasticsearch.index}") 50 private String CLUSTER_INDEX; 51 52 @Value("${elasticsearch.type}") 53 private String CLUSTER_TYPE; 54 55 56 /** 57 * @return 58 */ 59 @Override 60 public List<String> getSuggestList(String keyword, int number) { 61 List<String> suggestList = new ArrayList<String>(); 62 63 CompletionSuggestionBuilder suggestZhNameBuilder = new CompletionSuggestionBuilder(suggestZhName).prefix(keyword).size(100); 64 SuggestBuilder suggestBuilder = new SuggestBuilder() 65 .addSuggestion(suggestZhName, suggestZhNameBuilder); 66 SearchResponse suggestResponse = esClient.prepareSearch(CLUSTER_INDEX) 67 .setTypes(CLUSTER_TYPE) 68 // .setQuery(QueryBuilders.matchAllQuery()) 69 .suggest(suggestBuilder) 70 .execute() 71 .actionGet(); 72 73 List<? extends Suggest.Suggestion.Entry<? extends Suggest.Suggestion.Entry.Option>> result1 = suggestResponse 74 .getSuggest().getSuggestion(suggestZhName).getEntries(); 75 for (Suggest.Suggestion.Entry<? extends Suggest.Suggestion.Entry.Option> op : result1) { 76 List<? extends Suggest.Suggestion.Entry.Option> options = op.getOptions(); 77 if (suggestList.size() < number) { 78 for (Suggest.Suggestion.Entry.Option pp : options) { 79 if(suggestList.size() < number && !suggestList.contains(pp.getText().toString())){ 80 suggestList.add(pp.getText().toString()); 81 } 82 } 83 }else{ 84 break; 85 } 86 } 87 88 return suggestList; 89 } 90 91 }
Elasticsearch 5.4.3实战--Java API调用:搜索建议
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。