阅读 901

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

前言

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

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

具体的效果图如下:

screenshots

主要的就是借助 TreeNodetitle 属性,它的类型是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,欢迎有同样需要的小伙伴参考,starfork 也是极好的。

关注下面的标签,发现更多相似文章
评论