V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
推荐工具
RoboMongo
推荐书目
50 Tips and Tricks for MongoDB Developers
Related Blogs
Snail in a Turtleneck
Te11UA
V2EX  ›  MongoDB

MongoDB 对某列去重后,查询该列对应的另外一值是否都为 xx

  •  
  •   Te11UA · 2020-10-28 22:38:00 +08:00 · 3398 次点击
    这是一个创建于 1505 天前的主题,其中的信息可能已经有所发展或是发生改变。

    数据形如

    [
    {'status':'ok','message_id': '1', 'group_id':1},
    {'status':'bad','message_id': '2', 'group_id':1},
    {'status':'ok','message_id': '1', 'group_id':2}
    ],
    

    需求是要判断当前库中同一 group 的 status 都为 ok,有一个 bad 则返回 False 。由于不太熟悉 MongoDB,使用了比较难看的做法:

    group_ids = db.tasks.distinct('group_id')
    for group_id in group_ids:
    ____status = db.tasks.find_one({'group_id': group_id, 'status': 'bad'})
    ____return False if task else True
    

    请问大佬们,有什么好的办法可以完成这个需求呢

    3 条回复    2020-10-28 23:02:10 +08:00
    catinsides
        1
    catinsides  
       2020-10-28 22:57:17 +08:00
    ```
    aggregate([
    {
    $group: {
    _id: '$group_id',
    statusAry: {
    $addToSet: '$status'
    }
    }
    },
    {
    $addFields: {
    len: {
    $size: '$statusAry'
    }
    }
    },
    {
    $addFields: {
    results: {
    $cond: { if: { $eq: [ "$len", 1 ] }, then: true, else: false }
    }
    }
    }
    ])
    ```
    catinsides
        2
    catinsides  
       2020-10-28 22:59:31 +08:00
    没判断全是 bad
    lithiumii
        3
    lithiumii  
       2020-10-28 23:02:10 +08:00
    你这个是 pymongo ?如果可以在 py 里做的话

    group_ids = db.tasks.distinct('group_id')
    bad_ids = db.tasks.find({'status':'bad'}).distinct('group_id')
    good_ids = [x for x in group_ids if x not in bad_ids]

    或者用 MongoDB 的 aggregation
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   871 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 24ms · UTC 20:57 · PVG 04:57 · LAX 12:57 · JFK 15:57
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.