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

PaoShan

已于 2023-02-09 13:38:59 修改

阅读量214

点赞数

文章标签： elasticsearch 大数据 Powered by 金山文档

于 2023-01-29 17:27:06 首次发布

本文链接：https://blog.csdn.net/PaoShan/article/details/128791494

版权

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

我们用的是mp的框架。

初始化RestHighLevelClient

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

注入接口

@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);
}