Tree组件的快速定位更新节点的状态,以及修改节点的数据属性等操作

Evisu47（常安欧阳）2024-08-22 17:18

当我们点击树节点的时候我们常常只能获得树的id，那么我么如何获快速定位到树节点的内容呢，除此之外，当树已经存在时，但是缺少我们想要的内容时，我们想在树节点上添加我们需要的额外的内容时该怎么办，那么就是用以下方法可以快速定位到我们需要的节点并可以快速添加内容

JS 复制代码

/**
* @params
*     data：树节点
*     key：需要传的id（具体就是树节点唯一值就可以具体对于树而言）
*     callback：更新节点的状态,收集特定类型的节点信息,修改节点的数据属性等。
*       node：当前节点的所有数据
*       i：这个索引帮助我们理解
*       data：这个参数提供了当前层级所有节点的信息，可以用于上下文相关的操作。当前节点在其兄弟节点中的相对位置。
*       parent：允许访问父节点的信息，对于构建层级关系非常有用
*     parentNode：它表示当前节点的父节点。这个参数在递归调用 treeLoop 时传递给下一层，以便下一层的节点能够知道它们的父节点是谁。
*/

  const treeLoop = (
    data: treeDataNode[], 
    key: React.Key | null,
    callback: (node: treeDataNode, i: number, data: treeDataNode[], parent?: treeDataNode | null) => void,
    parentNode?: treeDataNode | null,
  ) => {
    for (let i = 0; i < data.length; i++) {
      if (key == null) {
        callback(data[i], i, data, parentNode)
        if (data[i].children) {
          treeLoop (data[i].children!, key, callback, data[i], type)
        }
      } else {
        if (data[i].id === key) {
          return callback(data[i], i, data, parentNode)
        }
        if (data[i].children) {
          treeLoop (data[i].children!, key, callback, data[i], type)
        }
      }
    }
  }

treeLoop 函数是一个通用的树形结构遍历工具，它可以用于多种场景，如查找特定节点、更新节点状态等。通过传入不同的 callback 函数，可以实现不同的功能。当 key 为 null 时，callback 会在遍历到每一个节点时被调用；而当 key 不为空时，只有在找到与 key 匹配的节点时才会调用 callback。