const treeData = [
{
title: '北京市',
key: '0-0',
children: [
{
title: '海淀区',
key: '0-0-0',
children: [
{ title: '中关村', key: '0-0-0-0' },
{ title: '后厂村', key: '0-0-0-1' },
{ title: '西二旗', key: '0-0-0-2' },
],
},
{
title: '朝阳区',
key: '0-0-1',
children: [
{ title: '东单', key: '0-0-1-0' },
{ title: '朝阳门', key: '0-0-1-1' },
{ title: '望京', key: '0-0-1-2' },
],
},
{
title: '丰台区',
key: '0-0-2',
},
],
},
{
title: '大连市',
key: '0-1',
children: [
{ title: '0-1-0-0', key: '0-1-0-0' },
{ title: '0-1-0-1', key: '0-1-0-1' },
{ title: '0-1-0-2', key: '0-1-0-2' },
],
},
{
title: '哈尔滨市',
key: '0-2',
},
];
// 定义树结构映射
const fields = {
title: 'title',
key: 'key',
parentId: 'parentId',
children: 'children',
}
/**
* 树关键字搜索,返回匹配的节点key及其父节点的key
* @param {Object} fields 字段映射
* @param {Array} data 树data
* @param {String} value 关键字值
* @param {String} condition 条件(key/title等及其映射)
* @returns {Array} 符合条件的所有key
*/
const getSearchKey = (fields, data, value,condition) => {
let result = {
flag: false,
keyArr: [],
};
for (let i = 0; i < data.length; i++) {
let item = data[i];
let children = item[fields.children];
if (children) {
let tmpResult = getSearchKey(fields, children, value,condition);
if (tmpResult.flag) {
result.keyArr.push(item[fields[condition]]);
result.flag = true;
}
result.keyArr = result.keyArr.concat(tmpResult.keyArr);
}
if (value && item[fields[condition]] === value) {
result.flag = true;
result.keyArr.push(item[fields[condition]]);
}
}
return result;
};
输出案例:
console.log(getSearchKey(fields,treeData,'望京','title'));
// { flag: true, keyArr: [ '北京市', '朝阳区', '望京' ] }
console.log(getSearchKey(fields,treeData,'0-0-0-1','key'));
// { flag: true, keyArr: [ '0-0', '0-0-0', '0-0-0-1' ] }