上篇文章讲到如何基于antd tree组件实现联动选择，即根据增删改选中list。其中提到通过选中的节点去找到父节点，然后根据父节点key拼接_list，判断新的key是否在checkedKeys中，如果在则不添加，不在的话就需要添加。
今天在做其他的需求时，发现这块不好使了，细看之下才发现，在一些情况些，比如修改个人信息操作，他有两个操作，一个是获取用户信息，一个是修改用户信息。当然当用户勾选了修改用户信息的节点，应该把获取用户信息的节点也给勾选上。他不需要list节点，用的是获取用户信息即xxxxxx_info作为node节点的key值，这就导致了不能通过上篇文章的方式去添加。
所以，在这里需要对代码进行改进。不能使用拼接的方式了，因为后台的节点key可以千变万化。
但是我们可以跟后台约定一种规则，比如：权限最小的放在第一个节点，比如获取用户信息，获取列表信息，这种权限最小的节点(所有的后续操作，都必须在第一个操作上。比如增删改一定要在list的基础上，否则你修改也没地方看)放在第一个。
那我们就可以去获取当前选中的节点同级的第一个节点，即选中节点父节点下的第一个子节点。
新写一个工具
import {map } from 'lodash'/*** * @param {*} tree 当前遍历的tree* @param {*} key 当前选中的节点父节点*/let first_key = ""export const getFirstNode = (tree, key) => {map(tree, (node) => {// 判断当前节点是否是父节点if (node.key === key) {first_key = node.children[0].keyreturn} else {// 如果匹配不到，那说明匹配的父节点不在当前外层节点中，需要遍历子节点if (node.children) {getFirstNode(node.children, key)}}})return first_key}
之前的代码修改后
// 选中的节点，如果选中某节点下的叶子节点则默认选中列表list节点onCheck = (checkedKeys, checked) => {// 每个模块下的第一个节点，lsit/info/result等const nodes = checked.checkedNodes// 排除checkedKey中的非叶子节点checkedKeys = _.filter(checkedKeys, (key) => {return !_.find(nodes, ["key", key]).children})console.log("checkedKeys", checkedKeys)_.map(nodes, node => {if (!node.children) {// 根据父节点推导出第一个子节点const parent = node.parentconst first_key = getFirstNode(this.state.treeData, parent)console.log("first_key", first_key)/*** 1.判断当前选中的节点是不是list节点* 2.如果是，则不添加* 3.如果不是则判断checkedKeys中是否存在list节点* 4.如果不存在则添加，如果存在则不添加* 5.如果选中任一个不是list节点，则选中list节点，且不可取消勾选list节点* 6.如果想取消list节点，则必须取消其他的节点勾选状态*/if (first_key !== node.key) {if (_.findIndex(checkedKeys, function (o) {return o === first_key; }) < 0) {checkedKeys.push(first_key)}}}})};
有疑问，请留言！
