代码日常学习

public class HotelSearchTest {

 

    private RestHighLevelClient client;

 

    @Test

    void testMatchAll() throws IOException {

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备dsl

        request.source().query(QueryBuilders.matchAllQuery());

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        SearchHits hits = search.getHits();

        //获取总条数

        long value = hits.getTotalHits().value;

        System.out.println(value);

        //获取数据

        SearchHit[] hits1 = hits.getHits();

        for (SearchHit documentFields : hits1) {

            String sourceAsString = documentFields.getSourceAsString();

            HotelDoc hotelDoc = JSON.parseObject(sourceAsString, HotelDoc.class);

            System.out.println(hotelDoc);

        }

    }

    @Test

    void testMatch() throws IOException {

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备dsl

        request.source().query(QueryBuilders.matchQuery("all", "如家"));

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        //响应解析

        handleResponse(search);

    }

    /**

     * 词条查询

     *

     * @throws IOException

     */

    @Test

    void testTerm() throws IOException {

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备dsl

        request.source().query(QueryBuilders.termQuery("city", "杭州"));

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        //响应解析

        handleResponse(search);

    }

    /**

     * 范围查询

     *

     * @throws IOException

     */

    @Test

    void testRange() throws IOException {

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备dsl

        request.source().query(QueryBuilders.rangeQuery("price").gte(100).lte(500));

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        //响应解析

        handleResponse(search);

    }

    /**

     * 符合查询之精确查询

     *

     * @throws IOException

     */

    @Test

    void testBool() throws IOException {

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备boolquery

        BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();

        boolQueryBuilder.must(QueryBuilders.termQuery("city", "上海"));

        boolQueryBuilder.filter(QueryBuilders.rangeQuery("price").lte(500));

        //准备dsl

        request.source().query(boolQueryBuilder);

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        //响应解析

        handleResponse(search);

    }

    @Test

    void testPageAndSort() throws IOException {

        //页码，每页大小

        int page = 1, size = 5;

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备dsl

        request.source().query(QueryBuilders.matchAllQuery());

        //排序  sort

        request.source().sort("price", SortOrder.ASC);

        //分页from

        request.source().from((page - 1) * size).size(5);

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        //响应解析

        handleResponse(search);

    }

    /**

     * 高亮查询

     *

     * @throws IOException

     */

    @Test

    void testHighlight() throws IOException {

        //页码，每页大小

        int page = 1, size = 5;

        //准备request

        SearchRequest request = new SearchRequest("hotel");

        //准备dsl

 

        //定义查询字段

        request.source().query(QueryBuilders.matchQuery("all", "如家"));

        request.source().highlighter(new HighlightBuilder().field("name").requireFieldMatch(false));

 

        //发送请求

        SearchResponse search = client.search(request, RequestOptions.DEFAULT);

        //响应解析

        handleResponse(search);

    }

    /**

     * 抽取后的解析方法

     *

     * @param search

     */

    private void handleResponse(SearchResponse search) {