Node.js+MongoDB场景下删除Customer文档后如何更新Connection关联引用

问题前置修正 #

你提供的Connection模型定义有误，重复粘贴了Customer的代码，我默认你的Connection模型中存在关联Customer的字段，字段名为customer，定义参考如下：

const mongoose = require('mongoose')
const connectionSchema = new mongoose.Schema({
  // 你的其他业务字段省略
  customer: {
    type: mongoose.Schema.Types.ObjectId,
    ref: 'Customer',
    required: false
  }
})
module.exports = mongoose.model('Connection', connectionSchema)

实现方案 #

你可以根据业务场景选择以下两种方案中的任意一种：

方案1：直接修改删除接口逻辑（适合单一场景删除Customer的情况） #

只需要调整原有删除接口，在Customer删除成功后同步处理Connection的关联数据即可，代码修改如下：

// 顶部先引入Connection模型，路径替换为你项目中的实际路径
const Connection = require('../models/connection')

router.delete('/customer/:id', async (req, res) => {
    try {
        const customer = await Customer.findByIdAndDelete(req.params.id)
        if (!customer) {
            return res.status(404).send()
        }

        // 以下两种处理逻辑二选一，按需选择
        // 逻辑1：仅清空Connection对该Customer的引用，保留Connection文档
        await Connection.updateMany(
            { customer: req.params.id },
            { $unset: { customer: "" } }
            // 如果不想删除字段，也可以用 { $set: { customer: null } }
        )

        // 逻辑2：直接删除所有关联了该Customer的Connection文档
        // await Connection.deleteMany({ customer: req.params.id })

        res.send(customer)
    } catch (e) {
        res.status(500).send()
    }
})

方案2：使用Mongoose钩子自动处理（适合多场景会触发Customer删除的情况） #

如果你的项目中多个地方都会调用Customer删除逻辑，推荐在Customer模型中添加删除钩子，自动触发关联数据清理，不需要每次写重复代码。
首先修改Customer模型定义：

const mongoose = require('mongoose')
// 引入Connection模型
const Connection = require('./connection')

// 注意原来的模型定义是直接传对象给mongoose.model，要先定义Schema才能加钩子
const customerSchema = new mongoose.Schema({
    cusName: {
        type: String,
        required: true,
        trim: true
    },
    nic: {
        type: String,
        default: false
    },
    cNumber: {
        type: String,
        default: true
    },
    cBranch: {
        type: String,
        default: true
    },
    cAddress: {
        type: String,
        default: true
    },
    cEmail: {
        type: String,
        default: true
    },
})

// 添加findOneAndDelete后置钩子，findByIdAndDelete内部会调用该钩子
customerSchema.post('findOneAndDelete', async function(deletedCustomer) {
    if (!deletedCustomer) return
    // 同样两种处理逻辑二选一
    await Connection.updateMany({ customer: deletedCustomer._id }, { $unset: { customer: "" } })
    // 或 await Connection.deleteMany({ customer: deletedCustomer._id })
})

const Customer = mongoose.model('Customer', customerSchema)
module.exports = Customer

使用该方案时，你原有删除接口不需要做任何修改，删除Customer时会自动清理Connection的关联引用。 #

额外注意 #

1. 代码中的关联字段名customer需要替换为你Connection模型中实际关联Customer的字段名
2. 如果你选择删除关联的Connection文档，需要额外考虑Payment模型对Connection的引用，避免出现无效的悬空引用，可参考同样的逻辑同步处理Payment的关联数据
3. 对数据一致性要求高的场景，建议添加MongoDB事务处理，避免Customer删除成功但关联数据更新失败导致的数据不一致问题。

内容的提问来源于stack exchange，提问作者tecs-x