Elasticsearch教程（七） elasticsearch Insert 插入数据（Java）

最新推荐文章于 2023-05-31 06:00:00 发布

转载最新推荐文章于 2023-05-31 06:00:00 发布 · 822 阅读

文章标签：

#大数据 #java #json

本文介绍了一种针对Elasticsearch进行批量数据插入的方法，并提供了一个简洁的封装示例，该示例支持单条和批量数据插入，适用于索引创建后的数据填充场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先我不赞成再采用一些中间件（jar包）来解决和 Elasticsearch 之间的交互，比如 Spring-data-elasticsearch.jar 系列一样，用就得依赖它。而 Elasticsearch 本身的用法就是很简洁的，也很容易，只要稍微加一下封装，即可解决。下面上我的简单封装的批量处理。

package com.sojson.core.elasticsearch.manager;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import net.sf.json.JSONObject;

import org.elasticsearch.action.bulk.BulkRequestBuilder;
import org.elasticsearch.action.bulk.BulkResponse;
import org.elasticsearch.action.index.IndexRequestBuilder;

import com.sojson.common.utils.StringUtils;
import com.sojson.core.elasticsearch.utils.ESTools;

public class InsertManager {

   /**
   * 添加数据到Elasticsearch
   * @param index       索引
   * @param type       类型
   * @param idName   Id字段名称
   * @param json       存储的JSON，可以接受Map
   * @return
   */
   public static Map save(String index, String type, String idName,JSONObject json) {
       List list = new ArrayList();
       list.add(json);
       return save(index, type, idName, list);
   }


   /**
   * 添加数据到Elasticsearch
   * @param index       索引
   * @param type       类型
   * @param idName   Id字段名称
   * @param listData 一个对象集合
   * @return
   */
   @SuppressWarnings("unchecked")
   public static Map save(String index, String type, String idName,List listData) {
       BulkRequestBuilder bulkRequest = ESTools.client.prepareBulk().setRefresh(true);
       Map resultMap = new HashMap();


       for (Object object : listData) {
           JSONObject json = JSONObject.fromObject(object);
           //没有指定idName 那就让Elasticsearch自动生成
           if(StringUtils.isBlank(idName)){
               IndexRequestBuilder lrb = ESTools.client.prepareIndex(index, type).setSource(json);
               bulkRequest.add(lrb);
           }
           else{
               String idValue = json.optString(idName);
               IndexRequestBuilder lrb = ESTools.client.prepareIndex(index, type,idValue).setSource(json);
               bulkRequest.add(lrb);
           }

       }

       BulkResponse bulkResponse = bulkRequest.execute().actionGet();
       if (bulkResponse.hasFailures()) {
           // process failures by iterating through each bulk response item
           System.out.println(bulkResponse.getItems().toString());
           resultMap.put("500", "保存ES失败!");
           return resultMap;
       }
       bulkRequest = ESTools.client.prepareBulk();
       resultMap.put("200", "保存ES成功!");
       return resultMap;
   }
}