需求：類似消息隊列 多個用戶同時從一個表中取數進行處理 為了保證每個用戶取得數都不同 取完數后立即加上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
-
qq_k_213不是原子性的，我踩坑了，并發下會報錯誤的 {"level":"error","msg":"[(DuplicateKey) E11000 duplicate key error collection: mk_dev.zhuqm_test index: _id_ dup key: { : \"fe2f3e8e-9380-11ea-bb26-88e9fe6ba529\" }]","stack":"*errors.errorString [(DuplicateKey) E11000 duplicate key error collection: mk_dev.zhuqm_test index: _id_ dup key: { : \"fe2f3e8e-9380-11ea-bb26-88e9fe6ba529\" }]\n/Users/zhuqiming/GoProjects/mk-data-scripts/logger/log.go:95 (0x148cbac)\n\tError: err := errors.New(args)\n/Users/zhuqiming/GoProjects/mk-data-scripts/controller/test/test.go:60 (0x1f16a08)\n\tsetOnInsert.func1: log.Error(err)\n/Users/zhuqiming/.g/go/src/runtime/asm_amd64.s:1357 (0x1064571)\n\tgoexit: BYTE\t$0x90\t// NOP\n","time":"2020-05-11 20:14:41","type":"app"}
拉風的咖菲貓
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 關注
- 3066 瀏覽
添加回答
舉報
0/150
提交
取消