/** * Created by Tony on 2021/10/6. */ let fs = require('fs'); // let data = fs.readFileSync('D:/GitHome/ConstructionOperation/tmp/testDataResultSectionData_1633407022828.js'); // let data = fs.readFileSync('D:/GitHome/YangHuOperation/tmp/testDataResult_SectionTree_广西公路日常养护预算指标(2021).js'); let data = fs.readFileSync('D:/GitHome/YangHuOperation/tmp/testDataResult_SectionTree_广西公路养护预算定额(2021).js'); //---------------------------------------- const NODE_ID = "ID", P_ID = "ParentID", NEXT_ID = "NextSiblingID", ADHOC_PRE_ID="Previous_ID", CHILDREN_NODE = "items", SUB_ID = "sub_ids", EMPTY_ID_VAL = -1, TREE_LEVEL = 'treeLevel', TOP_BILL_ID = "topBillID"; let tree_Data_Helper = { buildTreeNodeDirectly: function(data, addLevel) { let topArr = [], rst = [], tmpNodes = {}, prefix = "id_"; let private_getStartNode = function (idArr) { let tmpNodeRst = null; for (let i = 0; i < idArr.length; i++) { if (parseInt(tmpNodes[prefix + idArr[i]][ADHOC_PRE_ID]) === EMPTY_ID_VAL) { tmpNodeRst = tmpNodes[prefix + idArr[i]]; break; } } return tmpNodeRst; }; let private_buildNodeData = function(parentItem, idArr, treeLevel, tbID) { let iter = [], nextNode = private_getStartNode(idArr), pushedIds = []; while (nextNode !== null && nextNode !== undefined ) { if (parentItem) { parentItem[CHILDREN_NODE].push(nextNode); } else { rst.push(nextNode); } iter.push(nextNode); pushedIds.push(nextNode[NODE_ID]); nextNode[TOP_BILL_ID] = tbID; if (parentItem === null) { nextNode[TOP_BILL_ID] = nextNode[NODE_ID]; if (nextNode.flags && nextNode.flags.length > 0) { for (let flag of nextNode.flags) { if (flag.fieldName === "fixed") { nextNode[TOP_BILL_ID] = flag.flag; break; } } } } if (addLevel) nextNode[TREE_LEVEL] = treeLevel; nextNode = tmpNodes[prefix + nextNode[NEXT_ID]]; if (nextNode === null || nextNode === undefined) { //备注: 考虑到实际数据的健壮性,有些节点会掉链子,需要用 parentItem[SUB_ID] 比对已经加上的节点,如发现加上的节点数量不够,那就得在这里补充上去 if (parentItem) { if (parentItem[SUB_ID].length > iter.length) { for (let subId of parentItem[SUB_ID]) { if (pushedIds.indexOf(subId) < 0) { let restNode = tmpNodes[prefix + subId]; if (addLevel) restNode[TREE_LEVEL] = treeLevel; restNode[TOP_BILL_ID] = tbID; parentItem[CHILDREN_NODE].push(restNode); iter.push(restNode); } } } } else { if (idArr.length > iter.length) { for (let topId of idArr) { if (pushedIds.indexOf(topId) < 0) { let restNode = tmpNodes[prefix + topId]; if (addLevel) restNode[TREE_LEVEL] = treeLevel; restNode[TOP_BILL_ID] = restNode[NODE_ID]; if (restNode.flags && restNode.flags.length > 0) { for (let flag of restNode.flags) { if (flag.fieldName === "fixed") { restNode[TOP_BILL_ID] = flag.flag; break; } } } rst.push(restNode); iter.push(restNode); } } } } } } pushedIds = []; for (let i = 0; i < iter.length; i++) { let rtbID = tbID; if (parentItem === null) { rtbID = iter[i][TOP_BILL_ID]; } private_buildNodeData(iter[i], iter[i][SUB_ID], (treeLevel + 1), rtbID); } }; //1. 给每个节点设置key, 顺便找Top Node for (let i = 0; i < data.length; i++) { tmpNodes[prefix + data[i][NODE_ID]] = data[i]; data[i][ADHOC_PRE_ID] = EMPTY_ID_VAL; data[i][SUB_ID] = []; data[i][CHILDREN_NODE] = []; if (parseInt(data[i][P_ID]) === EMPTY_ID_VAL) { topArr.push(data[i][NODE_ID]); } } //2. 通过key,设置好兄弟/父子关系 for (let i = 0; i < data.length; i++) { if (parseInt(data[i][NEXT_ID]) !== EMPTY_ID_VAL) { if (tmpNodes[prefix + data[i][NEXT_ID]] !== undefined){ if (tmpNodes[prefix + data[i][NEXT_ID]][P_ID] === data[i][P_ID]) { tmpNodes[prefix + data[i][NEXT_ID]][ADHOC_PRE_ID] = data[i][NODE_ID]; } else { tmpNodes[prefix + data[i][NEXT_ID]][ADHOC_PRE_ID] = EMPTY_ID_VAL; data[i][NEXT_ID] = EMPTY_ID_VAL; } } } if (parseInt(data[i][P_ID]) !== EMPTY_ID_VAL) { tmpNodes[prefix + data[i][P_ID]][SUB_ID].push(data[i][NODE_ID]); } } //3. 开build private_buildNodeData(null, topArr, 0, -1); //try to release and return tmpNodes = null; topArr.length = 0; return rst; }, getFlatArray: function (srcArr, destArr) { let private_put = function (parentItem) { destArr.push(parentItem); if (parentItem.items) { for (let subItem of parentItem.items) { private_put(subItem); } } } for (let node of srcArr) { private_put(node); } for (let item of destArr) { delete item[CHILDREN_NODE]; delete item[SUB_ID]; delete item[ADHOC_PRE_ID]; } } }; let tmpTreeArr = tree_Data_Helper.buildTreeNodeDirectly(JSON.parse(data), true); function _setChapterLevel(chapterNode, cLv) { chapterNode.chapterLv = cLv; if (chapterNode.items && chapterNode.items.length > 0) { for (let subNode of chapterNode.items) { _setChapterLevel(subNode, cLv); } } } for (let idx = 0; idx < tmpTreeArr[0].items.length; idx++) { _setChapterLevel(tmpTreeArr[0].items[idx], idx + 1); } // console.log(tmpTreeArr); let sectionTreeArr = []; tree_Data_Helper.getFlatArray(tmpTreeArr, sectionTreeArr); sectionTreeArr.splice(0,1); // console.log(sectionTreeArr); let newSectionData = []; let firstStr = 'ID,ParentID,NextSiblingID,Name,Code,FullCode,Name2'; newSectionData.push(firstStr); for (let sectionNode of sectionTreeArr) { // let node = []; // node.push(sectionNode.ID); // node.push(sectionNode.ParentID); // node.push(sectionNode.NextSiblingID); // node.push(sectionNode.name); // node.push(sectionNode.chapterLv); // node.push(''); //full code // node.push(''); //name2 // let str = `INSERT INTO RationTree Values(${sectionNode.ID}, ${sectionNode.ParentID}, ${sectionNode.NextSiblingID}, '${sectionNode.name}', ${sectionNode.chapterLv === undefined?-1:sectionNode.chapterLv}, NULL, NULL );` // let str = `${sectionNode.ID},${sectionNode.ParentID},${sectionNode.NextSiblingID},'${sectionNode.name}',${sectionNode.chapterLv === undefined?-1:sectionNode.chapterLv},'',''` let str = `${sectionNode.ID},${sectionNode.ParentID},${sectionNode.NextSiblingID},${sectionNode.name},${sectionNode.chapterLv === undefined?-1:sectionNode.chapterLv},,,` newSectionData.push(str); } // let ttlSQL = JSON.stringify(newSectionData); // let ttlSQL = newSectionData.join(''); let ttlSQL = newSectionData.join('\n'); let regExp = new RegExp('"', "gm"); ttlSQL = ttlSQL.replace(regExp, ''); // ttlSQL.replace() //fs.writeFile(`D:/GitHome/ConstructionOperation/tmp/章节树临时Data文件_${(new Date()).getTime()}.js`, ttlSQL, { 'flag': 'a', 'encoding': 'utf-8' }, function(err){ // fs.writeFile(`D:/GitHome/ConstructionOperation/tmp/章节树临时Data文件_${(new Date()).getTime()}.txt`, ttlSQL, { 'flag': 'a', 'encoding': 'utf-8' }, function(err){ fs.writeFile(`D:/GitHome/YangHuOperation/tmp/章节树临时Data文件_${(new Date()).getTime()}.txt`, ttlSQL, { 'flag': 'a', 'encoding': 'utf-8' }, function(err){ if(err) throw err; }); // */