1. @Service("SearchServiceImpl")
    2. public class SearchServiceImpl implements MallSearchService {
    4.     @Autowired
    5.     private RestHighLevelClient restHighLevelClient;
    7.     @Override
    8.     public SearchResult search(SearchParam searchParam) {
    9.         SearchResult result = null;
    11.         //1.检索请求
    12.         SearchRequest searchRequest = buildSearchRequest(searchParam);
    13.         try {
    14.             //2.执行请求
    15.             SearchResponse response = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT);
    16.             //3.分析响应数据, 封装成我们需要的格式
    17.             result = buildSearchResult(response, searchParam);
    18.         } catch (IOException e) {
    19.             e.printStackTrace();
    20.         }
    21.         return result;
    22.     }
    25.     private SearchRequest buildSearchRequest(SearchParam param) {
    27.         //1.指定DSL检索条件, 构建查询语句
    28.         SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
    30.         BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
    32.         /* 查询: 模糊匹配, 过滤(按照属性/分类/品牌/价格区间/库存) */
    33.         //1.1 must-skuTitle
    34.         if (StringUtils.isNotEmpty(param.getKeyword())) {
    35.             boolQuery.must(QueryBuilders.termQuery("skuTitle", param.getKeyword()));
    36.         }
    38.         //1.2 filter过滤查询
    39.         if (param.getCatalog3Id() != null) {
    40.             boolQuery.filter(QueryBuilders.termQuery("catelogId", param.getCatalog3Id()));
    41.         }
    43.         if (param.getBrandId() != null && param.getBrandId().size() > 0) {
    44.             TermsQueryBuilder termsQuery = QueryBuilders.termsQuery("brandId", param.getBrandId());
    45.             boolQuery.filter(termsQuery);
    46.         }
    48.         if (param.getAttrs() != null && param.getAttrs().size() > 0) {
    49.             for (String attr : param.getAttrs()) {
    50.                 BoolQueryBuilder nestedBoolQuery = QueryBuilders.boolQuery();
    51.                 //attrs=1_a13:b14&attrs=2_2018:2019
    52.                 String[] s = attr.split("_");
    53.                 String attrId = s[0];
    54.                 String[] attrValues = s[1].split(":");
    55.                 nestedBoolQuery.must(QueryBuilders.termQuery("attrs.attrId", attrId));
    56.                 nestedBoolQuery.must(QueryBuilders.termsQuery("attrs.attrValue", attrValues));
    58.                 NestedQueryBuilder nestedQuery = QueryBuilders.nestedQuery("attrs", nestedBoolQuery, ScoreMode.None);
    59.                 boolQuery.filter(nestedQuery);
    60.             }
    61.         }
    63.         if (param.getHasStock() != null) {
    64.             boolQuery.filter(QueryBuilders.termQuery("hasStock", param.getHasStock() == 1));
    65.         }
    67.         if (param.getSkuPrice() != null) {
    68.             RangeQueryBuilder rangeQuery = QueryBuilders.rangeQuery("skuPrice");
    69.             //4000_6000/_6000/4000_
    70.             String[] s = param.getSkuPrice().split("_");
    71.             if (param.getSkuPrice().endsWith("_")) {
    72.                 rangeQuery.gte(s[0]);
    73.             } else {
    74.                 rangeQuery.gte(s[0]).lte(s[1]);
    75.             }
    76.             boolQuery.filter(rangeQuery);
    77.         }
    79.         sourceBuilder.query(boolQuery);
    82.         /* 排序, 分页, 高亮 */
    83.         //2.1 排序 skuPrice_asc/desc
    84.         if (StringUtils.isNotEmpty(param.getSort())) {
    85.             String sort = param.getSort();
    86.             String[] s = sort.split("_");
    87.             SortOrder order = s[1].equalsIgnoreCase("asc") ? SortOrder.ASC : SortOrder.DESC;
    88.             sourceBuilder.sort(s[0], order);
    89.         }
    91.         //2.2 分页  pageSize: 5
    92.         // pageNum:1  from:0  size:5  [0,1,2,3,4]
    93.         // pageNum:2  from:5  size:5  [5,6,7,8,9]
    94.         // from = (pageNum-1)*size
    95.         sourceBuilder.from((param.getPageNum() - 1) * EsConstant.PRODUCT_PAGESIZE);
    96.         sourceBuilder.size(EsConstant.PRODUCT_PAGESIZE);
    98.         //2.3 高亮
    99.         if (StringUtils.isNotEmpty(param.getKeyword())) {
    100.             HighlightBuilder builder = new HighlightBuilder();
    101.             builder.field("skuTitle");
    102.             builder.preTags("<b style='color:red'>");
    103.             builder.postTags("</b>");
    104.             sourceBuilder.highlighter(builder);
    105.         }
    107.         /**
    108.          * 聚合分析
    109.          */
    110.         //1.品牌聚合
    111.         TermsAggregationBuilder brand_agg = AggregationBuilders.terms("brand_agg").field("brandId");
    112.         TermsAggregationBuilder brand_name_agg = AggregationBuilders.terms("brand_name_agg").field("brandName").size(1);
    113.         TermsAggregationBuilder brand_img_agg = AggregationBuilders.terms("brand_img_agg").field("brandImg").size(1);
    114.         brand_agg.subAggregation(brand_name_agg);
    115.         brand_agg.subAggregation(brand_img_agg);
    116.         sourceBuilder.aggregation(brand_agg);
    118.         //2.分类聚合
    119.         TermsAggregationBuilder catelog_agg = AggregationBuilders.terms("catelog_agg").field("catelogId");
    120.         TermsAggregationBuilder catelog_name_agg = AggregationBuilders.terms("catelog_name_agg").field("catelogName").size(1);
    121.         catelog_agg.subAggregation(catelog_name_agg);
    122.         sourceBuilder.aggregation(catelog_agg);
    124.         //3.属性聚合
    125.         NestedAggregationBuilder attr_agg = AggregationBuilders.nested("attr_agg", "attrs");
    126.         TermsAggregationBuilder attr_id_agg = AggregationBuilders.terms("attr_id_agg").field("attrs.attrId");
    127.         TermsAggregationBuilder attr_name_agg = AggregationBuilders.terms("attr_name_agg").field("attrs.attrName");
    128.         TermsAggregationBuilder attr_value_agg = AggregationBuilders.terms("attr_value_agg").field("attrs.attrValue");
    129.         attr_id_agg.subAggregation(attr_name_agg);
    130.         attr_id_agg.subAggregation(attr_value_agg);
    131.         attr_agg.subAggregation(attr_id_agg);
    132.         sourceBuilder.aggregation(attr_agg);
    135.         System.out.println("构造的DSL" + sourceBuilder.toString());
    137.         return new SearchRequest(new String[]{EsConstant.PRODUCT_INDEX}, sourceBuilder);
    138.     }
    140.     private SearchResult buildSearchResult(SearchResponse response, SearchParam param) {
    141.         SearchResult result = new SearchResult();
    143.         SearchHits hits = response.getHits();
    144.         List<SkuEsModelTo> skuEsModelTos = Arrays.stream(hits.getHits()).map(hit -> {
    145.             String sourceAsString = hit.getSourceAsString();
    146.             SkuEsModelTo model = JSON.parseObject(sourceAsString, SkuEsModelTo.class);
    147.             //高亮
    148.             if (StringUtils.isNotEmpty(param.getKeyword())) {
    149.                 Map<String, HighlightField> highlightFields = hit.getHighlightFields();
    150.                 HighlightField highlightField = highlightFields.get("skuTitle");
    151.                 String highLight = highlightField.getFragments()[0].string();
    152.                 System.out.println("highlightFields" + highlightField.getFragments()[0]);
    153.                 model.setSkuTitle(highLight);
    154.             }
    155.             return model;
    156.         }).collect(Collectors.toList());
    157.         result.setProducts(skuEsModelTos);
    159.         /**
    160.          * 聚合分析结果
    161.          */
    162.         Aggregations aggregations = response.getAggregations();
    164.         //2.当前命中结果聚合分析出来的所有品牌信息
    165.         ParsedLongTerms brand_agg = aggregations.get("brand_agg");
    166.         List<SearchResult.BrandVo> brandVos = brand_agg.getBuckets().stream().map(bucket -> {
    167.             SearchResult.BrandVo brandVo = new SearchResult.BrandVo();
    168.             Long brandId = bucket.getKeyAsNumber().longValue();
    169.             ParsedStringTerms brand_img_agg = bucket.getAggregations().get("brand_img_agg");
    170.             ParsedStringTerms brand_name_agg = bucket.getAggregations().get("brand_name_agg");
    171.             String brandImg = brand_img_agg.getBuckets().get(0).getKeyAsString();
    172.             String brandName = brand_name_agg.getBuckets().get(0).getKeyAsString();
    173.             brandVo.setBrandId(brandId);
    174.             brandVo.setBrandName(brandName);
    175.             brandVo.setBrandImg(brandImg);
    176.             return brandVo;
    177.         }).collect(Collectors.toList());
    178.         result.setBrands(brandVos);
    180.         //3.当前所有商品涉及到的所有分类信息
    181.         ParsedLongTerms catelog_agg = aggregations.get("catelog_agg");
    182.         List<SearchResult.CatalogVo> catalogVos = catelog_agg.getBuckets().stream().map(bucket -> {
    183.             SearchResult.CatalogVo catalogVo = new SearchResult.CatalogVo();
    184.             long catelogId = bucket.getKeyAsNumber().longValue();
    185.             ParsedStringTerms catelog_name_agg = bucket.getAggregations().get("catelog_name_agg");
    186.             String catelogName = catelog_name_agg.getBuckets().get(0).getKeyAsString();
    187.             catalogVo.setCatelogName(catelogName);
    188.             catalogVo.setCatelogId(catelogId);
    189.             return catalogVo;
    190.         }).collect(Collectors.toList());
    191.         result.setCatalogs(catalogVos);
    193.         //4.当前所有商品涉及到的所有属性信息
    194.         ParsedNested attr_agg = aggregations.get("attr_agg");
    195.         ParsedLongTerms attr_id_agg = attr_agg.getAggregations().get("attr_id_agg");
    196.         List<SearchResult.AttrVo> attrVos = attr_id_agg.getBuckets().stream().map(bucket -> {
    197.             SearchResult.AttrVo attrVo = new SearchResult.AttrVo();
    198.             //4.1 获取属性id
    199.             long attrId = bucket.getKeyAsNumber().longValue();
    201.             //4.2 获取属性名字和值
    202.             ParsedStringTerms attr_name_agg = bucket.getAggregations().get("attr_name_agg");
    203.             ParsedStringTerms attr_value_agg = bucket.getAggregations().get("attr_value_agg");
    204.             String attrName = attr_name_agg.getBuckets().get(0).getKeyAsString();
    205.             List<String> attrValues = attr_value_agg.getBuckets().stream()
    206.                     .map(value -> value.getKeyAsString()).collect(Collectors.toList());
    208.             attrVo.setAttrId(attrId);
    209.             attrVo.setAttrName(attrName);
    210.             attrVo.setAttrValue(attrValues);
    211.             return attrVo;
    212.         }).collect(Collectors.toList());
    213.         result.setAttrs(attrVos);
    215.         return result;
    216.     }
    217. }