SpringBoot集成es以及ik分词器实现基本的相关性搜索

最新推荐文章于 2024-05-09 10:29:09 发布

Dpaper

最新推荐文章于 2024-05-09 10:29:09 发布

阅读量605

点赞数 2

文章标签： elasticsearch 大数据搜索引擎 spring boot java

本文链接：https://blog.csdn.net/weixin_59441154/article/details/132217613

版权

前言

最近正在学习elasticsearch，这篇文章我们就用es来实现一个基本的搜索功能，大家感兴趣的就点个赞吧😀

实现功能

实现的功能都是一些es的基础操作，如布尔查询、高亮、分页、分词查询等。

代码

1、es的配置类ElasticSearchConfig

package com.xueyu.search.config;

import lombok.Getter;
import lombok.Setter;
import org.apache.http.HttpHost;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Getter
@Setter
@Configuration
@ConfigurationProperties(prefix = "elasticsearch")
public class ElasticSearchConfig {
    private String host;
    private int port;

    @Bean
    public RestHighLevelClient client(){
        System.out.println(host);
        System.out.println(port);
        return new RestHighLevelClient(RestClient.builder(
                new HttpHost(
                        host,
                        port,
                        "http"
                )
        ));
    }
}

2、查询条件构造器

//设置查询请求
SearchRequest searchRequest = new SearchRequest("jmu_post_info");
//查询条件构造
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
//bool查询构建
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();

3、分词后再查询

//分词之后再进行查询
QueryStringQueryBuilder queryStringQueryBuilder = QueryBuilders.queryStringQuery(userSearchVO.getSearchWords()).field("title").field("content").defaultOperator(Operator.OR);
boolQueryBuilder.must(queryStringQueryBuilder);

4、分页查询

//分页
searchSourceBuilder.from(userSearchVO.getPage());
searchSourceBuilder.size(userSearchVO.getPageSize());

5、查询结果排序

//按照发布时间倒序查询
searchSourceBuilder.sort("createTime", SortOrder.DESC);

6、高亮条件设置

//设置高亮
HighlightBuilder highlightBuilder = new HighlightBuilder();
highlightBuilder.field("title");
highlightBuilder.preTags("<font style='color: red; font-size: inherit;'>");
highlightBuilder.postTags("</font>");

7、执行查询

//执行查询
searchSourceBuilder.highlighter(highlightBuilder);
searchSourceBuilder.query(boolQueryBuilder);
searchRequest.source(searchSourceBuilder);
SearchResponse searchResponse = null;
try {
    searchResponse = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT);
} catch (IOException e) {
    throw new RuntimeException(e);
}

8、查询结果封装和解析（高亮结果处理）

//查询结果解析
SearchHit[] hits = searchResponse.getHits().getHits();
//查询结果封装
List<Map<String,String>> list = new ArrayList<>();
for (SearchHit hit : hits) {
    String json = hit.getSourceAsString();
    Map<String,String> map = JSON.parseObject(json, Map.class);
    //处理高亮
    if(hit.getHighlightFields() != null && hit.getHighlightFields().size() > 0){
        Text[] titles = hit.getHighlightFields().get("title").getFragments();
        String title = StringUtils.join(titles);
        //高亮标题
        map.put("h_title",title);
    }else {
        //原始标题
        map.put("h_title",map.get("title"));
    }
    list.add(map);
}
return list;

响应结果解析对应es的部分