js递归遍历树形json结构 根据最后一个节点找到整个家族,根据父节点找到所有的子节点,根据层级关系写成树形结构

dataTree: [
        { id: 1, address: "安徽", pid: 0 },
        { id: 6, address: "安徽a", pid: 1 },
        { id: 12, address: "安徽aa", pid: 6 },
        { id: 14, address: "安徽aaa", pid: 12 },
        { id: 15, address: "安徽aaaa", pid: 14 },
        { id: 7, address: "安徽b", pid: 1 },
        { id: 13, address: "安徽bb", pid: 7 },
        { id: 2, address: "江苏", pid: 0 },
        { id: 8, address: "江苏a", pid: 2 },
        { id: 9, address: "江苏b", pid: 2 },
        { id: 3, address: "合肥", pid: 0 },
        { id: 10, address: "合肥a", pid: 3 },
        { id: 11, address: "合肥b", pid: 3 }
      ]

定义好层级关系的数组

//根据最后一个找到整个家族
    familyTree(arr, pid) {
      var temp = [];
      var forFn = function(arr, pid) {
        for (var i = 0; i < arr.length; i++) {
          var item = arr[i];
          if (item.id == pid) {
            temp.push(item);
            forFn(arr, item.pid);
          }
        }
      };
      forFn(arr, pid);
      return temp;
    },
//找到某一父节点下的所有子节点
    sonsTree(arr, id) {
      var temp = [],
        lev = 0;
      var forFn = function(arr, id, lev) {
        for (var i = 0; i < arr.length; i++) {
          var item = arr[i];
          if (item.pid == id) {
            item.lev = lev;
            temp.push(item);
            forFn(arr, item.id, lev + 1);
          }
        }
      };
      forFn(arr, id, lev);
      return temp;
    },
//将数据写成树结构
    toTree(data) {
      // 删除 所有 children,以防止多次调用
      data.forEach(function(item) {
        delete item.children;
      });

      // 将数据存储为 以 id 为 KEY 的 map 索引数据列
      var map = {};
      data.forEach(function(item) {
        map[item.id] = item;
      });
      //        console.log(map);

      var val = [];
      data.forEach(function(item) {
        // 以当前遍历项,的pid,去map对象中找到索引的id
        var parent = map[item.pid];

        // 如果找到索引,那么说明此项不在顶级当中,那么需要把此项添加到,他对应的父级中
        if (parent) {
          (parent.children || (parent.children = [])).push(item);
        } else {
          //如果没有在map中找到对应的索引ID,那么直接把 当前的item添加到 val结果集中,作为顶级
          val.push(item);
        }
      });
      return val;
    },

猜你喜欢

转载自blog.csdn.net/wodebokecsdn/article/details/82851647
今日推荐