十一.全文检索ElasticSearch经典入门-SpringBoot整合ElasticSearch7

Posted 2022-05-04 墨家巨子@俏如来

前言

你可以通过官网文档看到ElasticSearch7和6的变化，本文的重点在于在SpringBoot环境中使用ElasticSearch7。

软件安装

从官网下载ElasticSearch ，我这里使用的是7.9.3版本 ， 下载Kibana也是7.9.3，以及IK分词器，下载好之后把ElasticSearch进行解压：

把IK分词器解压到plugins/ik目录中，如下：

解压Kibana，目录结构如下

启动ElasticSearch ： elasticsearch-7.9.3\\bin\\elasticsearch.bat ， 然后启动Kibana ：\\kibana-7.9.3-windows-x86_64\\bin\\kibana.bat ,访问kibana: http://localhost:5601

Spring整合ES7

第一步：导入依赖,SpringBoot使用的是2.4；需要注意版本兼容7.9.3

 <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.4.0</version>
        <relativePath/>
    </parent>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-elasticsearch</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.12</version>
            <scope>compile</scope>
        </dependency>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.71</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>

第二步：启动类常规写法，yaml配置es地址

server:
  port: 1000
spring:
  elasticsearch:
    rest:
      uris:
        - http://localhost:9200

第三步：编写ES的Doc对象

@Data
//定义索引，7以上的版本去掉了type属性
@Document(indexName = "goodsorder", shards = 1, replicas = 1)
@AllArgsConstructor
@NoArgsConstructor
public class OrderDoc implements Serializable 
    @Id
    @Field(type = FieldType.Keyword)
    private String id;

    @Field(type = FieldType.Long)
    private Long orderNo;
	
    @Field(type = FieldType.Text, analyzer = "ik_smart", searchAnalyzer = "ik_max_word")
    private String title;
	
	//用来封装高亮的结果
    private Map<String, List<String>> highlights;

    public OrderDoc(String id, Long orderNo ,String title)
        this.id = id;
        this.orderNo = orderNo;
        this.title =title;
    

第四步：编写Repository接口

@Repository
//泛型是Doc对象和ID的类型
public interface OrderRepository extends ElasticsearchRepository<OrderDoc, String> 

第五步：编写Service接口,定义基本的CRUD方法

public interface OrderDocService 

    void saveAll(List<OrderDoc> orderDocs);

    OrderDoc findById(String id);

    void deleteById(String id);

    void updateById(OrderDoc orderDoc);

    PageResponse<OrderDoc> findList(OrderDoc orderDoc, Integer pageIndex, Integer pageSize);

    PageResponse<OrderDoc> findAll(Integer pageIndex, Integer pageSize);

    PageResponse<OrderDoc> findHighlight(OrderDoc orderDoc, Integer pageIndex, Integer pageSize);

下面是实现类，注入OrderRepository和ElasticsearchRestTemplate实现CRUD工作

@Service
public class OrderDocServiceImpl implements OrderDocService 

    @Autowired
    private OrderRepository orderRepository;
    @Autowired
    private ElasticsearchRestTemplate template;

    @Override
    public void saveAll(List<OrderDoc> orderDocs) 
        orderRepository.saveAll(orderDocs);
    

    @Override
    public OrderDoc findById(String id) 
        return orderRepository.findById(id).orElse(null);
    

    @Override
    public void deleteById(String id) 
        orderRepository.deleteById(id);
    

    @Override
    public void updateById(OrderDoc orderDoc) 
        orderRepository.save(orderDoc);
    

    @Override
    public PageResponse<OrderDoc> findList(OrderDoc orderDoc, Integer pageIndex, Integer pageSize) 
        CriteriaQuery criteriaQuery = new CriteriaQuery(
                new Criteria().and("title").contains(orderDoc.getTitle()),PageRequest.of(pageIndex,pageSize));

        SearchHits<OrderDoc> searchHits = template.search(criteriaQuery, OrderDoc.class);

        List<OrderDoc> orderDocList = searchHits.getSearchHits().stream()
                .map(orderDocSearchHit -> orderDocSearchHit.getContent()).collect(Collectors.toList());

        return new PageResponse<OrderDoc>(searchHits.getTotalHits() ,orderDocList);
    

    @Override
    public PageResponse<OrderDoc> findAll(Integer pageIndex, Integer pageSize) 
        Page<OrderDoc> page = orderRepository.findAll(PageRequest.of(pageIndex, pageSize));
        return new PageResponse<OrderDoc>(page.getTotalElements(),page.getContent());
    

    @Override
    public PageResponse<OrderDoc> findHighlight(OrderDoc orderDoc, Integer pageIndex, Integer pageSize) 
        //查询条件
        CriteriaQuery criteriaQuery = new CriteriaQuery(
                new Criteria().and("title").contains(orderDoc.getTitle()),
                PageRequest.of(pageIndex,pageSize));
        //高亮字段
        HighlightBuilder highlightBuilder = new HighlightBuilder();
        highlightBuilder.field("title").preTags("<font style='color:red'>").postTags("</font>");
        criteriaQuery.setHighlightQuery(new HighlightQuery(highlightBuilder));
        //搜索结果
        SearchHits<OrderDoc> searchHits = template.search(criteriaQuery, OrderDoc.class);

        List<OrderDoc> orderDocList = searchHits.get().map(searchHit -> 
            //把高亮结果添加到doc中中
            OrderDoc doc = searchHit.getContent();
            //获取高亮
            doc.setHighlights(searchHit.getHighlightFields());
            return doc;
        ).collect(Collectors.toList());

        //处理结果
        return new PageResponse<>(searchHits.getTotalHits() , orderDocList);
    

下面是需要的结果对象

@Data
@AllArgsConstructor
@NoArgsConstructor
public class PageResponse<T> 
    private Long total;
    private List<T> list;

第六步：编写单元测试

@RunWith(SpringRunner.class)
@SpringBootTest(classes = ApplicationStarter.class)
public class OrderDocServiceTest 

    @Autowired
    private OrderDocService orderDocService;

    @Test
    public void saveAll() 
        orderDocService.saveAll(Arrays.asList(
                new OrderDoc("1",100000l,"苹果电脑"),
                new OrderDoc("2",100000l,"苹果电脑"),
                new OrderDoc("3",100000l,"苹果电脑"),
                new OrderDoc("4",100000l,"苹果电脑"),
                new OrderDoc("5",100000l,"苹果电脑"),
                new OrderDoc("6",100000l,"苹果电脑"),
                new OrderDoc("7",100000l,"苹果电脑")
        ));
    


    @Test
    public void findById() 
        System.out.println(orderDocService.findById("1"));
    

    @Test
    public void deleteById() 
        orderDocService.deleteById("1");
    

    @Test
    public void updateById() 
        OrderDoc orderDoc = orderDocService.findById("1");
        orderDoc.setTitle("华为电脑");
        orderDocService.updateById(orderDoc);
    

    @Test
    public void findList() 
        OrderDoc orderDoc = new OrderDoc(null, 100l, "电脑");
        PageResponse<OrderDoc> response = orderDocService.findList(orderDoc, 0, 10);
        System.out.println(response.getTotal());
        response.getList().forEach(System.out::println);

    

    @Test
    public void findAll() 
        PageResponse<OrderDoc> response = orderDocService.findAll( 0, 10);
        System.out.println(response.getTotal());
        response.getList().forEach(System.out::println);
    

    @Test
    public void findHighlight() 
        OrderDoc orderDoc = new OrderDoc(null, 100l, "电脑",null);
        PageResponse<OrderDoc> response = orderDocService.findHighlight( orderDoc,0, 10);
        System.out.println(response.getTotal());
        response.getList().forEach(System.out::println);