如何基于EnergyOffer查询结果批量更新EnergyOfferLog集合？

问题场景 #

我维护着energyOffers和energyOfferLogs两个MongoDB集合。用户注销账户时，需要先筛选出符合以下条件的活跃energyOffers：

• 用户对应的entity存在于文档的assignees数组中
• 该entity不在declinedEntities数组内
• offerValidTill的时间戳晚于当前时间

目前用来查询这些offer的代码是：

const energyOffers = await EnergyOffer.find([{ 
    'assignees.id': entityID, 
    declinedEntities: { 
        $ne: leadID 
    },
    offerValidTill: { $gt: Date.now() }
}], { session });

接下来需要更新对应的energyOfferLogs记录，单条日志的匹配规则是{ entityID: entityID, 'offer.offerID': offer._id }。但我不想遍历查询结果执行多次更新，希望用一次updateMany操作完成。试过考虑用$lookup做聚合，但觉得energyOffers的查询条件太复杂，不知道怎么落地。现在的更新代码框架是：

await EnergyOfferLog.updateMany({ ??? }, {
    $set: {
        'offer.action': 'declined',
        'offer.action_date': Math.floor(Date.now()),
        'offer.action_user': user.first_name,
        'offer.action_user_id': userID
    }
});

实现方案 #

方案一：先获取ID列表再批量匹配 #

这是最直接且兼容性好的方式，分两步走：

1. 查询符合条件的offer，只提取_id字段：

const energyOffers = await EnergyOffer.find(
    { 
        'assignees.id': entityID, 
        declinedEntities: { $ne: leadID },
        offerValidTill: { $gt: Date.now() }
    },
    { _id: 1, session } // 仅返回_id，减少数据传输量
);
const targetOfferIds = energyOffers.map(offer => offer._id);

2. 用$in操作符匹配所有目标ID，执行批量更新：

await EnergyOfferLog.updateMany(
    {
        entityID: entityID,
        'offer.offerID': { $in: targetOfferIds }
    },
    {
        $set: {
            'offer.action': 'declined',
            'offer.action_date': Math.floor(Date.now()),
            'offer.action_user': user.first_name,
            'offer.action_user_id': userID
        }
    },
    { session } // 如果用事务，记得带上session
);

方案二：用聚合管道实现数据库端关联更新（MongoDB 4.2+） #

如果希望把关联逻辑放在数据库侧，减少客户端和数据库的交互，可以用带聚合管道的updateMany（要求MongoDB版本≥4.2）：

await EnergyOfferLog.updateMany(
    {},
    [
        // 关联energyOffers集合，筛选符合条件的关联文档
        {
            $lookup: {
                from: 'energyOffers', // 注意这里要填实际的集合名称
                let: {
                    logEntity: '$entityID',
                    logOfferId: '$offer.offerID'
                },
                pipeline: [
                    {
                        $match: {
                            $expr: {
                                $and: [
                                    { $eq: ['$assignees.id', entityID] },
                                    { $ne: ['$declinedEntities', leadID] },
                                    { $gt: ['$offerValidTill', new Date()] },
                                    { $eq: ['$_id', '$$logOfferId'] },
                                    { $eq: [entityID, '$$logEntity'] }
                                ]
                            }
                        }
                    }
                ],
                as: 'matchedOffer'
            }
        },
        // 只保留关联到有效offer的日志
        { $match: { matchedOffer: { $ne: [] } } },
        // 执行字段更新
        {
            $set: {
                'offer.action': 'declined',
                'offer.action_date': Math.floor(Date.now()),
                'offer.action_user': user.first_name,
                'offer.action_user_id': userID
            }
        }
    ],
    { session }
);

方案对比 #

• 方案一逻辑简单，对MongoDB版本没有要求，性能开销低，适合大多数场景。
• 方案二将关联逻辑放在数据库端，减少了一次数据传输，但需要高版本MongoDB支持，且聚合管道的条件需要仔细校验，避免逻辑错误。

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