xxm-test-js/dist/lib/convertToTree.js

xxm-js通用js工具(utils)库

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.convertToTree = convertToTree;
/**
 * 将扁平数据结构转换为树形结构。
 *
 * @template T 泛型类型，表示节点对象的类型。
 *
 * @param {T[]} flatData - 扁平数据数组。
 * @param {TreeOptions<T>} options - 树形结构的配置选项。
 * @returns {Array<TreeNode<T>>} 转换后的树形结构数组。
 *
 * @throws {Error} 如果存在重复 ID 或找不到父节点时抛出错误。
 *
 * @example
 * ```typescript
 * // 假设我们有以下扁平数据结构：
 * interface DataItem {
 *   id: number;
 *   name: string;
 *   parentId?: number | null;
 * }
 *
 * const flatData: DataItem[] = [
 *   { id: 1, name: 'Root', parentId: null },
 *   { id: 2, name: 'Child 1', parentId: 1 },
 *   { id: 3, name: 'Child 2', parentId: 1 },
 *   { id: 4, name: 'Grandchild 1', parentId: 2 },
 *   { id: 5, name: 'Grandchild 2', parentId: 2 },
 *   { id: 6, name: 'Another Root', parentId: null }
 * ];
 *
 * const options: TreeOptions<DataItem> = {
 *   idKey: 'id',
 *   parentIdKey: 'parentId',
 *   rootId: null,
 * };
 *
 * // 使用 convertToTree 函数将扁平数据转换为树形结构
 * const treeData = convertToTree(flatData, options);
 * console.log(JSON.stringify(treeData, null, 2));
 * ```
 */
function convertToTree(flatData, options) {
    const { idKey, parentIdKey, rootId } = options;
    const map = new Map();
    const tree = [];
    // 检查是否有重复ID
    flatData.forEach((item) => {
        if (map.has(item[idKey])) {
            throw new Error(`Duplicate ID found: ${item[idKey]}`);
        }
        const node = Object.assign(Object.assign({}, item), { children: [] });
        map.set(item[idKey], node);
    });
    // 建立父子关系
    flatData.forEach((item) => {
        var _a;
        const node = map.get(item[idKey]);
        if (item[parentIdKey] === rootId) {
            tree.push(node); // 根节点直接添加到树中
        }
        else {
            const parent = map.get(item[parentIdKey]);
            if (parent) {
                (_a = parent.children) === null || _a === void 0 ? void 0 : _a.push(node); // 添加到父节点的子节点列表
            }
            else {
                console.warn(`Parent not found for item with ID: ${item[idKey]}`);
            }
        }
    });
    return tree;
}
//# sourceMappingURL=convertToTree.js.map