【lucene-plus】更新文档的实现逻辑

最新推荐文章于 2022-02-12 18:10:11 发布

非非非常大的草履虫

最新推荐文章于 2022-02-12 18:10:11 发布

阅读量339

点赞数

CC 4.0 BY-SA版权

分类专栏： java 文章标签： lucene

本文链接：https://blog.youkuaiyun.com/shenya2/article/details/120754611

java 专栏收录该内容

23 篇文章

订阅专栏

Lucene默认不支持部分Field更新，Lucene-Plus通过query-delete-add方式实现了这一功能。具体步骤包括：1) 使用SearchByAfter检索相关文档；2) 删除匹配条件的文档；3) 合并数据并重新保存，更新部分字段。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【lucene-plus】初始化索引

【lucene-plus】索引初始化的实现逻辑

【lucene-plus】保存文档的实现逻辑

lucene并不支持部分Field的更新，updateDocument的实现逻辑实际是delete-add。lucene-plus基于updateDocument的实现实现了部分feild的更新，大致逻辑：query-delete-add。

DocumentPlusService#updateDocument：

 /**
     * 更新文档
     * @param indexName 索引名称
     * @param builder 更新文档的条件
     * @param params 参数
     */
    public void updateDocument(String indexName, BooleanQuery.Builder builder, Map<String, Object> params) throws IOException {
        List<Document> docList = CollUtil.newArrayList();
        SearchAfterRequestDTO searchAfterRequestDTO = new SearchAfterRequestDTO();
        searchAfterRequestDTO.setIndexName(indexName);
        searchAfterRequestDTO.setBuilder(builder);
        searchAfterRequestDTO.setPageSize(1000);
        SortField sortField = new SortField(IndexEnum.TIMESTAMP.getName(), Type.LONG);
        searchAfterRequestDTO.setSort(new Sort(sortField));
        boolean flag = true;
        SearchAfterResultDTO resultDTO;
        while (flag) {
            resultDTO = this.searchByAfter(searchAfterRequestDTO);
            if(CollUtil.isEmpty(resultDTO.getResult())) {
                flag = false;
                continue;
            }
            docList.addAll(resultDTO.getResult());
            searchAfterRequestDTO.setLastScoreDoc(resultDTO.getLastDoc());
        }
        // 删除
        this.deleteDocument(indexName, builder);
        // 新增
        Map<String, Object> mergeParams;
        for (Document doc : docList) {
            mergeParams = this.mergeFields(doc.getFields(), params);
            this.addDocument(indexName, mergeParams);
        }
    }

1.根据指定条件检索出所有相关的文档，这里用到了lucene-plus已封装的的SearchByAfter；

2.根据指定条件执行删除文档；

3.实现数据合并，重新保存文档，合并文档逻辑：

   /**
     * 合并文档
     * @param sourceFields 文档
     * @param targetMap 新字段值
     */
    private Map<String, Object> mergeFields(List<IndexableField> sourceFields, Map<String, Object> targetMap) {
        Set<String> systemFields = CollUtil.newHashSet(IndexEnum.TIMESTAMP.getName());
        Map<String, Object> params = CollUtil.newHashMap();
        sourceFields.stream()
            .filter(t->!systemFields.contains(t.name()))
            .filter(t->!targetMap.containsKey(t.name())).forEach(t->params.put(t.name(), t.stringValue()));
        targetMap.forEach((key, value) -> {
            if(!systemFields.contains(key)) {
                params.put(key, value);
            }
        });
        return params;
    }