一日一技：在 MongoDB 中，如何批量更新不同數(shù)據(jù)為不同值？

我們知道，當(dāng)使用 Pymongo 更新MongoDB 字段的時(shí)候，我們有兩種常見(jiàn)的方法：

handler.update_one({'name': 'value'}, {'$set': {'aa': 'bb'}}) handler.update_many({'name': 'value'}, {'$set': {'aa': 'bb'}})

其中，update_one是更新第一條滿足查詢條件的數(shù)據(jù)；update_many是更新所有滿足查詢條件的數(shù)據(jù)。大家在使用update_many的時(shí)候，不知道有沒(méi)有想過(guò)一個(gè)問(wèn)題：update_many會(huì)對(duì)所有滿足條件的文檔更新相同的字段。例如，對(duì)于上面第二行代碼，所有name字段為value的數(shù)據(jù)，在更新以后，新的數(shù)據(jù)的aa字段的值全都是bb。那么，有沒(méi)有辦法一次性把不同的字段更新成不同的數(shù)據(jù)呢？

例如，我們的 MongoDB 中有如下數(shù)據(jù)：

假設(shè)這是一份成績(jī)表，一開始，要求男生不低于80分，女生不低于70分，is_qualified字段才會(huì)為True。后來(lái)改了計(jì)分規(guī)則，變成男生不低于90分，女生不低于60分，is_qualified就能為 True，否則為 False。所以現(xiàn)在需要批量更新數(shù)據(jù)。顯然，對(duì)男生而言，有一些原本為True的需要變成 False；對(duì)女生而言，有一些原本為 False 的，要變成 True。如果讓你直接使用update_many，你可能需要寫成兩條更新語(yǔ)句：

handler.update_many({'sex': '男', 'result': {'$lt': 90}}, {'$set': {'is_qualified': False}}) handler.update_many({'sex': '女', 'result': {'$gte': 60}}, {'$set': {'is_qualified': True}})

那有沒(méi)有辦法只發(fā)一次請(qǐng)求，就同時(shí)更新兩組數(shù)據(jù)呢？其實(shí)方法也是有的，就是bulk_write：

import pymongo handler = pymongo.MongoClient().test_db.test_col handler.bulk_write([ pymongo.UpdateMany({'sex': '男', 'result': {'$lt': 90}}, {'$set': {'is_qualified': False}}), pymongo.UpdateMany({'sex': '女', 'result': {'$gte': 60}}, {'$set': {'is_qualified': True}}) ])

bulk_write接收一個(gè)列表作為參數(shù)。這個(gè)列表里面的每一個(gè)元素是一個(gè)pymongo.X對(duì)象，這里的 X 可能是InsertOne/InsertMany/DeleteOne/DeleteMany/UpdateOne/?UpdateMany……，基本上就是你想使用的對(duì)應(yīng)操作的駝峰命名法形式。

這種方式，Pymongo 會(huì)在一次請(qǐng)求同時(shí)提交這兩組操作，減少網(wǎng)絡(luò)連接的時(shí)間消耗。

批量操作不僅支持UpdateOne，還支持各種其他操作，你可以閱讀Bulk Write Operations — PyMongo 3.11.4 documentation[1]。

參考資料

MongoDB

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實(shí)的內(nèi)容，請(qǐng)聯(lián)系我們jiasou666@gmail.com 處理，核實(shí)后本網(wǎng)站將在24小時(shí)內(nèi)刪除侵權(quán)內(nèi)容。

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實(shí)的內(nèi)容，請(qǐng)聯(lián)系我們jiasou666@gmail.com 處理，核實(shí)后本網(wǎng)站將在24小時(shí)內(nèi)刪除侵權(quán)內(nèi)容。