springboot ElasticSearch 简单的全文检索高亮-白红宇

springboot ElasticSearch 简单的全文检索高亮

阅读量：5950 次

发布时间：2019-06-19

本文共 4124 字，大约阅读时间需要 13 分钟。

前阵子和张三丰聊天提到了es。这次正好有机会学习并使用

首先引入依赖


                
     
      org.springframework.boot
                 
     
      spring-boot-starter-data-elasticsearch

配置文件

spring.data.elasticsearch.local=truespring.data.elasticsearch.repositories.enabled=truespring.data.elasticsearch.cluster-name=yournamespring.data.elasticsearch.cluster-nodes=127.0.0.1:9300

然后创建接口并继承ElasticsearchRepository

idea 类继承 ElasticsearchRepository

package com.school.service;import com.school.model.Idea;import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;import org.springframework.stereotype.Component;@Componentpublic interface IdeaRepository extends ElasticsearchRepository
    
      {}

下一步在需要使用的service 或 Controller中引用

@Autowired    private IdeaRepository ideaRepository;      //esjap类    @Autowired    private ElasticsearchTemplate elasticsearchTemplate;    //es工具

使用save方法把数据保存到es中

业务代码忽略... 保存就完事了

public void save(Long ideaId) {        // 插入数据到es中        Idea idea = this.selectById(ideaId);        idea.setId(ideaId);        ideaRepository.save(idea);    }

全文检索并高亮数据

这里注意分页的页数是从0开始... 搞得我以为没查到数据debug了很久

import org.elasticsearch.action.search.SearchResponse;import org.elasticsearch.search.SearchHit;import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.data.domain.PageRequest;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.data.elasticsearch.core.query.SearchQuery;    @Autowired    private IdeaRepository ideaRepository;      //esjap类    @Autowired    private ElasticsearchTemplate elasticsearchTemplate;    //es工具/**     * 从es检索数据     *     * @param content  搜索关键字     * @param pageNum  页     * @param pageSzie 条     * @return     */public AggregatedPage
    
      getIdeaListBySrt(String content, Integer pageNum, Integer pageSzie) {        Pageable pageable = PageRequest.of(pageNum, pageSzie);        String preTag = "
     ";//google的色值        String postTag = "";        SearchQuery searchQuery = new NativeSearchQueryBuilder().                withQuery(matchQuery("ideaTitle", content)).                withQuery(matchQuery("ideaContent", content)).                withHighlightFields(new HighlightBuilder.Field("ideaTitle").preTags(preTag).postTags(postTag),                        new HighlightBuilder.Field("ideaContent").preTags(preTag).postTags(postTag)).build();        searchQuery.setPageable(pageable);                // 不需要高亮直接return ideas         // AggregatedPage
     
       ideas = elasticsearchTemplate.queryForPage(searchQuery, Idea.class);                // 高亮字段        AggregatedPage
      
        ideas = elasticsearchTemplate.queryForPage(searchQuery, Idea.class, new SearchResultMapper() {            @Override            public 
       
         AggregatedPage
        
          mapResults(SearchResponse response, Class
         
           clazz, Pageable pageable) { List
          
            chunk = new ArrayList<>(); for (SearchHit searchHit : response.getHits()) { if (response.getHits().getHits().length <= 0) { return null; } Idea idea = new Idea(); //name or memoe HighlightField ideaTitle = searchHit.getHighlightFields().get("ideaTitle"); if (ideaTitle != null) { idea.setIdeaTitle(ideaTitle.fragments()[0].toString()); } HighlightField ideaContent = searchHit.getHighlightFields().get("ideaContent"); if (ideaContent != null) { idea.setIdeaContent(ideaContent.fragments()[0].toString()); } chunk.add(idea); } if (chunk.size() > 0) { return new AggregatedPageImpl<>((List
           
            ) chunk); } return null; } }); return ideas; }