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

Answer 2

应用加锁解决不了程序分布式部署的问题。这有点类似队列里的出队列。A取的了一条消息， 该消息就不能被B取到。这种并发控制用mongo的话，我一般酱紫：记录加上一个处理状态字段,如processStatus, 默值认为0未处理 1处理中 2处理成功 3处理失败 4重试。利用findAndModify具有原子性操作的特性。db.foo.findAndModify({    query:{processStatus:[0,4]},    update: { $set: { processStatus: 1} }, });// 再根据处理结果 置processStatus为成功、失败或者重试。参考文档：mongo的原则性和事务性