需求:類似消息隊列 多個用戶同時從一個表中取數(shù)進行處理 為了保證每個用戶取得數(shù)都不同 取完數(shù)后立即加上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開發(fā)了一個腳本 驗證了一下 似乎是線程安全的 但是又不確定 為了保險起見 應(yīng)用代碼還是顯式加了鎖 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)
mongodb findOneAndUpdate 線程安全嗎?
慕哥9229398
2018-07-21 08:56:14