Java实现对ES数据的新增，删除，修改，及合并

新增数据

代码：

@Autowired
private RestHighLevelClient client;
/**
* @description ES写入数据
* @author zae
* @date 2022/1/13 14:40
* @param index 索引库
* @param dataList 数据集合（size为插入数据的条数）
*/
public void insertEsData(String index,List<Map<String,Object>> dataList) {
BulkProcessor bulkProcessor = null;
try {
bulkProcessor = getBulkProcessor(client);
for(Map<String,Object> dataMap:dataList){
bulkProcessor.add(new IndexRequest(index).source(dataMap));
}
// 将数据刷新到ES中
bulkProcessor.flush();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
boolean terminatedFlag = bulkProcessor.awaitClose(150L,
TimeUnit.SECONDS);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}

private BulkProcessor getBulkProcessor(RestHighLevelClient client) {
BulkProcessor bulkProcessor = null;
try {
BulkProcessor.Listener listener = new BulkProcessor.Listener() {
@Override
public void beforeBulk(long executionId, BulkRequest request) {
logger.info("Try to insert data number : "
+ request.numberOfActions());
}

@Override
public void afterBulk(long executionId, BulkRequest request,
BulkResponse response) {
logger.info("************** Success insert data number : "
+ request.numberOfActions() + " , id: " +
executionId);
}

@Override
public void afterBulk(long executionId, BulkRequest request,
Throwable failure) {
logger.error("Bulk is unsuccess : " + failure + ", executionId: " + executionId);
}
};
BiConsumer<BulkRequest, ActionListener<BulkResponse>> bulkConsumer =
(request, bulkListener) -> client
.bulkAsync(request, RequestOptions.DEFAULT, bulkListener);
BulkProcessor.Builder builder = BulkProcessor.builder(bulkConsumer,
listener);
builder.setBulkActions(5000);//刷新条数
builder.setBulkSize(new ByteSizeValue(100L, ByteSizeUnit.MB));// 刷新大小
builder.setConcurrentRequests(10);//并发线程数
builder.setFlushInterval(TimeValue.timeValueSeconds(100L));// 时间频率
builder.setBackoffPolicy(BackoffPolicy.constantBackoff(TimeValue.timeValueSeconds(1L), 3));//重试补偿策略
bulkProcessor = builder.build();
} catch (Exception e) {
e.printStackTrace();
try {
bulkProcessor.awaitClose(100L, TimeUnit.SECONDS);
} catch (Exception e1) {
logger.error(e1.getMessage());
}
}
return bulkProcessor;
}

注意点：

• 传入数据中的List中的每个Map为一条数据，Map中的key为字段Field，value为值。

• 假如在插入数据前index库中没有定义Field及映射关系，在插入数据时会自动新增字段Field，字段的类型会默认映射为value值的类型。

• 假如新插入的数据和之前第一次插入的数据类型不一致，(相同的key但是value的类型不一样)，这样在执行代码时可能不会报错失败，但是实际上并没有插入数据成功。

删除数据

代码

/**
* @description ES数据删除
* @author zae
* @date 2022/1/13 17:14
* @param index 索引库
* @param id 数据id
*/
public void delete(String index,String id){
DeleteRequest deleteRequest = new DeleteRequest(index,id);
DeleteResponse response = null;
try {
response = client.delete(deleteRequest,RequestOptions.DEFAULT);
} catch (IOException e) {
e.printStackTrace();
}
System.out.println(response);
}

删除的话直接指定索引库和要删除的数据的id进行删除，至于数据的id怎么获取，参考下面更新数据中的代码。

更新数据

直接修改

/**
* @description ES数据更新
* @author zae
* @date 2022/1/13 16:10
* @param index 索引库
* @param key 字段名
* @param value 更新后的值
* @param id  需要修改的那条数据的id
*/
public void update(String index,String key,Object value,String id){
try {
UpdateRequest request = new UpdateRequest();
request.index(index) //索引名
.id(id)//id
.doc(
XContentFactory.jsonBuilder()
.startObject()
.field(key, value)//要修改的字段 及字段值
.endObject()
);
UpdateResponse response= client.update(request,RequestOptions.DEFAULT);
System.out.println(response.status());
} catch (Exception e) {
e.printStackTrace();
}
}

• 更新数据需要传入指定的索引库，需要修改的字段名称，修改后的value值，以及代表那条数据的唯一id

• 关于唯一id怎么获得，可以参考以下代码。

  public void selectAndUpdate(){
  int count = 0;
  //step1：根据条件获取需要修改的数据的id
  SearchHit[] searchHits = esDeal.selectIdByKey(PRODUCT_DEV_INDEX, "name");
  
  // step2:遍历更新查询出来的数据
  for (SearchHit searchHit : searchHits) {
  // 获取到单条记录的id
  String id = searchHit.getId();
  //调用更新数据的核心方法
  esDeal.update(PRODUCT_DEV_INDEX,"name","张三",id);
  count++;
  }
  System.out.println("一共更新了"+count+"条数据");
  }
  
  public SearchHit[] selectIdByKey(String index,String key){
  BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
  boolQueryBuilder.must(QueryBuilders.existsQuery(key));
  SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
  sourceBuilder.query(boolQueryBuilder)
  .trackTotalHits(true)
  .size(22);
  
  SearchRequest searchRequest = new SearchRequest()
  .indices(index)
  .source(sourceBuilder);
  try {
  SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
  SearchHit[] hits = searchResponse.getHits().getHits();
  return hits;
  } catch (IOException e) {
  e.printStackTrace();
  }
  return null;
  }

数据合并

数据合并其实也是更新的一种，关键在于怎么提取及封装需要更新的数据，根据业务场景的不同达到合并的效果，以上面的图片为例，我需要将以上数据的20004.value里面的值合并上新的数据却也保留着原有的数据。

@Test
public void selectAndUpdate(){
int count = 0;
//step1：根据条件获取需要修改的数据的id
SearchHit[] searchHits = esDeal.selectIdByKey(PRODUCT_DEV_INDEX, "20004.value");
for (SearchHit searchHit : searchHits) {
String id = searchHit.getId();
// step2:组装数据
Map<String, Object> sourceAsMap = searchHit.getSourceAsMap();
List<Map<String,Object>> mapList = (List<Map<String, Object>>) sourceAsMap.get("20004");
Map<String, Object> mapData = mapList.get(0);
// 获取之前20004.value的数据，并添加上新的数据
List value = (List) mapData.get("value");
value.add("120021");
value.add("3990993");
mapData.put("value",value);
// step3：根据id更新数据
esDeal.update(PRODUCT_DEV_INDEX,"20004",mapList,id);
count++;
}
System.out.println("一共更新了"+count+"条数据");
}

备注：

• 代码中调用的selectIdByKey()以及update()在更新数据的第一部分都有代码，直接使用就好。根据以上代码更新后的数据为：

• 在更新20004.value的值时，不需要保持和20004.value的原有数据类型一致，更新成其他类型也是可以的，这点和插入key：value的数据是有区别的。