hotel-demohotel-demo

package cn.itcast.hotel.service.impl; import cn.itcast.hotel.mapper.HotelMapper; import cn.itcast.hotel.pojo.Hotel; import cn.itcast.hotel.pojo.HotelDoc; import cn.itcast.hotel.pojo.PageResult; import cn.itcast.hotel.pojo.RequestParams; import cn.itcast.hotel.service.IHotelService; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.elasticsearch.action.ActionListener; import org.elasticsearch.action.DocWriteResponse; import org.elasticsearch.action.delete.DeleteRequest; import org.elasticsearch.action.index.IndexRequest; import org.elasticsearch.action.index.IndexResponse; import org.elasticsearch.action.search.SearchRequest; import org.elasticsearch.action.search.SearchResponse; import org.elasticsearch.client.RequestOptions; import org.elasticsearch.client.RestHighLevelClient; import org.elasticsearch.common.geo.GeoPoint; import org.elasticsearch.common.unit.DistanceUnit; import org.elasticsearch.common.xcontent.XContentType; import org.elasticsearch.index.query.BoolQueryBuilder; import org.elasticsearch.index.query.QueryBuilders; import org.elasticsearch.index.query.functionscore.FunctionScoreQueryBuilder; import org.elasticsearch.index.query.functionscore.ScoreFunctionBuilders; import org.elasticsearch.rest.RestStatus; import org.elasticsearch.search.SearchHit; import org.elasticsearch.search.SearchHits; import org.elasticsearch.search.aggregations.AggregationBuilders; import org.elasticsearch.search.aggregations.Aggregations; import org.elasticsearch.search.aggregations.bucket.terms.Terms; import org.elasticsearch.search.fetch.subphase.highlight.HighlightField; import org.elasticsearch.search.sort.SortBuilders; import org.elasticsearch.search.sort.SortOrder; import org.elasticsearch.search.suggest.Suggest; import org.elasticsearch.search.suggest.SuggestBuilder; import org.elasticsearch.search.suggest.SuggestBuilders; import org.elasticsearch.search.suggest.completion.CompletionSuggestion; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; import java.io.IOException; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; /** * 酒店服务 * * @author ZY * @date 2022/11/05 */ @Slf4j @Service public class HotelService extends ServiceImpl<HotelMapper, Hotel> implements IHotelService { @Autowired private RestHighLevelClient restHighLevelClient; @Override public PageResult search(RequestParams params) { try { // 1.准备Request SearchRequest request = new SearchRequest("hotel"); // 2.准备请求参数 // 2.1.query buildBasicQuery(params, request); // 2.2.分页 int page = params.getPage(); int size = params.getSize(); request.source().from((page - 1) * size).size(size); // 2.3.距离排序 String location = params.getLocation(); if (StringUtils.isNotBlank(location)) { request.source().sort(SortBuilders .geoDistanceSort("location", new GeoPoint(location)) .order(SortOrder.ASC) .unit(DistanceUnit.KILOMETERS) ); } // 3.发送请求 SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT); // 4.解析响应 return handleResponse(response); } catch (IOException e) { throw new RuntimeException("搜索数据失败", e); } } @Override public Map<String, List<String>> filters() { try { // 1.准备请求 SearchRequest request = new SearchRequest("hotel"); // 2.请求参数 // 2.1.query //buildBasicQuery(params, request); // 2.2.size request.source().size(0); // 2.3.聚合 buildAggregations(request); // 3.发出请求 SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT); // 4.解析结果 Aggregations aggregations = response.getAggregations(); Map<String, List<String>> filters = new HashMap<>(3); // 4.1.解析品牌 List<String> brandList = getAggregationByName(aggregations, "brandAgg"); filters.put("brand", brandList); // 4.1.解析品牌 List<String> cityList = getAggregationByName(aggregations, "cityAgg"); filters.put("city", cityList); // 4.1.解析品牌 List<String> starList = getAggregationByName(aggregations, "starAgg"); filters.put("starName", starList); return filters; } catch (IOException e) { throw new RuntimeException(e); } } @Override public Map<String, List<String>> getFilters(RequestParams params) { try { // 1.准备请求 SearchRequest request = new SearchRequest("hotel"); // 2.请求参数 // 2.1.query buildBasicQuery(params, request); // 2.2.size request.source().size(0); // 2.3.聚合 buildAggregations(request); // 3.发出请求 SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT); // 4.解析结果 Aggregations aggregations = response.getAggregations(); //System.out.println("aggregations:" + JSON.toJSON(aggregations)); Map<String, List<String>> filters = new HashMap<>(3); // 4.1.解析品牌 List<String> brandList = getAggregationByName(aggregations, "brandAgg"); filters.put("brand", brandList); // 4.1.解析品牌 List<String> cityList = getAggregationByName(aggregations, "cityAgg"); filters.put("city", cityList); // 4.1.解析品牌 List<String> starList = getAggregationByName(aggregations, "starAgg"); filters.put("starName", starList); System.out.println(JSON.toJSON(filters)); return filters; } catch (IOException e) { throw new RuntimeException(e); } } @Override public List<String> getSuggestion(String key) { try { // 1.准备请求 SearchRequest request = new SearchRequest("hotel"); // 2.请求参数 request.source().suggest(new SuggestBuilder() .addSuggestion( "hotelSuggest", SuggestBuilders .completionSuggestion("suggestion") .size(10) .skipDuplicates(true) .prefix(key) )); // 3.发出请求 SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT); // 4.解析 Suggest suggest = response.getSuggest(); // 4.1.根据名称获取结果 CompletionSuggestion suggestion = suggest.getSuggestion("hotelSuggest"); // 4.2.获取options List<String> list = new ArrayList<>(); for (CompletionSuggestion.Entry.Option option : suggestion.getOptions()) { // 4.3.获取补全的结果 String str = option.getText().toString();