elasticsearch6.3 自动补齐suggest java代码

package com.sinosafe.essuggest.service.Impl;

import com.sinosafe.essuggest.provider.Impl.EsSingleCon;
import com.sinosafe.essuggest.service.EsSuggestService;
import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.search.suggest.Suggest;
import org.elasticsearch.search.suggest.SuggestBuilder;
import org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author : zengxiaoyi
 * @E-mail :
 * @Date : 2018/6/29
 */
@Service
public class EsSuggestServiceImpl implements EsSuggestService {

 @Autowired
 private EsSingleCon esSingleCon;
 
 /**
 * es get suggest
 * @param index 索引
 * @param type type，用的是es6.3，可以不写type，因为index下只允许一个type了
 * @param field 字段
 * @param context 内容
 * @return
 */
 @Override
 public List getSuggest(String index, String type, String field, String context) {
 List<String> result = new ArrayList<>();
 TransportClient transportClient=EsSingleCon.transportClient;
 SearchRequestBuilder searchRequestBuilder = transportClient.prepareSearch(index);
 //completion suggest
 CompletionSuggestionBuilder completionSuggestionBuilder = new CompletionSuggestionBuilder(field);
 //前缀查询 每次返回最多10条数据
 completionSuggestionBuilder.prefix(context).size(10);
 //"mysuggest"自定义名字
 SuggestBuilder suggestBuilder = new SuggestBuilder().addSuggestion("mysuggest", completionSuggestionBuilder);
 SearchResponse searchResponse = searchRequestBuilder.suggest(suggestBuilder).execute().actionGet();
 //保存es返回结果
 List<? extends Suggest.Suggestion.Entry<? extends Suggest.Suggestion.Entry.Option>> list = searchResponse
 .getSuggest().getSuggestion("mysuggest").getEntries();
 if (list == null) {
return null;
 }
else {
//转为list保存结果字符串
 for (Suggest.Suggestion.Entry<? extends Suggest.Suggestion.Entry.Option> e : list) {
for (Suggest.Suggestion.Entry.Option option : e) {
 result.add(option.getText().toString());
 System.out.println(option.getText().toString());
 }
 }
 }
 System.out.println(result);
 return result;

 }
}

转载于:https://www.cnblogs.com/zengxiaoyi36/p/9257478.html