ConstructionOperation/modules/ration_repository/models/ration_section_tree.js

/**
 * Created by Tony on 2017/4/21.
 */
const mongoose = require('mongoose');
let async = require('async');
let moment = require('moment');
let counter = require('../../../public/counter/counter');
let rationRepositoryDao = require('./repository_map');
const rationChapterTreeModel = mongoose.model('std_ration_lib_ration_chapter_trees');
const rationModel = mongoose.model('std_ration_lib_ration_items');
const compleRationSectionTemp = mongoose.model('complementary_ration_section_templates');
var rationChapterTreeDAO = function () { };

rationChapterTreeDAO.prototype.importSection = async function (rationRepId, sheetData) {
    const counterRst = await counter.counterDAO.getIDAfterCount(counter.moduleName.rationTree, sheetData.length);
    const insertData = [];
    const beginID = counterRst.sequence_value - (sheetData.length - 1);
    for (let i = 0; i < sheetData.length; i++) {
        const curData = sheetData[i];
        const nextData = sheetData[i + 1];
        const curID = beginID + i;
        const nextID = nextData ? curID + 1 : -1;
        const name = curData[0];
        insertData.push({
            rationRepId,
            name,
            ID: curID,
            NextSiblingID: nextID,
            ParentID: -1
        });
    }
    await rationChapterTreeModel.insertMany(insertData);
};

rationChapterTreeDAO.prototype.sectionTemplateCount = async function (compilationId) {
    return await compleRationSectionTemp.find({ compilationId }).count();
};

rationChapterTreeDAO.prototype.initSectionTemplate = async function (rationLibId, compilationId) {
    let sectionTempCount = await compleRationSectionTemp.find({ compilationId }).count();
    if (sectionTempCount > 0) {
        await compleRationSectionTemp.remove({ compilationId });
    }
    let bulks = [];
    let stdRationSection = await rationChapterTreeModel.find({ rationRepId: rationLibId });
    for (let data of stdRationSection) {
        delete data._doc._id;
        data._doc.compilationId = compilationId;
        bulks.push({ insertOne: { document: data } });
    }
    if (bulks.length > 0) {
        await compleRationSectionTemp.bulkWrite(bulks);
    }
};

rationChapterTreeDAO.prototype.getRationChapterTree = function (rationLibId, callback) {
    rationChapterTreeModel.find({ "rationRepId": rationLibId, "$or": [{ "isDeleted": null }, { "isDeleted": false }] }, function (err, data) {
        if (data.length) callback(0, data);
        else if (err) callback("获取定额树错误！", [])
        else callback(0, []);
    })
}

rationChapterTreeDAO.prototype.getNewRationTreeID = function (callback) {
    counter.counterDAO.getIDAfterCount(counter.moduleName.rationTree, 1, function (err, result) {
        if (err) {
            callback(err, null);
        }
        else {
            callback(0, result.sequence_value);
        }
    });
};

rationChapterTreeDAO.prototype.getRationChapterTreeByRepId = function (repId, callback) {
    rationChapterTreeModel.find({ "rationRepId": repId, "$or": [{ "isDeleted": null }, { "isDeleted": false }] }, function (err, data) {
        if (data.length) callback(0, data);
        else if (err) callback("获取定额树错误！", [])
        else callback(false, []);
    })
}

rationChapterTreeDAO.prototype.createNewNode = function (lastOpr, libId, lastNodeId, nodeData, callback) {
    counter.counterDAO.getIDAfterCount(counter.moduleName.rationTree, 1, function (err, result) {
        nodeData.rationRepId = libId;
        nodeData.ID = result.sequence_value;
        var node = new rationChapterTreeModel(nodeData);
        async.parallel([
            function (cb) {
                node.save(function (err, result) {
                    if (err) {
                        cb("章节树ID错误！", false);
                    } else {
                        if (lastNodeId > 0) {
                            rationChapterTreeModel.update({ ID: lastNodeId }, { "NextSiblingID": nodeData.ID }, function (err, rst) {
                                if (err) {
                                    cb("章节树ID错误！", false);
                                } else {
                                    cb(false, result);
                                }
                            });
                        } else cb(false, result);
                    }
                });
            },
            function (cb) {
                rationRepositoryDao.updateOprArr({ ID: libId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                    if (err) {
                        cb(err);
                    }
                    else {
                        cb(null);
                    }
                });
            }
        ], function (err, result) {
            if (err) {
                callback("章节树ID错误！", false);
            }
            else {
                callback(false, result[0]);
            }
        });
    });
},

    rationChapterTreeDAO.prototype.removeNodes = function (repId, lastOpr, nodeIds, preNodeId, preNodeNextId, callback) {
        var functions = [];
        if (preNodeId != -1) {
            functions.push((function (nodeId, nextId) {
                return function (cb) {
                    rationChapterTreeModel.update({ ID: nodeId }, { "NextSiblingID": nextId }, cb);
                };
            })(preNodeId, preNodeNextId));
        }
        for (var i = 0; i < nodeIds.length; i++) {
            functions.push((function (nodeId) {
                return function (cb) {
                    rationChapterTreeModel.update({ ID: nodeId }, { "isDeleted": true }, cb);
                };
            })(nodeIds[i]));
        }
        functions.push((function () {
            return function (cb) {
                rationRepositoryDao.updateOprArr({ ID: repId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                    if (err) {
                        cb(err);
                    }
                    else {
                        cb(null);
                    }
                })
            }
        })());
        async.parallel(functions, function (err, results) {
            callback(err, results);
        });
    }

rationChapterTreeDAO.prototype.updateExplanation = function (lastOpr, repId, nodeId, explanation, callback) {
    rationChapterTreeModel.update({ rationRepId: repId, ID: nodeId }, { $set: { explanation: explanation } }, function (err, result) {
        if (err) {
            callback(err);
        }
        else {
            rationRepositoryDao.updateOprArr({ ID: repId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                callback(null);
            });
        }
    });
};
rationChapterTreeDAO.prototype.updateErratumRecord = function (lastOpr, repId, nodeId, erratumRecord, callback) {
    rationChapterTreeModel.update({ rationRepId: repId, ID: nodeId }, { $set: { erratumRecord: erratumRecord } }, function (err, result) {
        if (err) {
            callback(err);
        }
        else {
            rationRepositoryDao.updateOprArr({ ID: repId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                callback(null);
            });
        }
    });
};
rationChapterTreeDAO.prototype.updateRuleText = function (lastOpr, repId, nodeId, ruleText, callback) {
    rationChapterTreeModel.update({ rationRepId: repId, ID: nodeId }, { $set: { ruleText: ruleText } }, function (err, result) {
        if (err) {
            callback(err);
        }
        else {
            rationRepositoryDao.updateOprArr({ ID: repId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                callback(null);
            });
        }
    });
};
rationChapterTreeDAO.prototype.updateSituation = function (lastOpr, repId, nodeId, situation, callback) {
    rationChapterTreeModel.update({ rationRepId: repId, ID: nodeId }, { $set: { jobContentSituation: situation } }, function (err, result) {
        if (err) {
            callback(err);
        }
        else {
            rationRepositoryDao.updateOprArr({ ID: repId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                callback(null);
            });
        }
    });
};
rationChapterTreeDAO.prototype.updateAnnoSituation = function (lastOpr, repId, nodeId, situation, callback) {
    rationChapterTreeModel.update({ rationRepId: repId, ID: nodeId }, { $set: { annotationSituation: situation } }, function (err, result) {
        if (err) {
            callback(err);
        }
        else {
            rationRepositoryDao.updateOprArr({ ID: repId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                callback(null);
            });
        }
    });
};

rationChapterTreeDAO.prototype.updateNodes = function (lastOpr, updateData, callback) {
    var functions = [];
    for (var i = 0; i < updateData.length; i++) {
        //var md = new rationChapterTreeModel(nodes[i]);
        //md.isNew = false;
        functions.push((function (doc) {
            return function (cb) {
                if (doc.updateType === 'update' && !doc.updateData.isDeleted) {
                    rationChapterTreeModel.update({ rationRepId: doc.updateData.rationRepId, ID: doc.updateData.ID }, doc.updateData, function (err) {
                        if (err) {
                            cb(err);
                        }
                        else {
                            cb(null);
                        }
                    });
                }
                else if (doc.updateType === 'update' && doc.updateData.isDeleted) {
                    rationChapterTreeModel.remove({ rationRepId: doc.updateData.rationRepId, ID: doc.updateData.ID }, function (err) {
                        if (err) {
                            cb(err);
                        }
                        else {
                            rationModel.remove({ sectionId: doc.updateData.ID }, function (err) {
                                if (err) {
                                    cb(err);
                                }
                                else {
                                    cb(null);
                                }
                            });

                        }
                    });
                }
                else if (doc.updateType === 'new') {
                    rationChapterTreeModel.create(doc.updateData, function (err) {
                        if (err) {
                            cb(err);
                        }
                        else {
                            cb(null);
                        }
                    });
                }
            };
        })(updateData[i]));
    }
    if (updateData.length > 0) {
        functions.push((function () {
            return function (cb) {
                rationRepositoryDao.updateOprArr({ ID: updateData[0].updateData.rationRepId }, lastOpr, moment(Date.now()).format('YYYY-MM-DD HH:mm:ss'), function (err) {
                    if (err) {
                        cb(err);
                    }
                    else {
                        cb(null);
                    }
                })
            }
        })());
    }
    async.parallel(functions, function (err, results) {
        if (!err) {
            err = 0;
        }
        callback(err, results);
    });
};

module.exports = new rationChapterTreeDAO()