将同级数据处理成树形数据

将同级数据处理成树形数据

  <script>
    const oldArray = [
      {
        batchNo: "1",
        factoryCnName: "苹果",
        goodsCnName: "pingguo",
        inputNum: 1,
      },
      {
        batchNo: "1",
        factoryCnName: "苹果",
        goodsCnName: "pingguo",
        inputNum: 3,
      },
      {
        batchNo: "1",
        factoryCnName: "梨子",
        goodsCnName: "lizi",
        inputNum: 3,
      },
      {
        batchNo: "1",
        factoryCnName: "梨子",
        goodsCnName: "lizi",
        inputNum: 3,
      },
      {
        batchNo: "2",
        factoryCnName: "苹果",
        goodsCnName: "pingguo",
        inputNum: 3,
      },
    ];
    const inStoreTabData = TabFormartFun(oldArray);
    console.log(inStoreTabData, "inStoreTabData");
    //处理出入库表格数据
    function TabFormartFun(arr) {
      const data1 = dataformart1(arr, "goodsCnName");
      const data2 = dataformart2(data1, "factoryCnName");
      const result = dataformart3(data2, "batchIdNo");
      return result;
    }
    //格式化表格数据
    function dataformart1(beforeData, name) {
      var tempArr = [];
      var afterData = [];
      for (var i = 0; i < beforeData.length; i++) {
        if (tempArr.indexOf(beforeData[i][name]) == -1) {
          afterData.push({
            Name: beforeData[i][name],
            show: i == 0 || name == "batchIdNo" ? true : false,
            children: [beforeData[i]],
          });
          tempArr.push(beforeData[i][name]);
        } else {
          for (var j = 0; j < afterData.length; j++) {
            if (afterData[j].Name == beforeData[i][name]) {
              afterData[j].children.push(beforeData[i]);
              break;
            }
          }
        }
      }
      return afterData;
    }
    function dataformart2(beforeData, name) {
      beforeData.forEach((element, index) => {
        var afterData = [];
        var tempArr = [];
        var ele = element.children;
        var itemArr = dataformart1(ele, name);
        beforeData[index].children = itemArr;
      });
      return beforeData;
    }
    function dataformart3(beforeData, name) {
      beforeData.forEach((element, index) => {
        var afterData = [];
        var tempArr = [];
        var ele = element.children;
        var itemArr = dataformart2(ele, name);
        //计算数量
        itemArr.forEach((eve) => {
          eve.children.forEach((item) => {
            var total = 0;
            item.children.forEach((e) => {
              total = total + (e.inputNum - 0);
            });
            item.total = total;
          });
        });
        beforeData[index].children = itemArr;
      });
      return beforeData;
    }
  </script>