MongoDB聚合操作中如何按条件删除文档而非仅移除指定字段？

实现方案 #

你可以使用MongoDB的bulkWrite顺序执行两个原子操作实现需求，逻辑和你现有写法完全对齐，仅将移除字段的逻辑替换为删除文档即可：

db.your_collection.bulk_write([
    # 第一步：和原有逻辑一致，先过滤移除指定的badge_id
    {
        'updateOne': {
            'filter': {'id': user_id},
            'update': [
                {
                    '$set': {
                        'badges': {
                            '$filter': {
                                'input': '$badges',
                                'cond': {'$ne': ['$$this', badge_id]}
                            }
                        }
                    }
                }
            ]
        }
    },
    # 第二步：如果处理后badges为空/不存在，直接删除整条文档
    {
        'deleteOne': {
            'filter': {
                'id': user_id,
                '$or': [
                    {'badges': {'$size': 0}},
                    {'badges': None},
                    {'badges': {'$exists': False}}
                ]
            }
        }
    }
])

逻辑说明 #

• 两个操作按顺序执行，对同一个文档的操作具备原子性，不会出现中间状态被其他请求读取的问题
• 第一步完全复用你现有的过滤逻辑，保证和原有业务逻辑对齐
• 第二步的删除条件覆盖了所有需要删除文档的场景：badges为空数组、badges为null、badges字段不存在

如果你不需要完全复用原有写法，也可以直接把判断逻辑放到删除的查询条件里，减少一次写操作：如果过滤掉目标badge后数组长度为0就直接删文档，否则只执行pull操作，代码如下：

# 先判断删除badge后是否会空，空的话直接删文档
delete_result = db.your_collection.delete_one({
    'id': user_id,
    '$expr': {
        '$eq': [
            {'$size': {'$filter': {'input': '$badges', 'cond': {'$ne': ['$$this', badge_id]}}}},
            0
        ]
    }
})
# 如果没删到文档，说明还有其他badge，执行pull操作
if delete_result.deleted_count == 0:
    db.your_collection.update_one(
        {'id': user_id},
        {'$pull': {'badges': badge_id}}
    )

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