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); //为已存在的分组添加新字段
});

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

秒客网

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

相关文章