怎么定制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写入树形组件的数据源上：
<DirectoryTree style={{ width: 280 }} draggable onDrop={onDrop} defaultExpandAll onRightCpck={({ node }) => setRightCpckKey(node.key)} onSelect={onSelect} selectedKeys={rightConnect ? [activeTabKey] : selectedKeys} onExpand={onExpand} treeData={[ …deepTree(treeData, (item) => { return { …item, titleWord: item.title, title: ( <Dropdown trigger=”contextMenu” visible={rightCpckKey === item.key} onVisibleChange={() => setRightCpckKey()} overlayStyle={{ width: 80 }} overlay={menu(item)} > <div style={ searchWord && item.title.includes(searchWord) ? { color: ‘red’ } : {} } > {item.title} </div> </Dropdown> ), }; }), ]} />复制代码
关于右键菜单有几点需要补充说明一下：
Dropdown 的触发属性需要设置成contextMenu;
Dropdown 显示的位置是相对于title而言，需要设置外层容器宽度铺满剩余空间：
.ant-tree-node-content-wrapper { display: flex;}.ant-tree-title { flex: 1;}复制代码
Dropdown 的显示藏是通过右键点击记录的key来判断的；
Dropdown 的菜单需要传递当前item;
const menu = (node) => ( <Menu onCpck={({ key, domEvent }) => { domEvent.stopPropagation(); console.log(‘menuCpck’, 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; } }} > <Menu.Item key=”add”>新增</Menu.Item> <Menu.Item key=”delete” danger> 删除 </Menu.Item> <Menu.Item key=”edit”>编辑</Menu.Item> </Menu> );复制代码
添加/修改/删除功能
添加功能默认只能给文件夹添加，通过key值判断添加，这里处理的比较简单，只做核心功能演示，代码见上一小节；
修改功能也做了简单的实例，在正式项目中一般需要弹窗编辑或者在树组件的title中嵌入输入框，可以使用变量记录正在编辑的item, 最后保存通过递归插入到树形数据中：
删除功能做了判断，如果是删除最外层，则直接通过filter过滤，⚠️否则删除功能是通过children来过滤的，这里要特别注意下。
搜索功能
搜索功能是通过titile颜色变红来提示的：
实现上也只是做了点击搜索之后搜索，没有实时搜索提示，也没有做搜索词区分，这里可以再截取下字符串来实现，可以见官方实例，注意这个默认打开父节点的属性autoExpandParent，否则可能要费些功夫向上递归。
还有一种需求是要过滤数据源，可以对官方实例简单改造后实现；
Tab反向定位
点击Tree组件item，在右侧添加Tab，或者激活Tab,这可以算是正向定位；那反向定位就是当右侧Tab页切换时左侧Tree组件选中对应item，核心代码也就是指定selectedKeys，相比较而言也不难，难点在默认打开相关父节点，当然前面说过了控制好autoExpandParent这个属性，就好了。
拖拽移动
拖拽移动一是Tree组件本身支持，二是官方已经给出了拖拽移动实例，我也只是在官方实例稍微做了改造，这里也不多赘述：
结束
搜索和反向定位的难点其实是在，打开关联文件夹上，不过官方实例中使用了autoExpandParent这个属性，一下子简单了很多。
时候也不早了，今天就到这里了。
以上就是Ant Design创建一个树形组件，实现编辑、搜索和定位功能的详细内容，更多请关注24小时课堂在线其它相关文章！
