ES在idea里的crud（酒店网站例）

Posted 2023-03-02 PaoShan

RestClient操作文档：本例是酒店网站例

我们用的是mp的框架。

1. 初始化RestHighLevelClient

启动类里注入RestHighLevelClient，依赖及其idea里自查es索引库见这篇：http://t.csdn.cn/rH3Ms

2. 注入接口

@SpringBootTest
public class HotelDocumentTest 

    @Autowired
    private IHotelService hotelService;

    private RestHighLevelClient client;

    @BeforeEach
    void initClient() 
        this.client = new RestHighLevelClient(RestClient.builder(
                HttpHost.create("http://192.168.136.130:9200")
        ));
    

    @AfterEach
    void closeClient() throws IOException 
        this.client.close();
    

索引库实体类

数据库查询后的结果是一个Hotel类型的对象，与我们的索引库结构存在差异：

• longitude和latitude需要合并为location

创建es映射，需要和mysql表一一对应。

新增！下面的是官方api

代码实现

我们导入酒店数据，基本流程一致，但是需要考虑几点变化：

• 酒店数据来自于数据库，我们需要先查询出来，得到hotel对象

• hotel对象需要转为HotelDoc对象

• HotelDoc需要序列化为json格式

@Test
public void testAddDocument() throws Exception 
    // 1-1 根据id查询酒店数据
    Hotel hotel = hotelService.getById(61083L);
    // 1-2 转为文档实体类型
    HotelDoc hotelDoc = new HotelDoc(hotel);
    // 1-3 将hotelDoc转为json
    String json = JSON.toJSONString(hotelDoc);

    // 2-1 创建request
    IndexRequest request = new IndexRequest("hotel").id(hotelDoc.getId().toString());
    // 2-2 准备DSL（json文档）
    request.source(json, XContentType.JSON);
    // 3-3 发送请求
    client.index(request, RequestOptions.DEFAULT);

查询！下面的是官方api

查询结果是一个JSON，其中文档放在一个_source属性中，因此解析就是拿到_source，反序列化为Java对象即可。

@Test
public void testGetDocumentById()throws Exception
    // 1.创建request
    GetRequest request = new GetRequest("hotel", "61083");
    // 2.发送请求，得到响应
    GetResponse response = client.get(request, RequestOptions.DEFAULT);
    // 3.解析响应
    String json = response.getSourceAsString();
    // 4.json转为hotelDoc
    HotelDoc hotelDoc = JSON.parseObject(json, HotelDoc.class);
    System.out.println(hotelDoc);

查询！下面的是官方api，全局修改和局部修改

• 全局修改：本质是先根据id删除，再新增

• 局部修改：修改文档中的指定字段值

在RestClient的API中，全量修改与新增的API完全一致，这里不再赘述，我们主要关注增量修改。

@Test
public void testUpdateDocument() throws Exception 
    // 1.创建request
    UpdateRequest request = new UpdateRequest("hotel", "61083");
    // 2.准备DSL
    request.doc(
        "price", "999",
        "starName", "四钻"
    );
    // 3.发送请求
    client.update(request, RequestOptions.DEFAULT);

删除！下面的是官方api

批量处理BulkRequest，其本质就是将多个普通的CUD请求组合在一起发送

其中提供了一个add方法，用来添加其他请求：

可以看到，能添加的请求包括：

• IndexRequest，也就是新增

• UpdateRequest，也就是修改

• DeleteRequest，也就是删除

因此Bulk中添加了多个IndexRequest，就是批量新增功能了。

官方提供的是，集群的两个。

我们用的是一个测试。

@Test
public void testBulkRequest() throws Exception 
    // 查询所有酒店数据
    List<Hotel> hotelList = hotelService.list();

    // 1.创建request
    BulkRequest request = new BulkRequest();
    // 2.准备DSL
    for (Hotel hotel : hotelList) 
        // 2-1 转为文档类型
        HotelDoc hotelDoc = new HotelDoc(hotel);
        // 2-2 转为json
        String json = JSON.toJSONString(hotelDoc);
        // 2-3 新增document
        request.add(new IndexRequest("hotel")
                    .id(hotelDoc.getId().toString())
                    .source(json, XContentType.JSON));
    
    
    // 3.发送请求
    client.bulk(request, RequestOptions.DEFAULT);

知识小结

文档操作的基本步骤：

• 初始化RestHighLevelClient

• 创建XxxRequest。XXX是Index、Get、Update、Delete、Bulk

• 准备参数（Index、Update、Bulk时需要）

• 发送请求。调用RestHighLevelClient..xxx()方法，xxx是index、get、update、delete、bulk

• 解析结果（Get时需要）