当前位置：首页 > news >正文

使用 Java RestClient 与 Elasticsearch 进行商品文档操作

news 来源：原创 2024/9/21 10:59:14

文章目录

- 前期准备
- 配置 Elasticsearch 客户端
- 添加文档
- 查询文档
- 更新文档
- 删除文档
- 批量操作
- 结语

在现代应用中，Elasticsearch 已成为处理海量数据和实现高效搜索的利器。今天，我们将探讨如何在 Spring Boot 项目中使用 Elasticsearch 进行商品文档的增删改查操作。通过这篇文章，你将了解到如何配置 Elasticsearch 客户端以及具体的实现步骤。

前期准备

在开始之前，我们需要确保已经在项目中引入了必要的依赖，并且已经安装并运行了 Elasticsearch 服务。本文假设你已经完成了这些准备工作。

配置 Elasticsearch 客户端

首先，我们需要配置 Elasticsearch 客户端。在每个测试方法执行之前，我们会初始化 RestHighLevelClient，并在测试结束后关闭它，以确保资源的正确释放。以下是 setUp 和 tearDown 方法的实现：

@BeforeEach
void setUp() {client = new RestHighLevelClient(RestClient.builder(HttpHost.create("http://192.168.xxx.xxx:9200")));
}@AfterEach
void tearDown() throws IOException {if (client != null) {client.close();}
}

添加文档

接下来，我们来看如何将商品文档添加到 Elasticsearch 中。首先，我们需要从数据库中获取商品数据，然后将其转换为 Elasticsearch 文档格式并进行存储。以下是具体实现步骤：

@Test
void testIndexDoc() throws IOException {// 0.准备文档数据// 0.1.根据id查询数据库数据Item item = itemService.getById(317578L);// 0.2.把数据库数据转为文档数据ItemDoc itemDoc = BeanUtil.copyProperties(item, ItemDoc.class);itemDoc.setPrice(29900);// 1.准备Request对象IndexRequest request = new IndexRequest("items").id(itemDoc.getId());// 2.准备请求参数request.source(JSONUtil.toJsonStr(itemDoc), XContentType.JSON);// 3.发送请求IndexResponse response = client.index(request, RequestOptions.DEFAULT);System.out.println("response=" + response);
}

在上述代码中，我们通过 itemService 获取商品数据，并使用 BeanUtil 工具将其转换为 ItemDoc 对象，最后将该对象作为 JSON 字符串存储到 Elasticsearch 中。

查询文档

为了验证文档是否成功存储，我们可以通过文档 ID 来查询文档。以下是查询文档的示例代码：

@Test
void testGetDoc() throws IOException {// 1.准备Request对象GetRequest request = new GetRequest("items", "317578");// 2.发送请求GetResponse response = client.get(request, RequestOptions.DEFAULT);// 3.解析响应String json = response.getSourceAsString();ItemDoc itemDoc = JSONUtil.toBean(json, ItemDoc.class);System.out.println("doc=" + itemDoc);
}

更新文档

如果需要更新文档中的某些字段，我们可以使用 UpdateRequest 来实现。以下是更新文档的示例代码：

@Test
void testUpdateDoc() throws IOException {// 1.准备Request对象UpdateRequest request = new UpdateRequest("items", "317578");// 2.准备请求参数request.doc("price", 25600);// 3.发送请求client.update(request, RequestOptions.DEFAULT);
}

删除文档

最后，如果需要删除文档，可以使用 DeleteRequest 来实现。以下是删除文档的示例代码：

@Test
void testDeleteDoc() throws IOException {// 1.准备Request对象DeleteRequest request = new DeleteRequest("items", "317578");// 2.发送请求client.delete(request, RequestOptions.DEFAULT);
}

批量操作

在实际应用中，我们可能需要批量处理文档。以下是批量添加文档的示例代码：

@Test
void testBulkDoc() throws IOException {int pageNo = 1, pageSize = 500;while (true) {// 1.准备文档数据Page<Item> page = itemService.lambdaQuery().eq(Item::getStatus, 1).page(Page.of(pageNo, pageSize));List<Item> records = page.getRecords();if (records == null || records.isEmpty()) {return;}// 2.准备Request对象BulkRequest request = new BulkRequest();// 3.准备请求参数for (Item item : records) {request.add(new IndexRequest("items").id(item.getId().toString()).source(JSONUtil.toJsonStr(BeanUtil.copyProperties(item, ItemDoc.class)), XContentType.JSON));}// 4.发送请求client.bulk(request, RequestOptions.DEFAULT);// 5.翻页pageNo++;}
}