如何通过在对象数组中嵌套对象键来获取值

Question

我写了一个函数，它获取泛型对象的数据和应该返回的值的属性(键)：

resultToArray<T>(
data: T[],
prop: (keyof T | ((row: T) => string))[] = []
): string[][] {
// data: [{id: 1, name: 'first'}, {id: 2, name: 'second'}] -> [[1, 'first'], [2, 'second']]
// prop: ['id', 'name']
if (prop.length === 0) {
return data.map(Object.values);
} else {
const output = data.map(d => {
const a: string[] = [];
for (let i = 0; i < prop.length; i++) {
a.push(d[prop[i]])
}
return a;
});
return output
}
}

示例输入：
data-

[ 
{ name: 'Name', address: 'Address', branch: { locale: 'Locale' } },
{ name: 'Name2', address: 'Address2', branch: { locale: 'Locale2' } }
]

prop-[ 'name', (d) => d.branch.locale ]

输出应为[['Name', 'Locale'], ['Name2', 'Locale2']]

函数对简单键工作正常，但对嵌套键不工作

Answer 1

我相信一个简单的解决方案(用纯js测试)是检查prop[I]的类型并拆分所使用的功能。

for (let i = 0; i < prop.length; i++) {
if (typeof prop[i] === 'string') {
a.push(d[prop[i]]);
} else {
a.push(prop[i](d));
}
}

您必须将数据作为参数来运行函数，才能实际获得输出。

输入：

resultToArray(
[
{ name: 'Name', address: 'Address', branch: { locale: 'Locale' , extra: { data: 'Country1' } } },
{ name: 'Name2', address: 'Address2', branch: { locale: 'Locale2', extra: { data: 'Country2' } } }
],
[ 'name', (d) => d.branch.locale, (d) => d.branch.extra.data ]
);

输出：

[
[ 'Name', 'Locale', 'Country1' ],
[ 'Name2', 'Locale2', 'Country2' ]
]

Answer 2

您需要检查；道具"；您想要提取的是一个函数并调用它。

for (let i = 0; i < prop.length; i++) {
const propToUse = prop[i];
let value;
if (typeof propToUse === 'function') {
value = propToUse(d);
} else {
value = d[propToUse];
}
a.push(value);
}