ElasticSearch学习（六）—— Java检索&设置高亮

朴灿烈づ我的快乐病毒、 2023-02-27 05:46 3阅读 0赞

*  JAVA REST API
    
     *  [https://www.elastic.co/guide/en/elasticsearch/client/java-rest/7.2/java-rest-high-document-bulk.html][https_www.elastic.co_guide_en_elasticsearch_client_java-rest_7.2_java-rest-high-document-bulk.html]
 *  ES API
    
     *  [https://www.elastic.co/guide/en/elasticsearch/reference/current/keyword.html][https_www.elastic.co_guide_en_elasticsearch_reference_current_keyword.html]

private static int port = 9200;
        private static String host = "192.168.1.105";
    
        private static RestHighLevelClient getClient() {
            RestHighLevelClient client = new RestHighLevelClient(
                    RestClient.builder(new HttpHost(host, port, "http")));
            return client;
        }
    
        public void earch(List<String> searchName,
                                      List<String> searchValue,
                                      List<Integer> searchOrAnd,
                                      Boolean isFuzzyQuery,
                                      String titleSort,
                                      String authorSort,
                                      Integer pageNum, Integer pageSize) {
    
            RestHighLevelClient client = getClient();
            // 基础设置   new SearchRequest("patent"); "patent" 为索引名称
            SearchRequest searchRequest = new SearchRequest("book");
            // 搜索源构建对象
            SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
            HighlightBuilder highlightBuilder = new HighlightBuilder();
            if (searchName.size() == 0){
                // 查全部字段
                if (searchValue.size() == 0){
                    // 查所有数据
                    searchSourceBuilder.query(QueryBuilders.matchAllQuery());
                }else {
                    // 正常所有字段模糊查询
                    String valueJ = searchValue.get(0);
    
                    BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
                    boolQueryBuilder.should(QueryBuilders.matchQuery("title", valueJ));
                    boolQueryBuilder.should(QueryBuilders.matchQuery("author", valueJ));
                    boolQueryBuilder.should(QueryBuilders.matchQuery("classification", valueJ));
                    
                    boolQueryBuilder.minimumShouldMatch(1);
                    searchSourceBuilder.query(boolQueryBuilder);
    
                    // 设置高亮高亮
                    highlightBuilder.field(new HighlightBuilder.Field("title"));
                    highlightBuilder.field(new HighlightBuilder.Field("author"));
                    highlightBuilder.field(new HighlightBuilder.Field("classification"));
                }
            }else {
                // 高级查询
                if (isFuzzyQuery){
                    // 模糊查询
                    BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
                    for (int i = 0; i < searchName.size(); i++){
                        String valueJ = searchValue.get(i);
    
                        BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
                        boolQuery.should(QueryBuilders.matchQuery(searchName.get(i), valueJ));
    
                        if (searchOrAnd.get(i).equals(0)){
                            boolQueryBuilder.should(boolQuery);
                        }else {
                            boolQueryBuilder.must(boolQuery);
                        }
                    }
                    searchSourceBuilder.query(boolQueryBuilder);
                }else {
                    // 精确匹配
                    BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
                    for (int i = 0; i < searchName.size(); i++){
                        String valueJ = searchValue.get(i);
    
                        BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
                        boolQuery.should(QueryBuilders.termQuery(searchName.get(i)+".keyword", valueJ));
    
                        if (searchOrAnd.get(i).equals(0)){
                            boolQueryBuilder.should(boolQuery);
                        }else {
                            boolQueryBuilder.must(boolQuery);
                        }
                    }
                    searchSourceBuilder.query(boolQueryBuilder);
                }
                // 高亮
                for (int i = 0; i < searchName.size(); i++){
                    HighlightBuilder.Field highlightTitle = new HighlightBuilder.Field(searchName.get(i));
                    highlightBuilder.field(highlightTitle);
                }
            }
            // 设置高亮样式
            highlightBuilder.preTags("<font color='red'>");
            highlightBuilder.postTags("</font>");
    
            searchSourceBuilder.highlighter(highlightBuilder);
    
            //意思是从第几条开始查找
            searchSourceBuilder.from(pageNum * pageSize);
            //查几条
            searchSourceBuilder.size(pageSize);
    
            searchSourceBuilder.trackTotalHits(true);
    
            // 排序
            if ("asc".equals(titleSort)){
                searchSourceBuilder.sort("title.sort", ASC);
            }else if ("desc".equals(titleSort)){
                searchSourceBuilder.sort("title.sort", DESC);
            }
    
            if ("asc".equals(authorSort)){
                searchSourceBuilder.sort("author.sort", ASC);
            }else if ("desc".equals(authorSort)){
                searchSourceBuilder.sort("author.sort", DESC);
            }
            searchRequest.source(searchSourceBuilder);
            // 发起请求，获取结果
            SearchResponse searchResponse = null;
            try {
                searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
                SearchHits hits = searchResponse.getHits();
                // 获取检索结果总数
                int count = (int)hits.getTotalHits().value;
                if (count == 0){
                    client.close();
                }else {
                    for (SearchHit hit : hits.getHits()) {
                        Map<String, Object> sourceAsMap = hit.getSourceAsMap();
                        Map<String, HighlightField> highlightFieldMap = hit.getHighlightFields();
    
                        List<Map<String, String>> list = new ArrayList<Map<String, String>>();
                        
                        // 获取高亮
                        for (Map.Entry<String, HighlightField> entry : highlightFieldMap.entrySet()){
                            Map<String, String> map = new HashMap<String, String>();
                            Text[] texts = highlightFieldMap.get(entry.getKey()).getFragments();
                            String title = "";
                            for (Text text : texts) {
                                title += text;
                            }
                            map.put(entry.getKey(), title);
                            list.add(map);
    
                        }
                        sourceAsMap.put("highlight", list);
                    }
                    client.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

[https_www.elastic.co_guide_en_elasticsearch_client_java-rest_7.2_java-rest-high-document-bulk.html]: https://www.elastic.co/guide/en/elasticsearch/client/java-rest/7.2/java-rest-high-document-bulk.html
[https_www.elastic.co_guide_en_elasticsearch_reference_current_keyword.html]: https://www.elastic.co/guide/en/elasticsearch/reference/current/keyword.html