js封装更改树形结构指定节点字段的值

封装函数

export function updateNodeField (tree, targetNodeId, fieldName, fieldValue) {
    if (!tree || typeof tree !== 'object') {
        throw new Error('Invalid tree object');
    }

    // 创建一个新的树形结构对象
    const updatedTree = {};

    // 复制原始树的属性到更新后的树
    for (const key in tree) {
        if (tree.hasOwnProperty(key)) {
            updatedTree[key] = tree[key];
        }
    }

    // 遍历树的每个节点
    for (const node of Object.values(updatedTree)) {
        // 如果当前节点是目标节点，则更新字段值
        if (node.id === targetNodeId) {
            node[fieldName] = fieldValue;
        }

        // 如果当前节点有子节点，则递归调用 updateNodeField 方法
        if (node.children && node.children.length > 0) {
            node.children = updateNodeField(node.children, targetNodeId, fieldName, fieldValue);
        }
    }

    return updatedTree;
}

描述

这段代码实现了一个树形结构的节点字段更新功能。给定一个树形结构对象，目标节点的ID，字段名和字段值，函数会在树中找到目标节点，并将其对应字段的值更新为指定的值。如果目标节点有子节点，函数会递归调用自身来更新子节点。

• 例如，假设我们有以下的树形结构对象：

const tree = {
    1: { id: 1, name: 'Node 1', children: [2, 3] },
    2: { id: 2, name: 'Node 2', children: [] },
    3: { id: 3, name: 'Node 3', children: [4] },
    4: { id: 4, name: 'Node 4', children: [] }
};

我们想要将节点ID为2的节点的"name"字段更新为"Updated Node 2"，可以使用该函数进行更新：

const updatedTree = updateNodeField(tree, 2, 'name', 'Updated Node 2');

• 更新后的树形结构对象updatedTree将变为：

{
    1: { id: 1, name: 'Node 1', children: [2, 3] },
    2: { id: 2, name: 'Updated Node 2', children: [] },
    3: { id: 3, name: 'Node 3', children: [4] },
    4: { id: 4, name: 'Node 4', children: [] }
}