基于Ant Design的可编辑Tree的实现

前言

最近在用Ant Design写一个后台，遇到的需求就是实现一个可动态增减和编辑子节点的Tree。GitHub上看了一圈，没好用和合适的。索性就基于Ant Design中的Tree组件写一个。实现的效果如下：

可以增加子节点
可以删除子节点
可以编辑子节点信息
可以取消编辑信息

具体的效果图如下：

主要的就是借助 TreeNode 的 title 属性，它的类型是string|ReactNode。

正文

经过分析，一个节点的数据结构应该是

{
    value: 'Root', // 显示的信息
    defaultValue: 'Root', // 当某一节点进入编辑状态，然后点击close按钮，节点的信息应该恢复原始状态，
    key: '0-1', // 节点的Key，全局唯一
    parentKey: '0', // 父节点的Key
    isEditable: false // 是否处于可编辑状态
    children:[] // 子节点
}

通过数据结构组装TreeNode的代码如下：

    data= [
      {
        value: 'Root',
        defaultValue: 'Root',
        key: '0-1',
        parentKey: '0',
        isEditable: false
      }
    ];
    
    state={
        data: this.data
    }


renderTreeNodes = data => data.map((item) => {
    if (item.isEditable) {  // 编辑状态下
      item.title = (
        <div>
          <input value={item.value}
            onChange={(e) => this.onChange(e, item.key)}/>
          <Icon type='close'  style={{marginLeft:10}} onClick={() => this.onClose(item.key, item.defaultValue)}/>
          <Icon type='check'  style={{marginLeft:10}} onClick={() => this.onSave(item.key)}/>
        </div>
      );
    } else {
      item.title = (
        <div>
          <span>
            {item.value}
          </span>
          <Icon style={{ marginLeft: 10 }} type='edit' onClick={() => this.onEdit(item.key)} />
          <Icon style={{ marginLeft: 10 }} type='plus' onClick={() => this.onAdd(item.key)} />
          {item.parentKey === '0' ? null : (<Icon style={{ marginLeft: 10 }} type='minus' onClick={() => this.onDelete(item.key)} />)}  // 根节点没有删除按钮
        </div>
      )
    }

    if (item.children) {
      return (
        <TreeNode title={item.title} key={item.key} dataRef={item}>
          {this.renderTreeNodes(item.children)}
        </TreeNode>
      );
    }

    return <TreeNode {...item}/>;
  })
  
  ...
  
  // 渲染界面
  render() {
    return (
      <div>
        <Tree>
          {this.renderTreeNodes(this.state.data)}
        </Tree>
      </div>
    )
  }

之后所有的增删修改等都是先修改data这个数组中的数据，然后使用this.setState({ data: this.data })更新界面，具体的看下代码就成，很简单。

最后优化这个组件的时候，遇到一个比较坑的。本来想是当在某节点上增加子节点时，父节点自动展开，代码逻辑上没有问题，但是必须手动执行过一次展开或者搜索的操作，所写的逻辑才能生效。后来没办法，只能在生命周期函数中DOM加载完毕后主动触发下：

componentDidMount() {
    this.onExpand([]); // 手动触发，否则会遇到第一次添加子节点不展开的Bug
  }

代码放在GitHub上了，地址是 react-editable-tree，欢迎有同样需要的小伙伴参考，star和fork 也是极好的。