1. @Slf4j
    2. @Service
    3. public class HotelService extends ServiceImpl<HotelMapper, Hotel> implements IHotelService {
    5.     @Autowired
    6.     private RestHighLevelClient restHighLevelClient;
    8.     @Override
    9.     public PageResult search(RequestParams params) {
    10.         try {
    11.             // 1.准备Request
    12.             SearchRequest request = new SearchRequest("hotel");
    13.             // 2.准备请求参数
    14.             // 2.1.query
    15.             buildBasicQuery(params, request);
    16.             // 2.2.分页
    17.             int page = params.getPage();
    18.             int size = params.getSize();
    19.             request.source().from((page - 1) * size).size(size);
    20.             // 2.3.距离排序
    21.             String location = params.getLocation();
    22.             if (StringUtils.isNotBlank(location)) {
    23.                 request.source().sort(SortBuilders
    24.                                       .geoDistanceSort("location", new GeoPoint(location))
    25.                                       .order(SortOrder.ASC)
    26.                                       .unit(DistanceUnit.KILOMETERS)
    27.                                      );
    28.             }
    29.             // 3.发送请求
    30.             SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT);
    31.             // 4.解析响应
    32.             return handleResponse(response);
    33.         } catch (IOException e) {
    34.             throw new RuntimeException("搜索数据失败", e);
    35.         }
    36.     }
    38.     private void buildBasicQuery(RequestParams params, SearchRequest request) {
    39.         // 1.准备Boolean查询
    40.         BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
    42.         // 1.1.关键字搜索，match查询，放到must中
    43.         String key = params.getKey();
    44.         if (StringUtils.isNotBlank(key)) {
    45.             // 不为空，根据关键字查询
    46.             boolQuery.must(QueryBuilders.matchQuery("all", key));
    47.         } else {
    48.             // 为空，查询所有
    49.             boolQuery.must(QueryBuilders.matchAllQuery());
    50.         }
    52.         // 1.2.品牌
    53.         String brand = params.getBrand();
    54.         if (StringUtils.isNotBlank(brand)) {
    55.             boolQuery.filter(QueryBuilders.termQuery("brand", brand));
    56.         }
    57.         // 1.3.城市
    58.         String city = params.getCity();
    59.         if (StringUtils.isNotBlank(city)) {
    60.             boolQuery.filter(QueryBuilders.termQuery("city", city));
    61.         }
    62.         // 1.4.星级
    63.         String starName = params.getStarName();
    64.         if (StringUtils.isNotBlank(starName)) {
    65.             boolQuery.filter(QueryBuilders.termQuery("starName", starName));
    66.         }
    67.         // 1.5.价格范围
    68.         Integer minPrice = params.getMinPrice();
    69.         Integer maxPrice = params.getMaxPrice();
    70.         if (minPrice != null && maxPrice != null) {
    71.             maxPrice = maxPrice == 0 ? Integer.MAX_VALUE : maxPrice;
    72.             boolQuery.filter(QueryBuilders.rangeQuery("price").gte(minPrice).lte(maxPrice));
    73.         }
    75.         // 2.算分函数查询
    76.         FunctionScoreQueryBuilder functionScoreQuery = QueryBuilders.functionScoreQuery(
    77.             boolQuery, // 原始查询，boolQuery
    78.             new FunctionScoreQueryBuilder.FilterFunctionBuilder[]{ // function数组
    79.                 new FunctionScoreQueryBuilder.FilterFunctionBuilder(
    80.                     QueryBuilders.termQuery("isAD", true), // 过滤条件
    81.                     ScoreFunctionBuilders.weightFactorFunction(10) // 算分函数
    82.                 )
    83.                 }
    84.         );
    86.         // 3.设置查询条件
    87.         request.source().query(functionScoreQuery);
    88.     }
    90.     private PageResult handleResponse(SearchResponse response) {
    91.         SearchHits searchHits = response.getHits();
    92.         // 4.1.总条数
    93.         long total = searchHits.getTotalHits().value;
    94.         // 4.2.获取文档数组
    95.         SearchHit[] hits = searchHits.getHits();
    96.         // 4.3.遍历
    97.         List<HotelDoc> hotels = new ArrayList<>(hits.length);
    98.         for (SearchHit hit : hits) {
    99.             // 4.4.获取source
    100.             String json = hit.getSourceAsString();
    101.             // 4.5.反序列化，非高亮的
    102.             HotelDoc hotelDoc = JSON.parseObject(json, HotelDoc.class);
    103.             // 4.6.处理高亮结果
    104.             // 1)获取高亮map
    105.             Map<String, HighlightField> map = hit.getHighlightFields();
    106.             if (map != null && !map.isEmpty()) {
    107.                 // 2）根据字段名，获取高亮结果
    108.                 HighlightField highlightField = map.get("name");
    109.                 if (highlightField != null) {
    110.                     // 3）获取高亮结果字符串数组中的第1个元素
    111.                     String hName = highlightField.getFragments()[0].toString();
    112.                     // 4）把高亮结果放到HotelDoc中
    113.                     hotelDoc.setName(hName);
    114.                 }
    115.             }
    116.             // 4.8.排序信息
    117.             Object[] sortValues = hit.getSortValues();
    118.             if (sortValues.length > 0) {
    119.                 hotelDoc.setDistance(sortValues[0]);
    120.             }
    122.             // 4.9.放入集合
    123.             hotels.add(hotelDoc);
    124.         }
    125.         return new PageResult(total, hotels);
    126.     }
    127. }