spring cloud 集成elasticsearch 实现findByIds查询

128 阅读 0 评论 85 点赞

我是靠谱客的博主苹果荔枝，这篇文章主要介绍spring cloud 集成elasticsearch 实现findByIds查询，现在分享给大家，希望可以做个参考。

我们使用elasticsearch常使用它的分词查询，但是因为业务需要也需要使用多id查询，elasticsearch提供了MultiGetRequest实现多ids,多索引查询：

复制代码

/**
     * 根据多个索引查询
     */
    public <T> List<T> findByIds(List<String> keys, Class<T> clazz) {
        RestHighLevelClient client=this.restHighLevelClient();
        try {
            if(null == keys || keys.size() == 0){
                return  null;
            }
            // 通过MultiGetRequest.Item对象设置查询参数
            MultiGetRequest request = new MultiGetRequest();
            String[] ids = new String[keys.size()];
            for(int i = 0;i< keys.size();i++){
                ids[i] = keys.get(i);
                request.add( "索引名", keys.get(i));
            }
            List<T>  productList = Lists.newArrayList();
            MultiGetResponse response = client.mget(request, RequestOptions.DEFAULT);
            log.info("ES-findById:id[{}],result:{}", JSON.toJSONString(keys),response.getResponses());
            if(!response.isFragment()){
                for(MultiGetItemResponse multiGetItemResponse :response.getResponses()){
                    GetResponse firstGet = multiGetItemResponse.getResponse();
                    if (firstGet.isExists()) {
                        String value =firstGet.getSourceAsString();
                        productList.add(JSONObject.parseObject(value,clazz)) ;
                    }
                }
            }
            return productList;
        } catch (Exception e) {
            throw new BizException(" findByIndexIds 搜索引擎异常:"+e.getMessage());
        }finally {
            try {
                client.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
/**
     * 根据多个索引查询
     */
    public <T> List<T> findByIds(List<String> keys, Class<T> clazz) {
        RestHighLevelClient client=this.restHighLevelClient();
        try {
            if(null == keys || keys.size() == 0){
                return  null;
            }
            // 通过MultiGetRequest.Item对象设置查询参数
            MultiGetRequest request = new MultiGetRequest();
            String[] ids = new String[keys.size()];
            for(int i = 0;i< keys.size();i++){
                ids[i] = keys.get(i);
                request.add( "索引名", keys.get(i));
            }
            List<T>  productList = Lists.newArrayList();
            MultiGetResponse response = client.mget(request, RequestOptions.DEFAULT);
            log.info("ES-findById:id[{}],result:{}", JSON.toJSONString(keys),response.getResponses());
            if(!response.isFragment()){
                for(MultiGetItemResponse multiGetItemResponse :response.getResponses()){
                    GetResponse firstGet = multiGetItemResponse.getResponse();
                    if (firstGet.isExists()) {
                        String value =firstGet.getSourceAsString();
                        productList.add(JSONObject.parseObject(value,clazz)) ;
                    }
                }
            }
            return productList;
        } catch (Exception e) {
            throw new BizException(" findByIndexIds 搜索引擎异常:"+e.getMessage());
        }finally {
            try {
                client.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }