一日一技：在 MongoDB 中，如何批量更新不同數(shù)據(jù)為不同值？
我們知道，當(dāng)使用 Pymongo 更新MongoDB 字段的時(shí)候，我們有兩種常見的方法：
- 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í)候，不知道有沒有想過一個(gè)問題：update_many會(huì)對(duì)所有滿足條件的文檔更新相同的字段。例如，對(duì)于上面第二行代碼，所有name字段為value的數(shù)據(jù)，在更新以后，新的數(shù)據(jù)的aa字段的值全都是bb。那么，有沒有辦法一次性把不同的字段更新成不同的數(shù)據(jù)呢?
例如，我們的 MongoDB 中有如下數(shù)據(jù)：
| sid | name | sex | result | is_qualified |
|---|---|---|---|---|
| 1 | 王曉一 | 男 | 80 | true |
| 2 | 張小二 | 女 | 69 | false |
| 3 | 劉小三 | 男 | 76 | false |
| 4 | 朱小四 | 女 | 75 | true |
| 5 | 馬小五 | 男 | 50 | false |
| 6 | 趙小六 | 女 | 77 | true |
| 7 | 錢小七 | 男 | 60 | false |
| 8 | 孫小八 | 女 | 68 | false |
| 9 | 李小九 | 男 | 98 | true |
| 10 | 周小十 | 女 | 61 | false |
假設(shè)這是一份成績表，一開始，要求男生不低于80分，女生不低于70分，is_qualified字段才會(huì)為True。后來改了計(jì)分規(guī)則，變成男生不低于90分，女生不低于60分，is_qualified就能為 True，否則為 False。所以現(xiàn)在需要批量更新數(shù)據(jù)。顯然，對(duì)男生而言，有一些原本為True的需要變成 False;對(duì)女生而言，有一些原本為 False 的，要變成 True。如果讓你直接使用update_many，你可能需要寫成兩條更新語句：
- handler.update_many({'sex': '男', 'result': {'$lt': 90}}, {'$set': {'is_qualified': False}})
- handler.update_many({'sex': '女', 'result': {'$gte': 60}}, {'$set': {'is_qualified': True}})
那有沒有辦法只發(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]。
參考資料
[1]Bulk Write Operations — PyMongo 3.11.4 documentation: https://pymongo.readthedocs.io/en/stable/examples/bulk.html
