javascript - 使用 JavaScript 中的一个函数从数组和对象中获取键

Question

我需要一个函数来从对象中获取键，并从数组中获取条目，如下所示:

_key(["x", "y", "z"]); // ["x", "y", "z"];
_key({ a: "x", b: "y", c: "z" }); // ["a", "b", "c"];

有没有内置的方法可以做到这一点，如果没有，最短的方法是什么？

Answer 1

你可以先用Array.isArray()测试它是否是数组

function _key(data) {
  return Array.isArray(data) ? data : Object.keys(data);
}

console.log(_key(["x", "y", "z"]))
console.log(_key({ a: "x", b: "y", c: "z" }))

如果您需要传递多个参数、对象和数组，您可以使用剩余参数语法和 map() 返回一个数组。

function _key(...data) {
  return [].concat(...data.map(e => Array.isArray(e) ? e : Object.keys(e)))
}

console.log(_key(["x", "y", "z"], { a: "x", b: "y", c: "z" }, {z: 1}))