ConstructionCost/modules/reports/models/tpl_tree_node.js

/**
 * 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, "$or": [{"isDeleted": null}, {"isDeleted": false} ]};
    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.removeNodes = function(nodeIds, preNodeId, preNodeNextId, callback){
    var functions = [];
    if (preNodeId != -1) {
        functions.push((function(nodeId, nextId) {
            return function(cb) {
                TreeNodeModel.update({ID: nodeId}, {"NextSiblingID": nextId}, cb);
            };
        })(preNodeId, preNodeNextId));
    }
    for (var i=0; i < nodeIds.length; i++) {
        functions.push((function(nodeId) {
            return function(cb) {
                TreeNodeModel.update({ID: nodeId}, {"isDeleted": true}, cb);
            };
        })(nodeIds[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();