JavaRestClient操作Elasticsearch批量新增文档

比眉伴天荒 2022-12-23 14:29 189阅读 0赞

# 导包 #

导包可以根据 [文档][Link 1] 里导入依赖  
这里就不写依赖了

# 批量新增文档 #

import com.google.gson.Gson;
    import com.leyou.pojo.Item;
    import org.apache.http.HttpHost;
    import org.elasticsearch.action.bulk.BulkRequest;
    import org.elasticsearch.action.bulk.BulkResponse;
    import org.elasticsearch.action.delete.DeleteRequest;
    import org.elasticsearch.action.delete.DeleteResponse;
    import org.elasticsearch.action.get.GetRequest;
    import org.elasticsearch.action.get.GetResponse;
    import org.elasticsearch.action.index.IndexRequest;
    import org.elasticsearch.action.index.IndexResponse;
    import org.elasticsearch.client.RequestOptions;
    import org.elasticsearch.client.RestClient;
    import org.elasticsearch.client.RestHighLevelClient;
    import org.elasticsearch.common.xcontent.XContentType;
    import org.junit.After;
    import org.junit.Before;
    import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.boot.test.context.SpringBootTest;
    import org.springframework.test.context.junit4.SpringRunner;
    import java.io.IOException;
    import java.util.ArrayList;
    import java.util.HashSet;
    import java.util.List;
    
    @RunWith(SpringRunner.class)
    @SpringBootTest
    public class Esdemo2 { 
    
        private RestHighLevelClient client;
    
        private Gson gson = new Gson();
        /** * 先执行 */
        @Before
        public void init (){ 
            // 初始化HighLevel客户端
            client = new RestHighLevelClient(
                    RestClient.builder(
                            HttpHost.create("http://127.0.0.1:9201"),
                            HttpHost.create("http://127.0.0.1:9202"),
                            HttpHost.create("http://127.0.0.1:9203")
                    )
            );
        }
    
        /** * 批量新增 */
        @Test
        public void testBulkIndex() throws IOException { 
            //准备新增的文档
            List<Item> list = new ArrayList<>();
            list.add(new Item(1L, "小米手机7", "手机", "小米", 3299.00, "http://image.csdn.com/13123.jpg"));
            list.add(new Item(2L, "坚果手机R1", "手机", "锤子", 3699.00, "http://image.csdn.com/13123.jpg"));
            list.add(new Item(3L, "华为META10", "手机", "华为", 4499.00, "http://image.csdn.com/13123.jpg"));
            list.add(new Item(4L, "小米Mix2S", "手机", "小米", 4299.00, "http://image.csdn.com/13123.jpg"));
            list.add(new Item(5L, "荣耀V10", "手机", "华为", 2799.00, "http://image.csdn.com/13123.jpg"));
            // 创建批量新增请求
            BulkRequest bulkRequest = new BulkRequest();
            for (Item item : list) { 
                // 构建单个IndexRequest，并add到BulkRequest中去
                bulkRequest.add(new IndexRequest("索引名","类型名",item.getId().toString()).source(gson.toJson(item),XContentType.JSON));
            }
    
            // 发起请求
            BulkResponse bulk = client.bulk(bulkRequest,RequestOptions.DEFAULT);
            //打印
            System.out.println("结果::"+bulk);
        }
    
        /** * 后执行 */
        @After
        public void close(){ 
            try { 
                client.close();
            } catch (IOException e) { 
                e.printStackTrace();
            }
        }
    }

结果图:  
![在这里插入图片描述][2020112516300686.png_pic_center]

[Link 1]: https://blog.csdn.net/weixin_44257023/article/details/110130718
[2020112516300686.png_pic_center]: /images/20221120/79ed25402ef5474687373bc3b8ffe09f.png