V2EX = way to explore
V2EX 是一个关于分享和探索的地方
Sign Up Now
For Existing Member  Sign In
推荐工具
RoboMongo
推荐书目
50 Tips and Tricks for MongoDB Developers
Related Blogs
Snail in a Turtleneck
JCZ2MkKb5S8ZX9pq
V2EX  ›  MongoDB

mongodb 给数组插入不重复的内容

  •  
  •   JCZ2MkKb5S8ZX9pq · Jan 6, 2019 · 6428 views
    This topic created in 2672 days ago, the information mentioned may be changed or developed.

    需求

    mongodb 里有一个字段,比如tags,有多个来源,来源都是数组,需要插入并去重。
    直观的方法是查询后去重,然后更新回去。
    网上看到还有 addToSet 这个命令。
    用两种方法都做了测试。

    先建一个临时表

    s = [i for i in range(10000)]
    for i in range(10000):
        db['test'].insert_one({'test_set': s})
    

    方法 1

    查询 /用旧列表和新列表合并 /更新

    s = [i for i in range(0, 20000, 2)]
    for i in db['test'].find():
        i['test_set'] = sorted(set(i['test_set']) | set(s))
        db['test'].update_one({'_id': i['_id']}, 
                              {'$set': {'test_set': i['test_set']}})
    

    方法 2

    网上看到的,用 addToSet。

    s = [i for i in range(0, 20000, 2)]
    for i in db['test'].find():
        i['test_set'] = sorted(set(i['test_set']) | set(s))
        db['test'].update_one({'_id': i['_id']},
                              {'$addToSet': {'test_set': {'$each': s}}})
    

    结论

    方法 2 超级超级慢,可能是 each 之后做了太多次查重导致的。
    本来还以为都在数据库端处理可能会快点,失败。
    跟大家分享下实验结果。

    附:$addToSet — MongoDB Manual

    如果有其它好的方法欢迎讨论。

    3 replies    2020-04-07 11:05:11 +08:00
    swulling
        1
    swulling  
       Jan 6, 2019 via iPhone
    方法一有并发冲突问题,谨慎使用。
    myKing
        2
    myKing  
       Feb 4, 2020
    还有其他方法?
    sego
        3
    sego  
       Apr 7, 2020
    看了[官网的文档]( https://docs.mongodb.com/manual/reference/operator/update/addToSet/),想不重复地添加元素到数组里,好像只能 addToSet 。
    要不看下 CPU 占用高不高,不高的话多线程试试?
    About   ·   Help   ·   Advertise   ·   Blog   ·   API   ·   FAQ   ·   Solana   ·   831 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 35ms · UTC 21:53 · PVG 05:53 · LAX 14:53 · JFK 17:53
    ♥ Do have faith in what you're doing.