前陣子和張三豐聊天提到了es。這次正好有機(jī)會(huì)學(xué)習(xí)并使用

首先引入依賴

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

配置文件

spring.data.elasticsearch.local=true
spring.data.elasticsearch.repositories.enabled=true
spring.data.elasticsearch.cluster-name=yourname
spring.data.elasticsearch.cluster-nodes=127.0.0.1:9300

然后 創(chuàng)建接口并繼承ElasticsearchRepository

idea 類(lèi)繼承 ElasticsearchRepository

package com.school.service;

import com.school.model.Idea;
import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;
import org.springframework.stereotype.Component;

@Component
public interface IdeaRepository extends ElasticsearchRepository {
}

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

    @Autowired
    private IdeaRepository ideaRepository;      //esjap類(lèi)

    @Autowired
    private ElasticsearchTemplate elasticsearchTemplate;    //es工具

使用save方法把數(shù)據(jù)保存到es中
業(yè)務(wù)代碼忽略... 保存就完事了

public void save(Long ideaId) {
        // 插入數(shù)據(jù)到es中
        Idea idea = this.selectById(ideaId);
        idea.setId(ideaId);
        ideaRepository.save(idea);
    }

全文檢索并高亮數(shù)據(jù)

這里注意分頁(yè)的頁(yè)數(shù)是從0開(kāi)始... 搞得我以為沒(méi)查到數(shù)據(jù)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類(lèi)

    @Autowired
    private ElasticsearchTemplate elasticsearchTemplate;    //es工具


/**
     * 從es檢索數(shù)據(jù)
     *
     * @param content  搜索關(guān)鍵字
     * @param pageNum  頁(yè)
     * @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;
    }

其他基礎(chǔ)接口直接使用 ideaRepository.

高亮寫(xiě)法借鑒代碼地址點(diǎn)我
其他方式查詢寫(xiě)法地址點(diǎn)我