NodeJS:树的序列化

接着上周的工作,我们把上周反序列得到的dtree对象输出到JSON,再将其序列化后存入MongoDB。


  1. 存入文档
    先将上次得到的决策树对象整个保存到JSON文件中,这个需要使用Node内建模块fs的.writeFile()方法,需要记得的一点是,JSON对象需要先使用stringify将其转换为字符串再存入到文件,不然存在文件中的数据就是[object Object]这种JS的对象。

    fs.writeFile(__dirname + '/../testjson/trees.json', JSON.stringify(dtrees), function (err) {
        if (err) throw err;
        console.log("It's writed.");
    });  
    
  2. 序列化
    树结构的序列化参照了python实现二叉树的序列化这篇文章的方法:使用先序来遍历树的节点,将其存入数组。

    首先是解开树结构将节点存入数组当中。

    /**
     * 递归,先序优先历遍每个节点,将节点的子节点存入数组
     * @function serialize
     * @param {Object Node} node 树结构的节点集,递归时为存入node_array的元素(节点)
     *         {Array} node_array 数组结构的节点集
     * @return {Object Node} node 包含树结构的节点,每次递归,节点的children数组元素(节点)少一
     *          {Array} node_array 数组结构的节点集,每次递归,该数组元素(节点)添一
     */
    var serialize = function (node, node_array) {
        while (node.children.length) {
            node_array = serialize(node.children [0], node_array).node_array;
            node.children.splice(0, 1); //.splice(i, 1) 删除node.children[]中index i的元素
            //需要注意 当子节点children数组为空,此步无效
        }
        node_array.push(node); //.push()到node_array
        return ({node: node, node_array: node_array});
    };  
    

    每次递归从节点的第一个子节点开始递归,递归回调时删除这个子节点。当跳出循环时,当前节点就是一个没有子节点的单独的节点,这样就可以妥妥地将单个节点存入数组当中,最后将修改后的结果返回给调用处。

    /**
     * 调用递归方法.serialize(),将dtree的树结构节点序列化为数组
     * @function dtreeToArray
     * @param {Object Node} nodes 树结构的节点dtree文档中的node_array节点
     * @return {Array} node_array 升序节点数组
     */
    var dtreeToArray = function (nodes) {
        var node_array = [];
        node_array = serialize(nodes, node_array).node_array;
        node_array.sort(function (a, b) {
            return a.node_id - b.node_id
        });
        //console.log(node_array);
        return node_array;
    };
    

    当序列化结束后,将返回的结果整理排序再返回给上一层的路由控制函数。

  3. 存入数据库
    最后就是将结果更新到数据库,使用方法.update(),它的第一个参数是条件(等同于关系型中的where),第二个就是需要更新的内容,最后回调函数给出后续操作。

    Dtree.update({"_id": dtreeJsonData._id}, {"node_array": nodes, "config.last_saved_time": Date.now()}, function(err){
            if(err){
                res.send({'success':false,'err':err});
            }else{
                //res.send({'success':true});
                //console.log("dtree created and saved: " + dtree);
                res.redirect('/');
            }
    });    
    
  4. 总结
    这样,项目后端比较重要的数据库操作工作就完成了,当然这还需要真实数据来检测。将整个文档读取操作是否足够好和安全,数据库中的参数项是否需要过滤掉,在更新数据库时v值也没有响应的变化,这些问题也需要好好考虑解决。