'use strict'; /** * * * @author Mai * @date * @version */ const TreeService = require('./base_tree_service'); // sql拼装器 const rootId = -1; class BaseBillsSerivce extends TreeService { /** * 从标准数据中提取有效数据 * @param {Object} stdData - 从标准库中查询所得 * @returns {name, unit, source, code, b_code} * @private */ _filterStdData(stdData) { const result = { name: stdData.name, unit: stdData.unit, source: stdData.source, node_type: stdData.node_type, }; result.code = stdData.code ? stdData.code : ''; result.b_code = stdData.b_code ? stdData.b_code : ''; return result; } async addBillsNode(tenderId, selectId, data, reviseId) { if (data) { if (reviseId) data.crid = reviseId; return await this.addNode(tenderId, selectId, data); } else { return await this.addNode(tenderId, selectId, {crid: reviseId}); } } /** * 新增子节点,并排在所有子节点的末尾 * @param {Number} tenderId - 标段Id * @param {Number} selectId - 父节点Id * @param {Object} data - 新增节点数据(编号名称等) * @returns {Promise<*>} */ async addChild(tenderId, selectId, data, reviseId) { if ((tenderId <= 0) || (selectId <= 0)) { return []; } const selectData = await this.getDataByKid(tenderId, selectId); if (!selectData) { throw '新增节点数据错误'; } const children = await this.getChildrenByParentId(tenderId, selectId); const maxId = await this._getMaxLid(tenderId); data.id = this.uuid.v4(); data.tender_id = tenderId; data.ledger_id = maxId + 1; data.ledger_pid = selectData.ledger_id; data.level = selectData.level + 1; data.order = children.length + 1; data.full_path = selectData.full_path + '-' + data.ledger_id; data.is_leaf = true; if (reviseId) data.crid = reviseId; this.transaction = await this.db.beginTransaction(); try { const result = await this.transaction.insert(this.tableName, data); if (children.length === 0) { await this.transaction.update(this.tableName, {is_leaf: false, quantity: null, unit_price: null, total_price: null, deal_qty: null, deal_tp: null}, { where: {tender_id: tenderId, ledger_id: selectData.ledger_id} }); } await this.transaction.commit(); } catch(err) { this.transaction.rollback(); throw err; } this._cacheMaxLid(tenderId, maxId + 1); // 查询应返回的结果 const resultData = {}; resultData.create = await this.getDataByKid(tenderId, data.ledger_id); if (children.length === 0) { resultData.update = await this.getDataByKid(tenderId, selectId); } return resultData; } /** * 添加节点(来自标准清单) * @param {Number} tenderId * @param {Number} selectId * @param {Object} stdData * @return {Promise<*>} */ async addStdNode(tenderId, selectId, stdData, reviseId) { const newData = this._filterStdData(stdData); const result = await this.addBillsNode(tenderId, selectId, newData, reviseId); return result; } /** * 添加标准节点,将选择的标准节点,添加为子节点(排序为末尾) * @param {Number} tenderId - 标段Id * @param {Number} selectId - 添加目标节点Id * @param {Object} stdData - 标准节点数据 * @returns {Promise<*>} */ async addStdNodeAsChild(tenderId, selectId, stdData, reviseId) { const newData = this._filterStdData(stdData); const result = await this.addChild(tenderId, selectId, newData, reviseId); return result; } /** * 根据parentData, data新增数据(新增为parentData的最后一个子项) * @param {Number} tenderId - 标段id * @param {Object} parentData - 父项数据 * @param {Object} data - 新增节点,初始数据 * @return {Promise<*>} - 新增结果 * @private */ async _addChildNodeData(tenderId, parentData, data, reviseId) { if (tenderId <= 0) { return undefined; } if (!data) { data = {}; } const pid = parentData ? parentData.ledger_id : rootId; const maxId = await this._getMaxLid(tenderId); data.id = this.uuid.v4(); data.tender_id = tenderId; data.ledger_id = maxId + 1; data.ledger_pid = pid; if (data.order === undefined) { data.order = 1; } data.level = parentData ? parentData.level + 1 : 1; data.full_path = parentData ? parentData.full_path + '-' + data.ledger_id : '' + data.ledger_id; if (data.is_leaf === undefined) { data.is_leaf = true; } if (reviseId) data.crid = reviseId; const result = await this.transaction.insert(this.tableName, data); this._cacheMaxLid(tenderId, maxId + 1); return [result, data]; } /** * 根据parentData, data新增数据(自动排序) * @param tenderId * @param parentData * @param data * @return {Promise} * @private */ async _addChildAutoOrder(tenderId, parentData, data, reviseId) { const self = this; const findPreData = function(list, a) { if (!list || list.length === 0) { return null; } for (let i = 0, iLen = list.length; i < iLen; i++) { if (self.ctx.helper.compareCode(list[i].code, a.code) > 0) { return i > 0 ? list[i - 1] : null; } } return list[list.length - 1]; }; const pid = parentData ? parentData.ledger_id : rootId; const children = await this.getChildrenByParentId(tenderId, pid); const preData = findPreData(children, data); if (!preData || children.indexOf(preData) < children.length - 1) { await this._updateChildrenOrder(tenderId, pid, preData ? preData.order + 1 : 1); } data.order = preData ? preData.order + 1 : 1; const [addResult, node] = await this._addChildNodeData(tenderId, parentData, data, reviseId); return [addResult, node]; } /** * 添加节点,并同步添加父节点 * @param {Number} tenderId - 标段id * @param {Number} selectId - 选中节点id * @param {Object} stdData - 节点数据 * @param {StandardLib} stdLib - 标准库 * @return {Promise} */ async addStdNodeWithParent(tenderId, stdData, stdLib, reviseId) { // 查询完整标准清单,并按层次排序 const fullLevel = await stdLib.getFullLevelDataByFullPath(stdData.list_id, stdData.full_path); fullLevel.sort(function(x, y) { return x.level - y.level; }); let isNew = false, node, firstNew, updateParent, addResult; const expandIds = []; this.transaction = await this.db.beginTransaction(); try { // 从最顶层节点依次查询是否存在,否则添加 for (let i = 0, len = fullLevel.length; i < len; i++) { const stdNode = fullLevel[i]; if (isNew) { const newData = this._filterStdData(stdNode); newData.is_leaf = (i === len - 1); [addResult, node] = await this._addChildNodeData(tenderId, node, newData, reviseId); } else { const parent = node; node = await this.getDataByCondition({ tender_id: tenderId, ledger_pid: parent ? parent.ledger_id : rootId, code: stdNode.code, name: stdNode.name, }); if (!node) { isNew = true; const newData = this._filterStdData(stdNode); newData.is_leaf = (i === len - 1); [addResult, node] = await this._addChildAutoOrder(tenderId, parent, newData, reviseId); if (parent && parent.is_leaf) { await this.transaction.update(this.tableName, { id: parent.id, is_leaf: false, unit_price: null, quantity: null, total_price: null, deal_qty: null, deal_tp: null}); updateParent = parent; } firstNew = node; } else { expandIds.push(node.ledger_id); } } } await this.transaction.commit(); } catch (err) { await this.transaction.rollback(); throw err; } // 查询应返回的结果 let createData = [], updateData = []; if (firstNew) { createData = await this.getDataByFullPath(tenderId, firstNew.full_path + '%'); updateData = await this.getNextsData(tenderId, firstNew.ledger_pid, firstNew.order); if (updateParent) { updateData.push(await this.getDataByCondition({ id: updateParent.id })); } } return { create: createData, update: updateData }; } } module.exports = BaseBillsSerivce;