Ant Design创建一个树形组件，实现编辑、搜索和定位功能

怎么定制ant design树形组件实现编辑、搜索和反向定位功能？下面本篇文章给大家介绍一下创建树形组件，实现这些功能的方法，希望对大家有所帮助！

这次在做了一个树形的展示功能，谁知产品意犹未尽，找我谈话：

PD: 什么？只有展开收起功能？这怎么行，咱们最基础的要支持编辑，支持搜索，如果可以的话还可以做个反向定位…

YY: 你咋不早说？需求文档上也没有啊…

PD: 你看谁家文档一次写到位的？哪家的PD不加需求？

YY: 话是这样说，可事情不是这么做的…

PD: 哎呀，别杵着浪费时间了，快去做吧！

YY: …

以上故事纯属虚构，如有雷同请评论区留言…

树形数据在开发中算是比较常见了，文件夹、组织架构、生物分类、国家地区等等，世间万物的大多数结构都是树形结构。使用树控件可以完整展现其中的层级关系，并具有展开收起选择等交互功能。

需求分析

编辑：添加/修改/删除/移动搜索功能：名称/创建人/ owner过滤定位：tab反向定位

项目仓库：https://github.com/speakice/editable-tree

功能实现

能实现以上功能的方法库和组件有很多种，这里只讲其中一种，都是Ant Design的组件：

Tree.DirectoryTree 目录树Dropdown 右键菜单容器Menu 菜单内容Tabs 右侧Tab页Input.Search 搜索框Switch 切换关联状态shortid 生成唯一id

import { Tree, Dropdown, Menu, Tabs, Input, Switch } from 'antd';import shortid from 'shortid';复制代码

登录后复制

递归方法

操作树行数据，最重要的前提是要有一个趁手的递归方法：

/** * 如果需要修改tree，action就返回修改后的item， 不修改就不返回 */export const deepTree = (tree = [], action = () => {}) => {  return tree.map((item) => {    const newItem = action({ ...item }) || item;    if (newItem.children) {      newItem.children = deepTree(newItem.children, action);    }    return newItem;  });};复制代码

登录后复制

鼠标右键菜单

右键菜单作用在title上，需要把Dropdown写入树形组件的数据源上：

     setRightClickKey(node.key)}          onSelect={onSelect}          selectedKeys={rightConnect ? [activeTabKey] : selectedKeys}          onExpand={onExpand}          treeData={[            ...deepTree(treeData, (item) => {              return {                ...item,                titleWord: item.title,                title: (                   setRightClickKey()}                    overlayStyle={{ width: 80 }}                    overlay={menu(item)}                  >                    
                      {item.title}                    
                                  ),              };            }),          ]}        />复制代码

登录后复制

关于右键菜单有几点需要补充说明一下：

Dropdown 的触发属性需要设置成contextMenu;Dropdown 显示的位置是相对于title而言，需要设置外层容器宽度铺满剩余空间：

.ant-tree-node-content-wrapper {  display: flex;}.ant-tree-title {  flex: 1;}复制代码

登录后复制Dropdown 的显示藏是通过右键点击记录的key来判断的；Dropdown 的菜单需要传递当前item;

  const menu = (node) => (     {        domEvent.stopPropagation();        console.log('menuClick', node, key);        // 如果要添加操作顶层文件夹,可以直接操作        switch (key) {          case 'add':            setTreeData(              deepTree(treeData, (item) => {                if (item.children && item.key === node.key) {                  return {                    ...item,                    children: [                      ...item.children,                      {                        title: 'new add',                        key: shortid.generate(),                        isLeaf: true,                      },                    ],                  };                }              })            );            break;          case 'delete':            const outer = treeData.find((item) => item.key === node.key);            if (outer) {              setTreeData(treeData.filter((item) => item.key !== node.key));              return;            }            setTreeData(              deepTree(treeData, (item) => {                if (item.children) {                  return {                    ...item,                    children: item.children.filter(                      ({ key }) => key !== node.key                    ),                  };                }                return item;              })            );            break;          case 'edit':            setTreeData(              deepTree(treeData, (item) => {                if (item.key === node.key) {                  console.log('editle', {                    ...item,                    title: 'new edit',                  });                  return {                    ...item,                    title: 'new edit',                  };                }                return item;              })            );            break;        }      }}    >      新增              删除            编辑      );复制代码

登录后复制

添加/修改/删除功能

添加功能默认只能给文件夹添加，通过key值判断添加，这里处理的比较简单，只做核心功能演示，代码见上一小节；

修改功能也做了简单的实例，在正式项目中一般需要弹窗编辑或者在树组件的title中嵌入输入框，可以使用变量记录正在编辑的item, 最后保存通过递归插入到树形数据中：

删除功能做了判断，如果是删除最外层，则直接通过filter过滤，⚠️否则删除功能是通过children来过滤的，这里要特别注意下。

搜索功能

搜索功能是通过titile颜色变红来提示的：

实现上也只是做了点击搜索之后搜索，没有实时搜索提示，也没有做搜索词区分，这里可以再截取下字符串来实现，可以见官方实例，注意这个默认打开父节点的属性autoExpandParent，否则可能要费些功夫向上递归。

还有一种需求是要过滤数据源，可以对官方实例简单改造后实现；

Tab反向定位

点击Tree组件item，在右侧添加Tab，或者激活Tab,这可以算是正向定位；那反向定位就是当右侧Tab页切换时左侧Tree组件选中对应item，核心代码也就是指定selectedKeys，相比较而言也不难，难点在默认打开相关父节点，当然前面说过了控制好autoExpandParent这个属性，就好了。

拖拽移动

拖拽移动一是Tree组件本身支持，二是官方已经给出了拖拽移动实例，我也只是在官方实例稍微做了改造，这里也不多赘述：

结束

搜索和反向定位的难点其实是在，打开关联文件夹上，不过官方实例中使用了autoExpandParent这个属性，一下子简单了很多。

时候也不早了，今天就到这里了。

更多编程相关知识，请访问：编程视频！！

以上就是Ant Design创建一个树形组件，实现编辑、搜索和定位功能的详细内容，更多请关注【创想鸟】其它相关文章！