#

luxiaotao1123

2024-02-19 0efe01c1702e336f7c46435bbe30ecd7a1af5076

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import { DataNode } from 'antd/es/tree';
import { parse } from 'querystring';
 
/**
 * 构造树型结构数据
 * @param {*} data 数据源
 * @param {*} id id字段 默认 'id'
 * @param {*} parentId 父节点字段 默认 'parentId'
 * @param {*} children 孩子节点字段 默认 'children'
 */
export function buildTreeData(data, id, name, parentId, parentName, children) {
  const config = {
    id: id || 'id',
    name: name || 'name',
    parentId: parentId || 'parentId',
    parentName: parentName || 'parentName',
    childrenList: children || 'children',
  };
 
  const childrenListMap = [];
  const nodeIds = [];
  const tree = [];
  data.forEach((item) => {
    const d = item;
    const pId = d[config.parentId];
    if (!childrenListMap[pId]) {
      childrenListMap[pId] = [];
    }
    d.key = d[config.id];
    d.title = d[config.name];
    d.value = d[config.id];
    d[config.childrenList] = null;
    nodeIds[d[config.id]] = d;
    childrenListMap[pId].push(d);
  });
 
  data.forEach((item) => {
    const d = item;
    const pId = d[config.parentId];
    if (!nodeIds[pId]) {
      d[config.parentName] = '';
      tree.push(d);
    }
  });
 
  function adaptToChildrenList(item) {
    const o = item;
    if (childrenListMap[o[config.id]]) {
      if (!o[config.childrenList]) {
        o[config.childrenList] = [];
      }
      o[config.childrenList] = childrenListMap[o[config.id]];
    }
    if (o[config.childrenList]) {
      o[config.childrenList].forEach((child) => {
        const c = child;
        c[config.parentName] = o[config.name];
        adaptToChildrenList(c);
      });
    }
  }
 
  tree.forEach((t) => {
    adaptToChildrenList(t);
  });
 
  return tree;
}
 
export const getPageQuery = () => parse(window.location.href.split('?')[1]);
 
export function formatTreeData(arrayList) {
  const treeSelectData = arrayList.map((item) => {
    const node = {
      id: item.id,
      title: item.label,
      key: `${item.id}`,
      value: item.id,
    };
    if (item.children) {
      node.children = formatTreeData(item.children);
    }
    return node;
  });
  return treeSelectData;
}