springboo3单体应用集成搜索引擎

import org.apache.lucene.queryparser.classic.ParseException;
import org.springframework.stereotype.Service;
import whoosh.Document;
import whoosh.Index;
import whoosh.Query;
import whoosh.Whoosh;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 基于Whoosh-java + IKAnalyzer的嵌入式检索服务
 */
@Service
public class WhooshSearchService {
    // 索引存储目录（内存索引：RAMDirectory，重启丢失；文件索引：FSDirectory）
    private final Index index;
    // 中文分词器
    private final IKAnalyzerLucene9 analyzer;

    /**
     * 初始化索引和分词器
     */
    public WhooshSearchService() throws IOException {
        // 1. 初始化IK分词器（智能分词模式）
        analyzer = new IKAnalyzerLucene9(true);

        // 2. 初始化索引（两种方式可选）
        // 方式1：内存索引（无需磁盘，适合测试/小数据量）
        index = Whoosh.createInMemory(analyzer);

        // 方式2：文件索引（持久化，适合生产）
        // String indexPath = "/tmp/whoosh-index"; // 自定义索引存储路径
        // index = Whoosh.create(indexPath, analyzer);

        // 3. 初始化测试数据（可选）
        initTestData();
    }

    /**
     * 初始化测试索引数据
     */
    private void initTestData() throws IOException {
        // 添加文档：字段名（name/category/price） + 字段值
        addDocument(Map.of(
                "id", "1",
                "name", "华为Mate60 Pro手机",
                "category", "数码产品",
                "price", "6999.0"
        ));
        addDocument(Map.of(
                "id", "2",
                "name", "苹果iPhone 15 Pro",
                "category", "数码产品",
                "price", "7999.0"
        ));
        addDocument(Map.of(
                "id", "3",
                "name", "小米笔记本Pro",
                "category", "数码产品",
                "price", "5999.0"
        ));
    }

    /**
     * 新增文档到索引
     * @param fieldMap 字段键值对（如：id、name、category等）
     */
    public void addDocument(Map<String, String> fieldMap) throws IOException {
        Document doc = new Document();
        // 遍历字段，添加到文档（TextField：分词存储，StringField：精确存储）
        fieldMap.forEach((field, value) -> {
            if ("name".equals(field)) {
                // 名称字段：分词存储（支持检索）
                doc.addTextField(field, value);
            } else {
                // 其他字段：精确存储（仅存储，不分词）
                doc.addStringField(field, value);
            }
        });
        // 写入索引
        index.addDocument(doc);
        // 提交索引（确保数据生效）
        index.commit();
    }

    /**
     * 检索文档
     * @param keyword 检索关键词
     * @param field 检索字段（如name）
     * @return 检索结果列表
     */
    public List<Map<String, String>> search(String keyword, String field) throws IOException, ParseException {
        // 1. 构建查询（检索指定字段包含关键词）
        Query query = Query.parse(field + ":" + keyword, analyzer);

        // 2. 执行查询（返回前10条）
        List<Document> docs = index.search(query, 10);

        // 3. 解析结果
        List<Map<String, String>> result = new ArrayList<>();
        for (Document doc : docs) {
            result.add(Map.of(
                    "id", doc.get("id"),
                    "name", doc.get("name"),
                    "category", doc.get("category"),
                    "price", doc.get("price")
            ));
        }
        return result;
    }

    /**
     * 关闭索引（应用关闭时调用）
     */
    public void close() throws IOException {
        index.close();
    }
}

4. 编写测试接口

import org.apache.lucene.queryparser.classic.ParseException;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.io.IOException;
import java.util.List;
import java.util.Map;

@RestController
public class SearchController {
    private final WhooshSearchService whooshSearchService;

    public SearchController(WhooshSearchService whooshSearchService) {
        this.whooshSearchService = whooshSearchService;
    }

    /**
     * 检索接口
     * @param keyword 关键词（如：手机、苹果）
     * @return 检索结果
     */
    @GetMapping("/search")
    public List<Map<String, String>> search(@RequestParam String keyword) {
        try {
            // 检索name字段包含关键词的文档
            return whooshSearchService.search(keyword, "name");
        } catch (IOException | ParseException e) {
            throw new RuntimeException("检索失败：" + e.getMessage());
        }
    }
}

// 删除文档（根据id）
index.deleteDocuments(Query.parse("id:1", analyzer));
// 更新文档（先删后加）
index.deleteDocuments(Query.parse("id:1", analyzer));
addDocument(Map.of("id", "1", "name", "华为Mate60 Pro+", "category", "数码产品", "price", "7999.0"));
index.commit();

// 检索name或category包含关键词
Query query = Query.parse("name:" + keyword + " OR category:" + keyword, analyzer);