本文也在我的博客edwardesire.com上,欢迎品尝。
接着上周的工作,我们把上周反序列得到的dtree对象输出到JSON,再将其序列化后存入MongoDB。
-
存入文档
先将上次得到的决策树对象整个保存到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.");
}); -
序列化
树结构的序列化参照了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;
};当序列化结束后,将返回的结果整理排序再返回给上一层的路由控制函数。
-
存入数据库
最后就是将结果更新到数据库,使用方法.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('/');
}
}); -
总结
这样,项目后端比较重要的数据库操作工作就完成了,当然这还需要真实数据来检测。将整个文档读取操作是否足够好和安全,数据库中的参数项是否需要过滤掉,在更新数据库时v值也没有响应的变化,这些问题也需要好好考虑解决。