elasticsearchTemplate 根据id主键进行操作

短命女 2022-12-28 09:04 194阅读 0赞

### 1. 主要代码 ###

#### 1.1 查询 ####

/** * 根据ID批量查询数据 * * @param index 数据所在索引 * @param ids id清单 * @author: leiming5 */
        @Override
        public List<ProcessEsVo> queryBatch(String index, List<String> ids) { 
    
            Client client = elasticsearchTemplate.getClient();
    
            //新建查询
            SearchRequestBuilder searchRequestBuilder = client.prepareSearch(index);
            TermsQueryBuilder termsQuery = QueryBuilders.termsQuery("_id", ids);
            searchRequestBuilder.setQuery(termsQuery);
            SearchResponse response = searchRequestBuilder.get();
    
            // 返回搜索结果
            SearchHits searchHits = response.getHits();
            return getProcessHitList(searchHits);
        }
    
    	/** * 分页查询 */
        public List<QesDocumentVo> getDocumentByIdList(List<String> docIdList) { 
            List<QesDocumentVo> qesDocumentVoList = Lists.newArrayList();
            if (CollectionUtils.isEmpty(docIdList)) { 
                return qesDocumentVoList;
            }
            SearchRequestBuilder searchRequestBuilder = initBaseRequestBuilder();
            searchRequestBuilder.setFrom(0).setSize(1000);
            searchRequestBuilder.setQuery(QueryBuilders.termsQuery("_id", docIdList));
            SearchResponse response = searchRequestBuilder.get();
            // 返回搜索结果
            SearchHits hits = response.getHits();
            if (hits.getTotalHits() <= 0) { 
                return qesDocumentVoList;
            }
            return getHitList(hits);
        }

#### 1.2 删除 ####

/** * 通过主数据id，删除es中的数据 * * @param index 主数据索引 * @param id 主数据id * @author: leiming5 */
        @Override
        public void deleteByIndexAndSubjectId(String index, Long id) { 
    
            List<String> list = Arrays.asList("question", "quality_college", "definition", "bms");
            if (!list.contains(index)) { 
                return;
            }
    
            EsEnum esEnum = EsEnum.getByIndexName(index);
    
            elasticsearchTemplate.delete(index, esEnum.getType(), String.valueOf(id));
        }
        /** * 根据ID批量删除指定索引数据 * * @param index 删除数据所在索引 * @param docType 删除数据文档类型 * @param ids 待删除id清单 */
        @Override
        public void batchDelete(String index, String docType, List<String> ids) { 
            if (CollectionUtils.isEmpty(ids)) { 
                return;
            }
            Client client = elasticsearchTemplate.getClient();
            BulkRequestBuilder bulkRequestBuilder = elasticsearchTemplate.getClient().prepareBulk();
            for (String docId : ids) { 
                bulkRequestBuilder.add(client.prepareDelete(index, docType, docId));
            }
            BulkResponse bulkResponse = bulkRequestBuilder.execute().actionGet();
            if (bulkResponse.hasFailures()) { 
                log.error(bulkResponse.buildFailureMessage());
            }
        }
    
        /** * 按指定Query条件删除ES数据。 * * @param index 删除数据所在索引 * @param docType 删除数据文档类型 * @param field 字段名称 * @param value 匹配的值 */
        @Override
        public void deleteByQuery(String index, String docType, String field, List<String> value) { 
            DeleteQuery deleteQuery = new DeleteQuery();
            deleteQuery.setIndex(index);
            deleteQuery.setType(docType);
            deleteQuery.setQuery(QueryBuilders.termsQuery(field, value));
    
            elasticsearchTemplate.delete(deleteQuery);
        }

#### 1.3 修改 ####

@SneakyThrows
        @Override
        @Transactional(rollbackFor = Exception.class)
        public Integer updateFieldByLLCenter(Long id, Long viewCount) { 
            //根据 任务id分组进行求和
            Client client = elasticsearchTemplate.getClient();
            //对应的es的索引库
            String index = "ll_center";
            String type = "center";
            JSONObject jsonObject = new JSONObject();
            jsonObject.put("viewcount", viewCount + 1L);
            UpdateRequest updateRequest = new UpdateRequest();
            updateRequest.index(index);
            updateRequest.type(type);
            updateRequest.id(id.toString());
            updateRequest.doc(jsonBuilder().startObject().field("viewcount", viewCount + 1L).endObject());
            RestStatus status = client.update(updateRequest).actionGet().status();
            return status.getStatus();
        }

### 2 相关代码 ###

private List<ProcessEsVo> getProcessHitList(SearchHits hits) { 
            List<ProcessEsVo> list = Lists.newArrayList();
            if (hits != null || hits.getTotalHits() != 0) { 
                for (SearchHit searchHit : hits) { 
                    list.add(convertSearchHit2Vo(searchHit, ProcessEsVo.class));
                }
            }
            return list;
        }
    
        private <T extends BaseESVo> T convertSearchHit2Vo(SearchHit searchHit, Class<T> clazz) { 
            JSONObject jsonObject = JSON.parseObject(searchHit.getSourceAsString());
            jsonObject.put("id", searchHit.getId());
            if (MapUtils.isNotEmpty(searchHit.getHighlightFields())) { 
                searchHit.getHighlightFields().forEach((k, v) -> { 
                    String hight = "";
                    for (Text text : v.getFragments()) hight += text.string();
                    setJsonProperty(jsonObject, v.getName(), hight);
                });
            }
            return JSONObject.toJavaObject(jsonObject, clazz);
        }