elasticsearch 根據主鍵_id更新部分字段

package com.better517na.ebookingbusiservice.helper;      

import com.alibaba.fastjson.JSON;      

import com.alibaba.fastjson.JSONObject;      

import com.better517na.component.security.utils.GsonUtil;      

import com.better517na.ebookingbusiservice.model.contracthotel.ContractHotelHitVo;      

import com.better517na.ebookingbusiservice.util.LogUtils;      

import com.better517na.component.security.utils.StringUtil;      

import com.google.common.reflect.TypeToken;      

import org.elasticsearch.action.bulk.BulkRequest;      

import org.elasticsearch.action.bulk.BulkResponse;      

import org.elasticsearch.action.index.IndexRequest;      

import org.elasticsearch.action.search.SearchRequest;      

import org.elasticsearch.action.search.SearchResponse;      

import org.elasticsearch.action.update.UpdateRequest;      

import org.elasticsearch.action.update.UpdateResponse;      

import org.elasticsearch.client.RequestOptions;      

import org.elasticsearch.client.RestHighLevelClient;      

import org.elasticsearch.common.unit.TimeValue;      

import org.elasticsearch.index.query.BoolQueryBuilder;      

import org.elasticsearch.index.query.Operator;      

import org.elasticsearch.index.query.QueryBuilders;      

import org.elasticsearch.index.reindex.UpdateByQueryRequest;      

import org.elasticsearch.rest.RestStatus;      

import org.elasticsearch.search.builder.SearchSourceBuilder;      

import org.springframework.stereotype.Component;      

import javax.annotation.Resource;      

import java.io.IOException;      

import java.util.ArrayList;      

import java.util.Arrays;      

import java.util.List;      

import java.util.Map;      

import java.util.concurrent.TimeUnit;      

import java.util.stream.Collectors;      

import static org.elasticsearch.common.xcontent.XContentFactory.jsonBuilder;      

/**      

* @author lulin      

* @project: ebookingbusiservice      

* @description EsHelper      

* @date 2021/09/24 11:17      

**/      

@Component      

public class EsHelper {      

private static RestHighLevelClient restClient;      

@Resource(name = "contractHotelRestClient")      

public void setProductRestClient(RestHighLevelClient restClient) {      

EsHelper.restClient = restClient;      

}      

/**      

* 查詢      

*      

* @param contractNo      

* @param hotelId      

* @return      

* @throws IOException      

*/      

public static List<ContractHotelHitVo> queryContractHotelList(String contractNo, String hotelId) throws IOException {      

if (StringUtil.isEmptyOrNull(contractNo) || StringUtil.isEmptyOrNull(hotelId)) {      

return new ArrayList<>();      

}      

int from = 0;      

int size = 200;      

int totalHits = 0;      

List<ContractHotelHitVo> resp = new ArrayList<>();      

String index = "contract_hotel";      

if (StringUtil.isEmptyOrNull(index)) {      

return new ArrayList<>();      

}      

SearchRequest request = new SearchRequest(index);      

SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();      

sourceBuilder.from(from);      

sourceBuilder.size(size);      

sourceBuilder.timeout(new TimeValue(60, TimeUnit.SECONDS));      

BoolQueryBuilder boolBuilder = QueryBuilders.boolQuery();      

boolBuilder.must(QueryBuilders.termQuery("isDelete", 0));      

boolBuilder.must(QueryBuilders.matchQuery("contactno", contractNo).operator(Operator.AND));      

boolBuilder.must(QueryBuilders.matchQuery("hotelIDs", hotelId));      

sourceBuilder.query(boolBuilder);      

request.source(sourceBuilder);      

SearchResponse search = restClient.search(request, RequestOptions.DEFAULT);      

if (search.status() != RestStatus.OK || search.getHits().getTotalHits().value <= 0) {      

return new ArrayList<>();      

}      

totalHits = (int) search.getHits().getTotalHits().value;      

List<ContractHotelHitVo> collect = Arrays.stream(search.getHits().getHits()).map(p -> {      

ContractHotelHitVo contractHotelHitVo = JSON.parseObject(JSON.toJSONString(p.getSourceAsMap()), ContractHotelHitVo.class);      

contractHotelHitVo.setEsId(p.getId());      

return contractHotelHitVo;      

}).collect(Collectors.toList());      

resp.addAll(collect);      

from += size;      

while (from < totalHits) {      

sourceBuilder.from(from);      

SearchResponse response1 = restClient.search(request, RequestOptions.DEFAULT);      

if (response1 == null || response1.status() != RestStatus.OK || response1.getHits().getTotalHits().value < 0) {      

return resp;      

}      

List<ContractHotelHitVo> resp1 = Arrays.stream(response1.getHits().getHits()).map(p -> {      

ContractHotelHitVo contractHotelHitVo = JSON.parseObject(JSON.toJSONString(p.getSourceAsMap()), ContractHotelHitVo.class);      

contractHotelHitVo.setEsId(p.getId());      

return contractHotelHitVo;      

}).collect(Collectors.toList());      

resp.addAll(resp1);      

from += size;      

}      

return resp;      

}      

/**      

* 批量更新酒店産品      

*      

* @param      

* @return      

*/      

public static boolean updateHotelProdects(ContractHotelHitVo contractHotelHitVo) {      

if (contractHotelHitVo == null) {      

return true;      

}      

try {      

String index = "contract_hotel";      

UpdateByQueryRequest updateByQueryRequest = new UpdateByQueryRequest("_all");      

UpdateRequest updateRequest = new UpdateRequest(index, contractHotelHitVo.getEsId()).doc(jsonBuilder().startObject().field("data", contractHotelHitVo.getData())      

.field("modifyStaff", contractHotelHitVo.getModifyStaff())      

.field("modifyStaffName", contractHotelHitVo.getModifyStaffName())      

.field("modifyTime", contractHotelHitVo.getModifyTime())      

.endObject());      

UpdateResponse updateResponse = restClient.update(updateRequest, RequestOptions.DEFAULT);      

if (updateResponse.getResult().toString().equals("UPDATED") || updateResponse.getResult().toString().equals("NOOP")) {      

return true;      

}      

} catch (Exception e) {      

e.printStackTrace();      

LogUtils.writeExceptionLog(e, "updateHotelPriceInfos");      

return false;      

}      

return false;      

}      

}