業務場景中,經常有這樣一個需求,就是根據A表中的信息,更新B表的某個字段,在關係型數據庫中,這個需求很簡單,一條SQL搞定:
update B set B.f1=A.a1 where xxxx
但在mongo數據庫中,一條SQL就解決不了。好在mongo支持js腳本,可以通過js寫類似於存儲過程的方式實現。譬如,業務場景中,有一個用戶表,一個車輛表,車輛表的狀態要彙總到用戶表中,
var cursor =
db.getCollection('vehicle').aggregate([
{$match:
{'checkStatus':2}
},
{$group:
{_id:"$userId",vehicleNum:{$sum:1}}
}
])
while(cursor.hasNext()){
r = cursor.next();
print(r["_id"] + "\t" + r["vehicleNum"]);
db.getCollection("user").update({_id: r["_id"]},
{$set: {vehicleNum: r["vehicleNum"],
updateDatetime: ISODate("2019-05-23T09:07:56.566+08:00")}
})
}
這個原理很簡單,就是根據mongo返回遊標遍歷,只要知道遊標節點訪問方式即可輕易寫出來。