RestHighLevelClient searchAfter用法

spring data elasticSearch在4.2版本以上才支持es的search after用法
所以在项目不支持的情况下 可以使用es官方提供的RestHighLevelClient实现

基于elasticsearch-rest-high-level-client-7.6.2.jar

import com.alibaba.fastjson.JSONObject;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.elasticsearch.search.sort.SortOrder;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;


public class Test01 {

    @Resource
    private RestHighLevelClient restHighLevelClient;


    
    public void demo() throws Exception{
        SearchRequest searchRequest = new SearchRequest();
        searchRequest.indices("demo_index");

        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
        searchSourceBuilder.from(0).size(1000);
        BoolQueryBuilder queryBuilder = QueryBuilders.boolQuery();
        searchSourceBuilder.query(queryBuilder).sort("create_time", SortOrder.ASC)
                .sort("code", SortOrder.ASC);

        List<Demo> totalList = new ArrayList<>();

        this.search(searchRequest,searchSourceBuilder,totalList,Demo.class,null);

        System.out.println(totalList);

    }

    private <T> void search(SearchRequest searchRequest,
                        SearchSourceBuilder searchSourceBuilder,
                        List<T> totalList,Class<T> tClass,Object[] objects) throws Exception{
        if(Objects.nonNull(objects)){
            searchSourceBuilder.searchAfter(objects);
        }
        searchRequest.source(searchSourceBuilder);
        SearchResponse searchResponse = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHit[] hits = searchResponse.getHits().getHits();
        if(Objects.nonNull(hits) && hits.length > 0){
            List<T> tList = convertToDataObject(hits,tClass);
            totalList.addAll(tList);
            search(searchRequest,searchSourceBuilder,totalList,tClass,hits[hits.length - 1].getSortValues());
        }
    }

    private <T> List<T> convertToDataObject(SearchHit[] hits,Class<T> tClass){
        return Arrays.stream(hits)
                .map(hit -> JSONObject.parseObject(hit.getSourceAsString(),tClass))
                .collect(Collectors.toList());
    }


}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值