mongodb findOneAndUpdate 线程安全吗？

Question

需求：

类似消息队列 多个用户同时从一个表中取数进行处理 为了保证每个用户取得数都不同 取完数后立即加上user_id

db.foo.findOneAndUpdate({"user_id":{$exists:false}, {$set:{user_id:"123456"}},{sort:{_id:1}}})

官方文档中并没有说明该命令是否线程安全 如有如下的记录

{"_id":1}
{"_id":2}
{"_id":3}

5个用户同时来请求 有没有可能都是取得_id为1的记录？

使用Python开发了一个脚本 验证了一下 似乎是线程安全的 但是又不确定 为了保险起见 应用代码还是显式加了锁 Python代码如下：

def query_and_set_user_id(user_id):
    result = db.test.find_one_and_update({"user_id":{"$exists":False}},{'$set': {'user_id': user_id}}, sort=[("id", 1)])    print(threading.current_thread(), result)    return result

max_workers = 5
pool = ThreadPoolExecutor(max_workers)
futures = []for i in range(max_workers):
    f = pool.submit(query_and_set_user_id, "user_{}".format(i+1))
    futures.append(f)

wait(futures)

Answer 1

是线程安全的。实际上findAndModify的行为跟update是一样的，这两个函数在查询和更新之间是不会被打断的。不同的是findAndModify在操作之后还可以返回更新前或更新后的文档。更多的差异可以看文档：Comparisons with the update Method