如何在Mongoose中实现分类文档的级联删除？

针对你需要删除父分类同时递归删除所有子级、孙级等关联文档的需求，这里提供几种比递归更高效的实现方案：

1. 先批量查询所有关联ID，再一次性删除 #

利用MongoDB的$graphLookup聚合操作，递归获取目标分类的所有后代ID，然后将父ID和后代ID一起批量删除，避免多次数据库请求。

示例代码：

async function deleteCategoryAndDescendants(categoryId) {
    // 递归查询目标分类的所有后代
    const categoryTree = await Category.aggregate([
        { $match: { _id: categoryId } },
        {
            $graphLookup: {
                from: 'categories', // 注意这里是集合的实际名称（通常是模型名复数）
                startWith: '$_id',
                connectFromField: '_id',
                connectToField: 'parent',
                as: 'descendants'
            }
        }
    ]);

    if (!categoryTree.length) return;

    // 收集所有需要删除的ID：父ID + 所有后代ID
    const idsToDelete = [categoryId, ...categoryTree[0].descendants.map(item => item._id)];

    // 执行批量删除
    await Category.deleteMany({ _id: { $in: idsToDelete } });
}

优势：仅需两次数据库操作（聚合查询+批量删除），性能远优于递归多次查询，且逻辑清晰易维护。

2. 利用Mongoose预删除钩子自动处理后代删除 #

通过Mongoose的pre钩子，在执行删除操作时自动触发后代删除逻辑，让业务代码更简洁。

示例代码：

// 在Category模型的Schema中定义预删除钩子
CategorySchema.pre('deleteOne', { document: false, query: true }, async function() {
    const targetId = this.getQuery()._id;
    
    // 查询当前分类的所有后代
    const descendants = await Category.aggregate([
        { $match: { parent: targetId } },
        {
            $graphLookup: {
                from: 'categories',
                startWith: '$_id',
                connectFromField: '_id',
                connectToField: 'parent',
                as: 'allDescendants'
            }
        }
    ]);

    // 收集所有后代ID
    const descendantIds = descendants.flatMap(node => [node._id, ...node.allDescendants.map(item => item._id)]);
    
    if (descendantIds.length) {
        await Category.deleteMany({ _id: { $in: descendantIds } });
    }
});

使用时直接调用删除方法即可：

// 调用此方法会自动删除目标分类及其所有后代
await Category.findByIdAndDelete(targetCategoryId);

注意：

• 钩子参数{ document: false, query: true }表示该钩子针对的是模型级别的删除操作（如deleteOne、findByIdAndDelete），如果是文档实例的remove()方法，需要修改钩子配置为{ document: true, query: false }。
• 确保parent字段已建立索引，提升查询效率：CategorySchema.index({ parent: 1 });

3. 用事务保证删除操作的原子性 #

如果你的MongoDB部署为副本集或分片集群，可以使用事务确保所有关联文档要么全部删除，要么全部保留，避免部分删除导致的数据不一致。

示例代码：

async function deleteWithTransaction(categoryId) {
    const session = await mongoose.startSession();
    session.startTransaction();

    try {
        // 在事务中执行聚合查询
        const categoryTree = await Category.aggregate([
            { $match: { _id: categoryId } },
            {
                $graphLookup: {
                    from: 'categories',
                    startWith: '$_id',
                    connectFromField: '_id',
                    connectToField: 'parent',
                    as: 'descendants'
                }
            }
        ]).session(session);

        if (!categoryTree.length) {
            await session.abortTransaction();
            return;
        }

        const idsToDelete = [categoryId, ...categoryTree[0].descendants.map(item => item._id)];
        // 在事务中执行批量删除
        await Category.deleteMany({ _id: { $in: idsToDelete } }).session(session);

        await session.commitTransaction();
    } catch (err) {
        await session.abortTransaction();
        throw err;
    } finally {
        session.endSession();
    }
}

适用场景：对数据一致性要求极高的业务场景，比如电商分类删除，避免出现父分类已删但子分类残留的情况。

内容的提问来源于stack exchange，提问作者Huy Tran