【笔记】nodejs读取JSON,数组转树

const fs = require('fs');

// --------------- 读取源文件 ---------------
const originData = require('./vuxArea3.json');

// --------------- 常量 ---------------
const constant = {
  ARRAY_TYPE: 'Array',
  JSON_TYPE: 'Json',
};

// --------------- 转换的配置 ---------------

const config = {
  originKey: {
    children: null,
    parent: 'parent',
    value: 'value',
    label: 'name'
  },
  originType: constant.ARRAY_TYPE,
  targetKey: {
    children: 'children',
    parent: null,
    value: 'value',
    label: 'label'
  },
  targetType: constant.JSON_TYPE
};


// --------------- 转换函数 ---------------
function convertArrayToJson(initData, { originKey, targetKey }) {
  if (!initData || initData.length === 0) return [];
  if (!initData.length === 1) return initData;
  if (!(initData instanceof Array)) return [initData];

  const result = [];
  const tempMap = [];

  const key = originKey.value;
  const parentKey = originKey.parent;
  const l = initData.length;
  let i;

  // key转换
  const data = initData.map(item => ({
    [originKey.parent]: item[originKey.parent],
    [targetKey.value]: item[originKey.value],
    [targetKey.label]: item[originKey.label]
  }));

  // 临时引用
  for (i = 0; i < l; i += 1) {
    tempMap[data[i][key]] = data[i];
  }

  // 叶子通过临时引用挂载,利用所有的节点都是单例,才可以通过临时引用去挂载叶子
  // 根元素直接push到根节点
  for (i = 0; i < l; i += 1) {
    const parentValue = data[i][parentKey];
    // 当前元素存在父元素,并且子不等于父
    if (tempMap[parentValue] && data[i][key] !== parentValue) {
      // 父元素上在children上加上当前元素
      tempMap[parentValue][targetKey.children] =
        tempMap[parentValue][targetKey.children] || [];
      tempMap[parentValue][targetKey.children].push(data[i]);
    } else {
      // 当前元素是根元素
      result.push(data[i]);
    }
  }
  return result;
}

function convert(data, { originKey, originType, targetKey, targetType }) {
  if (originType === constant.ARRAY_TYPE && targetType === constant.JSON_TYPE) {
    return convertArrayToJson(data, { originKey, targetKey });
  }
  return [];
}

// --------------- 执行 ---------------
const result = convert(originData, config);
let resultStr = '';
try {
  resultStr = JSON.stringify(result);
} catch (e) {
  console.log(e);
}
resultStr = resultStr.replace(/"parent":"[a-zA-Z\d]+",/g, '');

// --------------- 写入 ---------------
fs.writeFile('./result.json', resultStr, 'utf-8', (err) => {
  if (err) {
    console.log(err);
  } else {
    console.log(`### conver ${config.originType} to ${config.targetType} success! `);
  }
});
const fs = require('fs');
// --------------- 读取源文件 ---------------
const originData = require('./vuxArea3.json');
// --------------- 常量 ---------------
const constant = {
  ARRAY_TYPE: 'Array',
  JSON_TYPE: 'Json',
};
// --------------- 转换的配置 ---------------
const config = {
  originKey: {
    children: null,
    parent: 'parent',
    value: 'value',
    label: 'name'
  },
  originType: constant.ARRAY_TYPE,
  targetKey: {
    children: 'children',
    parent: null,
    value: 'value',
    label: 'label'
  },
  targetType: constant.JSON_TYPE
};
// --------------- 转换函数 ---------------
function convertArrayToJson(initData, { originKey, targetKey }) {
 if (!initData || initData.length === 0) return [];
 if (!initData.length === 1) return initData;
 if (!(initData instanceof Array)) return [initData];
 const result = [];
 const tempMap = [];
 const key = originKey.value;
 const parentKey = originKey.parent;
 const l = initData.length;
 let i;
 // key转换
 const data = initData.map(item => ({
    [originKey.parent]: item[originKey.parent],
    [targetKey.value]: item[originKey.value],
    [targetKey.label]: item[originKey.label]
  }));
 // 临时引用
 for (i = 0; i < l; i += 1) {
    tempMap[data[i][key]] = data[i];
  }
 // 叶子通过临时引用挂载,利用所有的节点都是单例,才可以通过临时引用去挂载叶子
 // 根元素直接push到根节点
 for (i = 0; i < l; i += 1) {
 const parentValue = data[i][parentKey];
 // 当前元素存在父元素,并且子不等于父
 if (tempMap[parentValue] && data[i][key] !== parentValue) {
 // 父元素上在children上加上当前元素
      tempMap[parentValue][targetKey.children] =
        tempMap[parentValue][targetKey.children] || [];
      tempMap[parentValue][targetKey.children].push(data[i]);
    } else {
 // 当前元素是根元素
      result.push(data[i]);
    }
  }
 return result;
}
function convert(data, { originKey, originType, targetKey, targetType }) {
 if (originType === constant.ARRAY_TYPE && targetType === constant.JSON_TYPE) {
 return convertArrayToJson(data, { originKey, targetKey });
  }
 return [];
}
// --------------- 执行 ---------------
const result = convert(originData, config);
let resultStr = '';
try {
  resultStr = JSON.stringify(result);
} catch (e) {
 console.log(e);
}
resultStr = resultStr.replace(/"parent":"[a-zA-Z\d]+",/g, '');
// --------------- 写入 ---------------
fs.writeFile('./result.json', resultStr, 'utf-8', (err) => {
 if (err) {
 console.log(err);
  } else {
 console.log(`### conver ${config.originType} to ${config.targetType} success! `);
  }
});

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏数据库

使用VBA创建Access数据表

导读: 本期介绍如何在Access数据库中创建一张空数据表。下期将介绍如何将工作表中的数据存入数据库对应的表中,随后还将介绍如何从数据库的表中取出数据输出到Ex...

2597
来自专栏python成长之路

(解释文)My SQL中主键为0和主键自排约束的关系

1945
来自专栏散尽浮华

mysql表名忽略大小写问题记录

问题描述: 一开发同事在linux下调一个程序老是报错说找不到表,但是登陆mysql,show tables查看明明是已经创建了这张表的!!如下: mysql>...

3987
来自专栏精讲JAVA

Java面试之数据库面试题

触发器是一中特殊的存储过程,主要是通过事件来触发而被执行的。它可以强化约束,来维护数据的完整性和一致性,可以跟踪数据库内的操作从而不允许未经许可的更新和变化。可...

1582
来自专栏PingCAP的专栏

TiDB 源码阅读系列文章(四)Insert 语句概览

本文为 TiDB 源码阅读系列文章的第四篇。上一篇文章简单介绍了整体流程,无论什么语句,大体上是在这个框架下运行,DDL 语句也不例外。

4365
来自专栏Java帮帮-微信公众号-技术文章全总结

第二十七天 数据库基础&JDBC使用&工具类Properties&连接池&DBUtils【悟空教程】

第二十七天 数据库基础&JDBC使用&工具类Properties&连接池&DBUtils【悟空教程】

1432
来自专栏Vamei实验室

来玩Play框架05 数据库

数据库是整个站点的数据储藏室。用户提交的数据可以存储在数据库中,以便未来使用。Play可以通过JDBC和数据库通信。我讲介绍Play和mysql数据库的连接。 ...

2349
来自专栏hotqin888的专栏

golang beego orm无限条件查询,多条件查询,不定条件查询,动态多条件查询sql语句,一个字段匹配多值

其实我的需求就是:一个树状目录,每个目录下都存有成果,给定某一个上级目录id,分页查询出这个目录下以及子孙目录下的所有成果,要求分页。

9882
来自专栏程序员的SOD蜜

使用OQL“语言”构造ORM实体类的复杂查询条件

OQL”语言“ 是PDF.NET数据开发框架的实体对象查询语言,一直以来,ORM的复杂查询条件都是困扰ORM的问题,所以很多时候不得不舍弃ORM,直接手工拼接S...

2176
来自专栏Java后端生活

JDBC(三)PreparedStatement

SQL 注入是利用某些系统没有对用户输入的数据进行充分的检查,而在用户输入数据中注入非法的 SQL 语句段或命令,从而利用系统的 SQL 引擎完成恶意行为的做法

821

扫码关注云+社区

领取腾讯云代金券