From 801278977b8afff7e434f29815300e3027b4b781 Mon Sep 17 00:00:00 2001 From: 0007 <0007@qq.com> Date: Wed, 27 Aug 2025 19:57:05 +0800 Subject: [PATCH] Add File --- .../engines/test/TestLuceneCRUD.java | 96 +++++++++++++++++++ 1 file changed, 96 insertions(+) create mode 100644 agents-flex-search-engine/agents-flex-search-engine-lucene/src/test/java/com/agentsflex/engines/test/TestLuceneCRUD.java diff --git a/agents-flex-search-engine/agents-flex-search-engine-lucene/src/test/java/com/agentsflex/engines/test/TestLuceneCRUD.java b/agents-flex-search-engine/agents-flex-search-engine-lucene/src/test/java/com/agentsflex/engines/test/TestLuceneCRUD.java new file mode 100644 index 0000000..60371d7 --- /dev/null +++ b/agents-flex-search-engine/agents-flex-search-engine-lucene/src/test/java/com/agentsflex/engines/test/TestLuceneCRUD.java @@ -0,0 +1,96 @@ +/* + * Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com). + *

+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

+ * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.agentsflex.engines.test; + +import com.agentsflex.core.document.Document; +import com.agentsflex.search.engine.lucene.LuceneConfig; +import com.agentsflex.search.engine.lucene.LuceneSearcher; + +import java.util.List; + +public class TestLuceneCRUD { + public static void main(String[] args) { + // 1. 配置 Lucene 索引路径 + LuceneConfig config = new LuceneConfig(); + config.setIndexDirPath("./2lucene_index"); // 设置索引目录路径 + // 2. 创建 LuceneSearcher 实例 + + LuceneSearcher luceneSearcher = new LuceneSearcher(config); + + // 文档ID(用于更新和删除) + // ---- Step 1: 添加文档 ---- + System.out.println("【添加文档】"); + Document doc1 = new Document(); + doc1.setId(1); + doc1.setTitle("利润最大化的原则"); + doc1.setContent("平台客服工具:是指拼多多平台开发并向企业提供的功能或工具,商家通过其专属账号登录平台客服工具后,可以与平台消费者取得\n" + + "联系并为消费者提供客户服务"); + + boolean addSuccess = luceneSearcher.addDocument(doc1); + System.out.println("添加文档1结果:" + (addSuccess ? "成功" : "失败")); + + + Document doc2 = new Document(); + doc2.setId(2); + doc2.setTitle("企业获取报酬的活动"); + doc2.setContent("研究如何最合理地分配稀缺资源及不同的用途"); + + boolean addSuccess1 = luceneSearcher.addDocument(doc2); + System.out.println("添加文档2结果:" + (addSuccess1 ? "成功" : "失败")); + + // 查询添加后的结果 + testSearch(luceneSearcher, "企业"); + testSearch(luceneSearcher, "报酬"); + + // ---- Step 2: 更新文档 ---- + System.out.println("\n【更新文档】"); + Document updatedDoc = new Document(); + updatedDoc.setId(1); + updatedDoc.setContent("平台客服工具:是指拼多多平台开发并向商家提供的功能或工具,商家通过其专属账号登录平台客服工具后,可以与平台消费者取得\n" + + "联系并为消费者提供客户服务2"); + + boolean updateSuccess = luceneSearcher.updateDocument(updatedDoc); + System.out.println("更新文档结果:" + (updateSuccess ? "成功" : "失败")); + + // 查询更新后的结果 + testSearch(luceneSearcher, "消费者"); + + // ---- Step 3: 删除文档 ---- + System.out.println("\n【删除文档】"); + boolean deleteSuccess = luceneSearcher.deleteDocument(2); + System.out.println("删除文档结果:" + (deleteSuccess ? "成功" : "失败")); + + // 查询删除后的结果 + testSearch(luceneSearcher, "报酬"); + + } + + // 封装一个搜索方法,打印搜索结果 + private static void testSearch(LuceneSearcher searcher, String keyword) { + List results = searcher.searchDocuments(keyword); + if (results.isEmpty()) { + System.out.println("没有找到匹配的文档。"); + } else { + System.out.println("找到 " + results.size() + " 个匹配文档:"); + for (com.agentsflex.core.document.Document doc : results) { + System.out.println("ID: " + doc.getId()); + System.out.println("标题: " + doc.getTitle()); + System.out.println("内容: " + doc.getContent()); + System.out.println("-----------------------------"); + } + } + } +}