ElasticSearch scroll查询 api

1、scroll深度搜索，查询符合条件的所有数据。如果不是scroll深度搜索默认之后返回20条数据，如果指定分页就返回分页的条数。

package com.example.demo;


import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.transport.client.PreBuiltTransportClient;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class TestDemo {


    @Autowired
    PreBuiltTransportClient preBuiltTransportClient;

    /**
     *  查询所有数据
     * @param indices
     * @param type
     * @param queryBuilder
     * @param pageSize
     * @return
     */
    public List<Map<String,Object>> searchAllData(String indices, String type, QueryBuilder queryBuilder, Integer pageSize){
        List result = new ArrayList<>();

        List tmpList = new ArrayList<>();

        SearchResponse scrollResp = preBuiltTransportClient.prepareSearch(indices)
                .setTypes(type)
                .setScroll(new TimeValue(60000))
                .setQuery(queryBuilder)
                .setSize(pageSize).get();

        do {
            for (SearchHit hit : scrollResp.getHits().getHits()) {
                tmpList.add(hit.getSourceAsMap());
            }

            result.addAll(tmpList);

            tmpList.clear();

            scrollResp = preBuiltTransportClient.prepareSearchScroll(scrollResp.getScrollId()).setScroll(new TimeValue(60000)).execute().actionGet();

        } while (scrollResp.getHits().getHits().length != 0);

        return result;
    }


    /**
     *  查询职位是经理而且工资1万元以上的员工，然后计算他们的年终奖金
     * @param index
     * @param type
     */
    public void calculateBonus(String index,String type){
        try{

            Integer pageSize = 100;

            BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();

            //查询条件1: 工资是1万元以上
            boolQueryBuilder.must(QueryBuilders.rangeQuery("salary").gt(10000));

            //查询条件2：职位是经理级别
            boolQueryBuilder.must(QueryBuilders.matchPhraseQuery("level","经理"));

            List<Map<String, Object>> dataList = this.searchAllData(index, type, boolQueryBuilder, pageSize);

            if(dataList != null){
                for(Map<String, Object> data : dataList){
                    //业务逻辑处理，例如：计算年终奖金
                }
            }

        }catch (Exception e){
            e.printStackTrace();
        }
    }

}

来源：oschina

链接：https://my.oschina.net/u/4311641/blog/3233918

标签

QueryBuilder

java

ElasticSearch