Mongo基于一张表的数据更新另一张表

Posted 2021-12-02

业务场景中，经常有这样一个需求，就是根据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返回游标遍历，只要知道游标节点访问方式即可轻易写出来。