Springboot集成ElasticSearch7.4

package com.swj.search.model;

import org.springframework.data.elasticsearch.annotations.Document;
import org.springframework.data.elasticsearch.annotations.Field;
import org.springframework.data.elasticsearch.annotations.FieldType;

import javax.persistence.Id;
import java.io.Serializable;
import java.util.Date;
import java.util.Map;

@Document(indexName = "skuinfo", type = "docs")
public class SkuInfo implements Serializable {
    //商品id，同时也是商品编号
    @Id
    @Field(index = true, store = true, type = FieldType.Keyword)
    private Long id;

    //SKU名称
    @Field(index = true, store = true, type = FieldType.Text, analyzer = "ik_smart")
    private String name;

    //商品价格，单位为：元
    @Field(index = true, store = true, type = FieldType.Double)
    private Long price;

    //库存数量
    @Field(index = true, store = true, type = FieldType.Integer)
    private Integer num;

    //商品图片
    @Field(index = false, store = true, type = FieldType.Text)
    private String image;

    //商品状态，1-正常，2-下架，3-删除
    @Field(index = true, store = true, type = FieldType.Keyword)
    private String status;

    //创建时间
    private Date createTime;

    //更新时间
    private Date updateTime;

    //是否默认
    @Field(index = true, store = true, type = FieldType.Keyword)
    private String isDefault;

    //SPUID
    @Field(index = true, store = true, type = FieldType.Long)
    private Long spuId;

    //类目ID
    @Field(index = true, store = true, type = FieldType.Long)
    private Long categoryId;

    //类目名称
    @Field(index = true, store = true, type = FieldType.Keyword)
    private String categoryName;

    //品牌名称
    @Field(index = true, store = true, type = FieldType.Keyword)
    private String brandName;

    //规格
    private String spec;

    //规格参数
    private Map<String, Object> specMap;

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public Long getPrice() {
        return price;
    }

    public void setPrice(Long price) {
        this.price = price;
    }

    public Integer getNum() {
        return num;
    }

    public void setNum(Integer num) {
        this.num = num;
    }

    public String getImage() {
        return image;
    }

    public void setImage(String image) {
        this.image = image;
    }

    public String getStatus() {
        return status;
    }

    public void setStatus(String status) {
        this.status = status;
    }

    public Date getCreateTime() {
        return createTime;
    }

    public void setCreateTime(Date createTime) {
        this.createTime = createTime;
    }

    public Date getUpdateTime() {
        return updateTime;
    }

    public void setUpdateTime(Date updateTime) {
        this.updateTime = updateTime;
    }

    public String getIsDefault() {
        return isDefault;
    }

    public void setIsDefault(String isDefault) {
        this.isDefault = isDefault;
    }

    public Long getSpuId() {
        return spuId;
    }

    public void setSpuId(Long spuId) {
        this.spuId = spuId;
    }

    public Long getCategoryId() {
        return categoryId;
    }

    public void setCategoryId(Long categoryId) {
        this.categoryId = categoryId;
    }

    public String getCategoryName() {
        return categoryName;
    }

    public void setCategoryName(String categoryName) {
        this.categoryName = categoryName;
    }

    public String getBrandName() {
        return brandName;
    }

    public void setBrandName(String brandName) {
        this.brandName = brandName;
    }

    public String getSpec() {
        return spec;
    }

    public void setSpec(String spec) {
        this.spec = spec;
    }

    public Map<String, Object> getSpecMap() {
        return specMap;
    }

    public void setSpecMap(Map<String, Object> specMap) {
        this.specMap = specMap;
    }
}

package com.swj.search.mapper;

import com.swj.search.model.SkuInfo;
import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;

public interface ESManagerMapper extends ElasticsearchRepository<SkuInfo,Long> {
}

package com.swj.search.service.impl;

import com.alibaba.fastjson.JSON;
import com.swj.crawler.fegin.SKUFegin;
import com.swj.crawler.model.TbSku;
import com.swj.search.mapper.ESManagerMapper;
import com.swj.search.model.SkuInfo;
import com.swj.search.service.ESManagerService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;

@Service
public class ESManagerServiceImpl implements ESManagerService {

  @Autowired private ElasticsearchTemplate elasticsearchTemplate;

  @Autowired private SKUFegin skuFegin;

  @Autowired private ESManagerMapper esManagerMapper;

  // 创建索引库结构
  @Override
  public void createMappingAndIndex() {
    // 创建索引
    elasticsearchTemplate.createIndex(SkuInfo.class);
    // 创建映射
    elasticsearchTemplate.putMapping(SkuInfo.class);
  }

  @Override
  public void importAll() {
    // 查询sku集合
    List<TbSku> tbSkus = skuFegin.selectAll();
    String jsonTbSkus = JSON.toJSONString(tbSkus);
    List<SkuInfo> skuInfos = JSON.parseArray(jsonTbSkus, SkuInfo.class);
    for (SkuInfo skuInfo : skuInfos) {
      Map spaceMap = JSON.parseObject(skuInfo.getSpec(), Map.class);
      skuInfo.setSpecMap(spaceMap);
    }
    // 导入索引库
    esManagerMapper.saveAll(skuInfos);
  }

  @Override
  public void importDataBySpuId(String spuId) {}

  @Override
  public void delDataBySpuId(String spuId) {}
}

package com.swj.search.service.impl;

import com.alibaba.fastjson.JSON;
import com.swj.search.model.SkuInfo;
import com.swj.search.service.SearchService;
import org.apache.commons.lang.StringUtils;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.Operator;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Pageable;
import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
import org.springframework.data.elasticsearch.core.SearchResultMapper;
import org.springframework.data.elasticsearch.core.aggregation.AggregatedPage;
import org.springframework.data.elasticsearch.core.aggregation.impl.AggregatedPageImpl;
import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
public class SearchServiceImpl implements SearchService {
  @Autowired private ElasticsearchTemplate elasticsearchTemplate;

  @Override
  public Map search(Map<String, String> searchMap) {
    Map<String, Object> resultMap = new HashMap<>();
    if (searchMap != null) {
      // 构建查询条件
      NativeSearchQueryBuilder nativeSearchQueryBuilder = new NativeSearchQueryBuilder();
      BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();

      // 按照关键字查询
      if (StringUtils.isNotEmpty(searchMap.get("keywords"))) {
        boolQuery.must(
            QueryBuilders.matchQuery("name", searchMap.get("keywords")).operator(Operator.AND));
      }

      // 按照品牌进行过滤查询
      if (StringUtils.isNotEmpty(searchMap.get("brand"))) {
        boolQuery.filter(QueryBuilders.termQuery("brandName", searchMap.get("brand")));
      }

      // 范围查询
      if (StringUtils.isNotEmpty(searchMap.get("price"))) {
        String[] prices = searchMap.get("price").split("-");
        if (prices.length == 2) {
          boolQuery.filter(QueryBuilders.rangeQuery("price").lte(prices[0]));
        }
        boolQuery.filter(QueryBuilders.rangeQuery("price").gte(prices[1]));
      }

      nativeSearchQueryBuilder.withQuery(boolQuery);

      // 开启查询
      /** 第一个参数: 条件构建对象 第二个参数: 查询操作实体类 第三个参数: 查询结果操作对象 */
      // 封装查询结果
      AggregatedPage<SkuInfo> aggregatedPage =
          elasticsearchTemplate.queryForPage(
              nativeSearchQueryBuilder.build(),
              SkuInfo.class,
              new SearchResultMapper() {
                @Override
                public <T> AggregatedPage<T> mapResults(
                    SearchResponse searchResponse, Class<T> aClass, Pageable pageable) {

                  List list = new ArrayList();
                  // 获取查询结果集
                  SearchHits searchHits = searchResponse.getHits();
                  for (SearchHit searchHit : searchHits) {
                    if (searchHit != null) {
                      SkuInfo skuInfo = JSON.parseObject(searchHit.toString(), SkuInfo.class);
                      Map<String, HighlightField> highlightFields = searchHit.getHighlightFields();
                      if (highlightFields != null && highlightFields.size() > 0) {
                        // 替换数据
                        skuInfo.setName(highlightFields.get("name").getFragments()[0].toString());
                      }

                      list.add((T) skuInfo);
                    }
                  }

                  return new AggregatedPageImpl<T>(
                      list, pageable, searchHits.getTotalHits(), searchResponse.getAggregations());
                }

                @Nullable
                @Override
                public <T> T mapSearchHit(SearchHit searchHit, Class<T> aClass) {
                  return null;
                }
              });

      resultMap.put("rows", aggregatedPage.getContent());
    }

    return null;
  }
}