需求:类似消息队列 多个用户同时从一个表中取数进行处理 为了保证每个用户取得数都不同 取完数后立即加上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)
2 回答
达令说
TA贡献1821条经验 获得超6个赞
是线程安全的。
实际上findAndModify
的行为跟update
是一样的,这两个函数在查询和更新之间是不会被打断的。不同的是findAndModify
在操作之后还可以返回更新前或更新后的文档。更多的差异可以看文档:Comparisons with the update Method
拉风的咖菲猫
TA贡献1995条经验 获得超2个赞
应用加锁解决不了程序分布式部署的问题。
这有点类似队列里的出队列。A取的了一条消息, 该消息就不能被B取到。这种并发控制用mongo的话,我一般酱紫:
记录加上一个处理状态字段,如
processStatus
, 默值认为0
未处理1
处理中2
处理成功3
处理失败4
重试。利用findAndModify具有原子性操作的特性。
db.foo.findAndModify({ query:{processStatus:[0,4]}, update: { $set: { processStatus: 1} }, });// 再根据处理结果 置processStatus为成功、失败或者重试。
参考文档:mongo的原则性和事务性
- 2 回答
- 0 关注
- 2979 浏览
添加回答
举报
0/150
提交
取消