antd的Tree控件实现点击展开功能

antd 的 Tree 控件没有提供点击展开的功能,只能通过左边的三角形实现展开和收起,没办法只好自己实现这个功能。

截图.gif

如图实现的是类似 Mac 文件目录形式的结构,有箭头代表是个文件夹,点击展开文件夹,点击外层文件夹可以收起整个文件夹。

首先根据服务器返回的 Json 数据生成树形结构

const data = {
    name: "root",
    children: [{
        name: "a",
        value: "/a",
        children: [{
            name: "file_1",
            value: "/a/file_1"
        }, {
            name: "a_1",
            value: "/a/a_1",
            children: [{
                name: "file_2",
                value: "/a/a_1/file_2"
            }, {
                name: "file_3",
                value: "/a/a_1/file_3"
            }]
        }, {
            name: "a_2",
            value: "/a/a_2",
            children: [{
                name: "file_4",
                value: "/a/a_2/file_4"
            }, {
                name: "file_5",
                value: "/a/a_2/file_5"
            }]
        }]
    }, {
        name: "b",
        value: "/b",
        children: [{
            name: "b_1",
            value: "/b/b_1",
            children: [{
                name: "file_6",
                value: "/b/b_1/file_6"
            }]
        }]
    }]
}

数据结构中有 children 字段代表是个文件夹,因此需要递归遍历出 TreeNode 视图

import React from "react";
import Tree from 'antd/lib/tree';
import 'antd/lib/tree/style/css';

const TreeNode = Tree.TreeNode;

class TreeView extends React.Component {
    //遍历json绘制出tree结构
    mapData = (children) => {
        if (children && Array.isArray(children)) {
            return children.map((ele) => {
                if (ele.children && Array.isArray(ele.children)) {
                    return <TreeNode title={ele.name} key={ele.value}>
                        {this.mapData(ele.children)}
                    </TreeNode>
                } else {
                    return <TreeNode title={ele.name} key={ele.value}/>
                }
            })
        }
        return []
    }

    render() {
        let content = []

        let {name, children} = data
        if (name) {
            content.push(<TreeNode title={name} key="/">{this.mapData(children)}</TreeNode>)
        }

        return (
            <Tree>
                {content}
            </Tree>
        );
    }
}

export default TreeView

已经完成第一步,生成树形结构视图。接下来就是要实现点击展开树形结构

   state = {
        expandedKeys: ["/"],
        autoExpandParent: true,
        selectedKeys: []
    }

    //选中的回调
    onSelect = (selectedKeys, obj) => {
        let {expandedKeys} = this.state
        let selectedKey = this.state.selectedKeys
        //选中的状态
        if (obj.selected) {
            //判断是否已经展开,未展开就添加到 expandedKeys 中
            //已经展开就不用管
            let index = expandedKeys.indexOf(selectedKeys[0])
            if (index === -1) {
                expandedKeys.push(selectedKeys[0])
                this.setState({
                    expandedKeys: expandedKeys,
                    selectedKeys: selectedKeys
                })
            } else {
                this.setState({
                    selectedKeys: selectedKeys
                })
            }

            // 没有 children 代表当前已没有下一级目录
            if (obj.event && obj.selectedNodes.length === 1 && !obj.selectedNodes[0].props.children) {
                //do something
            }
        } else {
            //selectedKey 是上次选中的元素 在 expandedKeys 肯定是存在的 
            //找到下标后需要过滤掉子类目录 例如上次选中的元素为 /a ,
            //子类 /a/a_1 已经展开就需要从 expandedKeys 中移除这个元素
            let index = expandedKeys.indexOf(selectedKey[0])
            if (index !== -1) {
                //过渡掉子类元素
                expandedKeys = expandedKeys.filter((ele) => {
                    return !ele.includes(selectedKey[0])
                })
                this.setState({
                    expandedKeys: expandedKeys,
                    selectedKeys: []
                })
            } else {
                this.setState({
                    selectedKeys: []
                })
            }
        }
    }

    //展开的回调
    onExpend = (expandedKey, obj) => {
        let {expandedKeys} = this.state
        //展开的状态
        if (obj.expanded) {
            this.setState({
                expandedKeys: expandedKey,
                selectedKeys: []
            })
        } else {
            //expandedKey 返回的是当前已经展开的元素 expandedKeys 是上次展开的元素
            //比较两个数组中缺少的元素得出当前被收起的是哪个元素
            let removeArray = this.diffArray(expandedKey, expandedKeys)
            //收起的时候需要把里面展开的元素一并移除,不然会造成收起动作无效
            expandedKeys = expandedKeys.filter((ele) => {
                return !ele.includes(removeArray[0])
            })
            this.setState({
                expandedKeys: expandedKeys,
                selectedKeys: []
            })
        }
    }

    //比较出2个数组中不一样的元素
    diffArray = (arr1, arr2) => {
        let arr3 = [];
        for (let i = 0; i < arr1.length; i++) {
            if (arr2.indexOf(arr1[i]) === -1)
                arr3.push(arr1[i]);
        }
        for (let j = 0; j < arr2.length; j++) {
            if (arr1.indexOf(arr2[j]) === -1)
                arr3.push(arr2[j]);
        }
        return arr3;
    }
    ......
    render() {
        return (
            <Tree
                onExpand={this.onExpend}
                expandedKeys={this.state.expandedKeys}
                autoExpandParent={this.state.autoExpandParent}
                onSelect={this.onSelect}
                selectedKeys={this.state.selectedKeys}>
                {content}
            </Tree>
        )
    }

以上就可以实现点击展开和收起功能
完整源码地址

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,185评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,445评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,684评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,564评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,681评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,874评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,025评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,761评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,217评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,545评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,694评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,351评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,988评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,778评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,007评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,427评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,580评论 2 349

推荐阅读更多精彩内容