es中java代码高亮显示_ElasticSearch如何设置返回结果关键词高亮显示-Fun言

前言

类似于百度等搜索引擎，我们的搜索词都会在搜索结果中显示红色或其他高亮颜色，这就叫高亮显示，ElasticSearch同样也是支持这个功能的，下面我们就看一下如何用es实现搜索结果的高亮显示

教程

1、重写高亮显示的mapper类package cn.funyan.blog.es;

import com.alibaba.fastjson.JSONObject;

import org.elasticsearch.action.search.SearchResponse;

import org.elasticsearch.common.bytes.ByteBufferReference;

import org.elasticsearch.common.text.Text;

import org.elasticsearch.search.SearchHit;

import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;

import org.springframework.data.domain.Pageable;

import org.springframework.data.elasticsearch.core.DefaultResultMapper;

import org.springframework.data.elasticsearch.core.aggregation.AggregatedPage;

import java.nio.ByteBuffer;

import java.util.Map;

/**

*高亮显示类

**/

public class HighlightResultMapper extends DefaultResultMapper

{

@Override

public AggregatedPage mapResults(SearchResponse response, Class clazz, Pageable pageable) {

for (SearchHit hit : response.getHits()) {

Map sourceMap = hit.getSourceAsMap();

for (Map.Entry entry : hit.getHighlightFields().entrySet()) {

String key = entry.getKey();

if (sourceMap.containsKey(key)) {

Text[] fragments = entry.getValue().getFragments();

sourceMap.put(key, transTextArrayToString(fragments));

}

}

hit.sourceRef(new ByteBufferReference(ByteBuffer.wrap(JSONObject.toJSONString(sourceMap).getBytes())));

}

return super.mapResults(response, clazz, pageable);

}

private String transTextArrayToString(Text[] fragments) {

if (null == fragments) {

return "";

}

StringBuffer buffer = new StringBuffer();

for (Text fragment : fragments) {

buffer.append(fragment.string());

}

return buffer.toString();

}

}

2、编写搜索方法public List searchArtV2(Integer pageNum, String keywords) {

if(StringUtils.isEmpty(keywords)){

return new Res().failed().setMsg("搜索词不得为空");

}

//构建分页数据

Pageable pageable=PageRequest.of(Page.getEsPageNum(pageNum),Page.DEFAULT_PAGESIZE);

//构建条件查询器

QueryStringQueryBuilder boolQueryBuilder = new QueryStringQueryBuilder(keywords);

//构建条件

boolQueryBuilder.field("head")

.field("introduce");

//构建查询器

//高亮的前缀和后缀，这里可以自定义，只要可以高亮显示

String preTag = "";

String postTag = "";

NativeSearchQuery searchQuery = new NativeSearchQueryBuilder()

.withQuery(boolQueryBuilder)

.withFields("id","head","titlePhoto","pageView","createdAt","description","introduce","classifyName")

//重点！！！这里支持检索的字段高亮，显示高亮的必须是你检索的字段

.withHighlightFields(new HighlightBuilder.Field("head").preTags(preTag).postTags(postTag),new HighlightBuilder.Field("introduce").preTags(preTag).postTags(postTag))

.withPageable(pageable)

.build();

//这里直接传入我们写的高亮类

AggregatedPage articles = elasticsearchRestTemplate.queryForPage(searchQuery, Article.class, new HighlightResultMapper());

List list=articles.getContent();

log.info("用户搜索文章：{}",JSONObject.toJSON(result));

return list;

}

3、这样就可以看到我们的搜索词都加上了标签，效果如下

你还没有登录，请先使用 QQ登录 或 注册！

文章评论

发表评论