需求:类似消息队列 多个用户同时从一个表中取数进行处理 为了保证每个用户取得数都不同 取完数后立即加上user_iddb.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)
1 回答
慕仙森
TA贡献1827条经验 获得超8个赞
是线程安全的。
实际上findAndModify
的行为跟update
是一样的,这两个函数在查询和更新之间是不会被打断的。不同的是findAndModify
在操作之后还可以返回更新前或更新后的文档。更多的差异可以看文档:Comparisons with the update Method
- 1 回答
- 0 关注
- 417 浏览
添加回答
举报
0/150
提交
取消