/** * Created by Tony on 2017/5/31. */ var mongoose = require('mongoose'); var dbm = require("../../../config/db/db_manager"); var db = dbm.getCfgConnection("Reports"); var async = require("async"); var rptTplDef = require("../../../public/rpt_tpl_def").getUtil(); var Schema = mongoose.Schema; var TreeNodeSchema = new Schema({ ID:Number, ParentID:Number, NextSiblingID:Number, grpType: Number, //建筑(const : 1)/公路(const 2)/其他 etc... nodeType: Number,//节点类型:树节点(枝) 或 模板节点(叶) tplType: Number, //概算、预算、招投标 etc... userId: String, //用户自定义模板用 refId: Number, // name: String, isDeleted: Boolean }); var TreeNodeModel = db.model("rpt_tpl_tree", TreeNodeSchema, "rpt_tpl_tree"); var counter = require('../../../public/counter/counter'); var RplTplTreeDAO = function(){}; RplTplTreeDAO.prototype.getTplTreeNodes = function(grpType, userId, tplType, callback) { var filter = {"grpType": grpType}; if (userId) { filter.userId = userId; } if ((tplType && tplType !== rptTplDef.TplType.ALL)) { filter.tplType = tplType; } TreeNodeModel.find(filter, '-_id', function(err, data){ if (err) { callback(true, null); } else callback(false,data); }); }; RplTplTreeDAO.prototype.updateTreeNodes = function(nodes, callback) { var functions = []; for (var i=0; i < nodes.length; i++) { functions.push((function(doc) { return function(cb) { TreeNodeModel.update({ID: doc.ID}, doc, cb); }; })(nodes[i])); } async.parallel(functions, function(err, results) { callback(err, results); }); }; RplTplTreeDAO.prototype.createTplTreeNode = function(nodeData, lastNodeId, callback) { counter.counterDAO.getIDAfterCount(counter.moduleName.report, 1, function(err, result){ nodeData.ID = result.value.sequence_value; var node = new TreeNodeModel(nodeData); node.save(function (err, result) { if (err) { callback("树节点错误!", false); } else { if (lastNodeId > 0) { TreeNodeModel.update({ID: lastNodeId}, {"NextSiblingID": nodeData.ID}, function(err, rst){ if (err) { callback("树节点错误!", false); } else { callback(false, result); } }); } else callback(false, result); } }); }); }; module.exports = new RplTplTreeDAO();