MongoDB Shell - 新版本上线, 生产数据修复的JS入坑

因为最近公司项目的【分组模块】优化，整个实体结构变为树形结构，以前的Group实体大致如下：

public class Group extends BaseAuditEntity {
    /**
     * 平台
     */
    private String storeId;
    /**
     * 分组名称
     */
    private String name;
}

新增分组等级，父分组ID，地址，负责人，负责人电话字段。

因为存在生产数据，所以新版本上线后需要对数据进行修复。
方式1：自己在代码里预留一个接口，实现修复数据的功能，上线后自己调用接口实现数据修复。
方式2：写一个MongoDB JS 脚本，上线后，统一执行。

自己写了一个脚本：因为走了些弯路，记录一下实现。

//查询出原来的所有分组
db.group1.find().forEach(function(oldGroup){
    var storeId = oldGroup.storeId; //获取原来分组的storeId
    //查询是否该平台已经添加rootGroup
    var storeRootGroup = db.group1.find({"storeId":storeId, "parentGroupId" : ""}).toArray();

    var dynamicValue = db.getCollection("user").findOne({"_id":ObjectId("59448ad103b5d0c2247505a7")}).username;

    if(storeRootGroup.length == 0) {
        var rootGroupEachStore = {
            "storeId" : storeId,
            "parentGroupId" : "",
            "name" : dynamicValue,
            "level" : NumberLong(1),
            "Address" : {"province":"", "city":"","county":"","detailAddress":""},
            "responsiblePerson" : "",
            "responsiblePersonPhone" : "",
            "updatedBy" : "SYSTEM",
            "updatedTime" : new Date(),
            "version" : NumberLong(0),
            "createdBy" : "SYSTEM",
            "createdTime" : new Date(),
            "documentState" : NumberLong(1)
        };
        var parimaryId = new ObjectId();
        rootGroupEachStore._id = parimaryId; //平台相关分组的ID
        db.group1.insert(rootGroupEachStore); //插入平台分组(一个平台一个Root分组)
        oldGroup.parentGroupId = parimaryId.str; //老的分组设置parentId
    } else {
        oldGroup.parentGroupId = storeRootGroup[0]._id.str;
    }
    oldGroup.level = NumberLong(2);
    oldGroup.Address = {"province":"", "city":"", "county":"", "detailAddress":""};
    oldGroup.responsiblePerson = "";
    oldGroup.responsiblePersonPhone = "";
    db.group1.update({"_id":oldGroup._id,}, oldGroup); //为已存在的分组添加新字段
});

1. db.group1.find() 方法需要调用toArray()才能获取到返回值
2. db.getCollection(“user”).findOne({“_id”:ObjectId(“59448ad103b5d0c2247505a7”)}) 通过ID查询方式
3. oldGroup.parentGroupId = parimaryId.str; 通过str属性获取ObjectId的字符串内容
4. var parimaryId = new ObjectId(); 直接产生一个ObjectId