ES实现高亮查询 es高亮搜索java

转载

mob64ca141a683a 2024-06-11 19:48:10

文章标签 ES实现高亮查询 elasticsearch spring 字段 文章分类 架构后端开发

springboot的整合spring-data该怎么整合网上这些东西太多了，大家可以百度一下。而且在文章最下面已经提供了该博客的所有源码。希望大家能够follow and star me。
我本来按照下面的方法进行查找，返回的内容就是FilmEntity实体bean映射的内容。如果我要是高亮查询内容，按照@query注解中添加highlight查询，name压根映射不上，真是痛苦。网上搜了一大堆千篇一律，都是我下面的那个使用spring-data的方法。不能说spring-data的crud不好，但是在向es导入数据的时候，spring-data的crud还是非常好用的。
然而在一些复杂的查询时候，spring-data封装的crud就不好用。我写了一个方法专门针对复杂类型的查询，我测试了高亮和多字段查询，封装返回实体bean的集合。
这样拿着spring-data的crud，esclient混合使用可以更好的达到特定的业务目的。

public interface FilmDao extends ElasticsearchRepository<FilmEntity, Long>{


    @Query("{\"bool\": {\"must\": [{\"match\": {\"name\": \"?0\"}}]}},\"highlight\": {\n" +
            "\"fields\" : {\n" +
            "\"name\" : {}\n" +
            "}\n" +
            "}")
    List<Object> findByAuthorsNameUsingCustomQuery(String name);

看了很多文档，总结写了一个config以注解的形式，就像mongotemplate那样使用es，效果很好。
大概意思就是在项目启动后生成一个client，然后大家都使用这一个client对elasticsearch进行操作。

import org.elasticsearch.client.Client;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.transport.InetSocketTransportAddress;
import org.elasticsearch.transport.client.PreBuiltTransportClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.elasticsearch.repository.config.EnableElasticsearchRepositories;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.net.InetAddress;

/**
 * <p>description: 创建TransportClient的类，esTemplate方法避免TransportClient每次使用创建和释放</p>
 *
 * @author chenrui
 * @since 2018-08-28
 */
@Configuration
@EnableElasticsearchRepositories(basePackages = "com.marsdl.dao.es")
@Component
public class EsConfig {

    @Value("${spring.data.elasticsearch.cluster-name}")
    private String clusterName;

    private Client esClient;

    @Bean
    public Client client() {
        TransportClient client = null;
        try {
            final Settings elasticsearchSettings = Settings.builder()
                    .put("client.transport.sniff", true)
                    .put("cluster.name", clusterName).build();
            client = new PreBuiltTransportClient(elasticsearchSettings);
            client.addTransportAddress(new InetSocketTransportAddress(InetAddress.getByName("127.0.0.1"), 9300));
        } catch (Exception e) {
            e.printStackTrace();
        }
        return client;
    }

    //避免TransportClient每次使用创建和释放
    public  Client esTemplate() {
        if( StringUtils.isEmpty(esClient) || StringUtils.isEmpty(esClient.admin())) {
            esClient = client();
            return esClient;
        }
        return esClient;
    }

}

注意：java代码中es调用的端口是9300，千万别搞错了，不然又出现扣脚的问题。
elasticsearch在application.yml配置文件是这样子的，在上一篇博客中提到，在java变量中获取yml内容。

spring:
  data:
    elasticsearch:
      cluster-name: chenrui
      cluster-nodes: localhost:9300
      repositories:
        #仓库中存储数据
        enabled: true

下面是service使用esclient封装多字段查询，也可以放在dao层。

@Autowired
    EsConfig esConfig;

   /**
     * <p>高亮查询内容, query的值查询两个字段name, director。当然了你可以配置查询更多个字段或者你可以改成你所需查询的字段</p>
     * @param query
     * @return List<FilmEntity> list
     */
    public List<FilmEntity> search(String query) {
        Client client = esConfig.esTemplate();
        HighlightBuilder highlightBuilder = new HighlightBuilder();
        //高亮显示规则
        highlightBuilder.preTags("<span style='color:red'>");
        highlightBuilder.postTags("</span>");
        //指定高亮字段
        highlightBuilder.field("name");
        highlightBuilder.field("director");
        //添加查询的字段内容
        String [] fileds = {"name", "director"};
        QueryBuilder matchQuery = QueryBuilders.multiMatchQuery(query, fileds);
        //搜索数据
        SearchResponse response = client.prepareSearch("chenrui")
                .setQuery(matchQuery)
                .highlighter(highlightBuilder)
                .execute().actionGet();

        SearchHits searchHits = response.getHits();
        System.out.println("记录数-->"+searchHits.getTotalHits());

        //List<String> list = new ArrayList<>();
        List<FilmEntity> list = new ArrayList<>();

        for(SearchHit hit : searchHits) {
            FilmEntity entity = new FilmEntity();
            Map<String, Object> entityMap = hit.getSourceAsMap();
            //高亮字段
            if(!StringUtils.isEmpty(hit.getHighlightFields().get("name"))) {
                Text[] text = hit.getHighlightFields().get("name").getFragments();
                entity.setName(text[0].toString());
                entity.setDirector(String.valueOf(entityMap.get("director")));
            }
            if(!StringUtils.isEmpty(hit.getHighlightFields().get("director"))) {
                Text[] text = hit.getHighlightFields().get("director").getFragments();
                entity.setDirector(text[0].toString());
                entity.setName(String.valueOf(entityMap.get("name")));
            }

            //map to object
            if(!CollectionUtils.isEmpty(entityMap)) {
                if(!StringUtils.isEmpty(entityMap.get("id"))) {
                    entity.setId(Long.valueOf(String.valueOf(entityMap.get("id"))));
                }
                if(!StringUtils.isEmpty(entityMap.get("language"))) {
                    entity.setLanguage(String.valueOf(entityMap.get("language")));
                }
            }
            list.add(entity);
        }
        return list;
    }