SpringBoot下Elasticsearch增删改查相关API使用

1、创建索引：

if(!template.indexOps(ChargeOrder.class).exists()){
            log.warn("索引【charge_order】不存在，创建索引");
            String doc=new BufferedReader(new InputStreamReader(this.getClass().getClassLoader().getResourceAsStream("ChargeOrderMapping.json")))
                    .lines().collect(Collectors.joining(System.lineSeparator()));
            createIndex(doc,ChargeOrder.class);
}else{
            log.info("索引【charge_order】已经存在，无需创建");
}

2、索引创建完成后，有时根据业务需求，需要增加mapping字段：

String doc = new BufferedReader(new InputStreamReader(this.getClass().getClassLoader().getResourceAsStream("chargeOrder.json")))
                    .lines().collect(Collectors.joining(System.lineSeparator()));
if (template.indexOps(ChargeOrder.class).putMapping(Document.parse(doc)))
   System.out.println("新增索引字段成功！");

3、新增、删除数据记录：

template.save(order);
template.delete("x86xEHgBFCiNVtifACcz",ChargeOrder.class);

4、ElasticsearchRestTemplate实现修改文档：

Document document = Document.create();
        document.setId(id);
        document.put("status", 3);
UpdateQuery build = UpdateQuery.builder(id).withDocument(document).withScriptedUpsert(true).build();
UpdateResponse update = template.update(build, IndexCoordinates.of(ChargeOrder.class.getAnnotation(org.springframework.data.elasticsearch.annotations.Document.class).indexName()));

5、RestHighLevelClient实现文档修改：

XContentBuilder builder=jsonBuilder().startObject();
builder.field("status",3);
UpdateRequest updateRequest = new UpdateRequest();
updateRequest.index(ChargeOrder.class.getAnnotation(org.springframework.data.elasticsearch.annotations.Document.class).indexName());
updateRequest.id(id);
updateRequest.doc(builder.endObject());
client.update(updateRequest, RequestOptions.DEFAULT).getGetResult();

6、查询符合条件文档数量，类似mysql里的count:

BoolQueryBuilder boolQuery= QueryBuilders.boolQuery();
List<Integer> status=new ArrayList<>();
status.add(0);
status.add(1);
boolQuery.must(QueryBuilders.termQuery("userOpenid",openID))
         .must(QueryBuilders.termsQuery("status",status));
NativeSearchQueryBuilder searchQueryBuilder= new NativeSearchQueryBuilder().withQuery(boolQuery);
long count = template.count(searchQueryBuilder.build(), ChargeOrder.class);

7、分页查询文档数据，注意当数据量大于1w条时，就不友好了，虽然可以setTrackTotalHits，但是会严重影响性能，慎重使用：

BoolQueryBuilder boolQuery= QueryBuilders.boolQuery();
if(StringUtils.isNotEmpty(time_start)&&StringUtils.isNotEmpty(time_end)){
            time_start+=" 00:00:00";
            time_end+=" 23:59:59";
            RangeQueryBuilder rangeQuery = QueryBuilders.rangeQuery("createTime");
            rangeQuery.from(time_start);
            rangeQuery.to(time_end);
            boolQuery.must(rangeQuery);
}
if(StringUtils.isNotEmpty(keywords)){
                
boolQuery.must(QueryBuilders.boolQuery().should(QueryBuilders.termQuery("orderNo",keywords))
                    .should(QueryBuilders.termQuery("stake.imei",keywords))
                    .should(QueryBuilders.termQuery("stake.number",keywords)));
}
if(orderType!=null){
   boolQuery.must(QueryBuilders.termQuery("orderType",orderType));
}
NativeSearchQueryBuilder searchQueryBuilder= new NativeSearchQueryBuilder().withQuery(boolQuery)
                .withPageable(PageRequest.of(page, pageSize))
                .addAggregation(AggregationBuilders.sum("sum_amount").field("amount"))
                .withSort(SortBuilders.fieldSort("createTime").order(SortOrder.DESC));
NativeSearchQuery searchQuery =searchQueryBuilder.build();
searchQuery.setTrackTotalHits(true);
SearchHits<ChargeOrder> searchHits=template.search(searchQuery,ChargeOrder.class);
List<ChargeOrder> chargeOrderList=new ArrayList<>();
Iterator<SearchHit<ChargeOrder>> iterator=searchHits.iterator();
while (iterator.hasNext()){
    ChargeOrder order=iterator.next().getContent();
    chargeOrderList.add(order);
}
ParsedSum sum=searchHits.getAggregations().get("sum_amount");
Map<String,Object> attach=new HashMap<>();
attach.put("sumAmount",sum.getValue());
long total=searchHits.getTotalHits();
return new TableResultResponse<>(total, chargeOrderList,attach);

8、分组统计文档，并排序：

String[] param= {"stake"};
TopHitsAggregationBuilder top1 = AggregationBuilders.topHits("top").fetchSource(param, Strings.EMPTY_ARRAY).size(1);
TermsAggregationBuilder tb = AggregationBuilders.terms("stake_group").field("stake.id")
                .subAggregation(top1).size(Integer.MAX_VALUE);
tb.subAggregation(AggregationBuilders.sum("sum_amount").field("amount"));
//        FilterAggregationBuilder sumBuilderFilter=AggregationBuilders.filter("month_incom",
//                QueryBuilders.boolQuery().must(QueryBuilders.termQuery("actionType",BillsActionType.monthOrderIn.getType())))
//                .subAggregation(AggregationBuilders.sum("sum_amount").field("amount"));
//        tb.subAggregation(sumBuilderFilter);
tb.subAggregation(new BucketSortPipelineAggregationBuilder("bucket_field",Arrays.asList(
                new FieldSortBuilder("sum_amount").order(SortOrder.DESC))).from(0).size(20000));

BoolQueryBuilder boolQueryBuilder=QueryBuilders.boolQuery();
boolQueryBuilder.must(QueryBuilders.termQuery("status",3));
NativeSearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(boolQueryBuilder)
                .withFields("id").withPageable(PageRequest.of(0, 1))
                .addAggregation(tb)
                .addAggregation(AggregationBuilders.sum("sum_amount").field("amount"))
                .build();
searchQuery.setTrackTotalHits(true);
SearchHits<ChargeOrder> searchHitsList = template.search(searchQuery, ChargeOrder.class);

Terms term = searchHitsList.getAggregations().get("stake_group");
for (Terms.Bucket bk : term.getBuckets()) {
    ParsedSum sumAmount=bk.getAggregations().get("sum_amount");
    TopHits top = bk.getAggregations().get("top");
    for (org.elasticsearch.search.SearchHit searchHit : top.getHits()) {
        JSONObject stake=JSON.parseObject(JSON.toJSONString(searchHit.getSourceAsMap())).getJSONObject("stake");
        System.out.println("设备【"+stake.getString("number")+"】收益："+
                        new BigDecimal(sumAmount.getValue()).divide(new BigDecimal(100),2, RoundingMode.HALF_UP));;
    }
}
    ParsedSum sumAmount=searchHitsList.getAggregations().get("sum_amount");
    System.out.println("所有设备收益总和："+new BigDecimal(sumAmount.getValue()).divide(new BigDecimal(100),2, RoundingMode.HALF_UP));

注意：我这里用的ES版本是7.6.2；spring-boot-starter-data-elasticsearch版本是2.3.7RELEASE

暂时就想到这么些，如有其它方面的需求，大家可以留言，我抽空再补上来