JavaScript 终于原生支持数组分组了!
在日常开发中,很多时候需要对数组进行分组,每次都要手写一个分组函数,或者使用lodash的groupBy
函数。好消息是,JavaScript 现在正在引入全新的分组方法:Object.groupBy
和Map.groupBy
,以后再也不需要手写分组函数了,目前最新版本的 Chrome(117)已经支持了这两个方法!
以前的数组分组
假设有一个由表示人员的对象组成的数组,需要按照年龄进行分组。可以使用forEach
循环来实现,代码如下:
const people = [
{ name: "Alice", age: 28 },
{ name: "Bob", age: 30 },
{ name: "Eve", age: 28 },
];
const peopleByAge = {};
people.forEach((person) => {
const age = person.age;
if (!peopleByAge[age]) {
peopleByAge[age] = [];
}
peopleByAge[age].push(person);
});
console.log(peopleByAge);
输出结果如下:
{
"28": [{"name":"Alice","age":28}, {"name":"Eve","age":28}],
"30": [{"name":"Bob","age":30}]
}
也可以使用reduce
方法:
const peopleByAge = people.reduce((acc, person) => {
const age = person.age;
if (!acc[age]) {
acc[age] = [];
}
acc[age].push(person);
return acc;
}, {});
无论哪种方式,代码都略显繁琐。每次都要检查对象,看分组键是否存在,如果不存在,则创建一个空数组,并将项目添加到该数组中。
使用 Object.groupBy 分组
可以通过以下方式来使用新的Object.groupBy
方法:
const peopleByAge = Object.groupBy(people, (person) => person.age);
可以看到,代码非常简洁!
不过需要注意,使用Object.groupBy
方法返回一个没有原型(即没有继承任何属性和方法)的对象。这意味着该对象不会继承Object.prototype
上的任何属性或方法,例如hasOwnProperty
或toString
等。虽然这样做可以避免意外覆盖Object.prototype
上的属性,但也意味着不能使用一些与对象相关的方法。
const peopleByAge = Object.groupBy(people, (person) => person.age);
console.log(peopleByAge.hasOwnProperty("28"));
// TypeError: peopleByAge.hasOwnProperty is not a function
在调用Object.groupBy
时,传递给它的回调函数应该返回一个字符串或 Symbol
类型的值。如果回调函数返回其他类型的值,它将被强制转换为字符串。
在这个例子中,回调函数返回的是一个数字类型的age
属性值,但由于Object.groupBy
方法要求键必须是字符串或 Symbol
类型,所以该数字会被强制转换为字符串类型。
console.log(peopleByAge[28]);
// => [{"name":"Alice","age":28}, {"name":"Eve","age":28}]
console.log(peopleByAge["28"]);
// => [{"name":"Alice","age":28}, {"name":"Eve","age":28}]
使用 Map.groupBy 分组
Map.groupBy
和Object.groupBy
几乎做的是相同的事情,只是返回的结果类型不同。Map.groupBy
返回一个Map对象,而不是像Object.groupBy
返回一个普通的对象。、
const ceo = { name: "Jamie", age: 40, reportsTo: null };
const manager = { name: "Alice", age: 28, reportsTo: ceo };
const people = [
ceo
manager,
{ name: "Bob", age: 30, reportsTo: manager },
{ name: "Eve", age: 28, reportsTo: ceo },
];
const peopleByManager = Map.groupBy(people, (person) => person.reportsTo);
这里根据人的汇报上级将他们进行了分组。如果想通过对象来从这个Map中获取数据,那么要求这些对象具有相同的身份或引用。这是因为Map在比较键时使用的是严格相等(===),只有两个对象具有相同的引用,才能被认为是相同的键。
peopleByManager.get(ceo);
// => [{ name: "Alice", age: 28, reportsTo: ceo }, { name: "Eve", age: 28, reportsTo: ceo }]
peopleByManager.get({ name: "Jamie", age: 40, reportsTo: null });
// => undefined
在上面的例子中,如果尝试使用与ceo
对象类似的对象作为键去访问Map中的项,由于这个对象与之前存储在Map中的ceo
对象不是同一个对象,所以无法检索到对应的值。
浏览器支持
这两个groupBy
方法是 proposal-array-grouping 提案提出的,该提案目前处于第3阶段,预计会在 2024 年成为正式标准。
9 月 12 日,Chrome 117发布,该版本支持了这两个方法。Firefox Nightly 在一个标志后已经实现了这两个方法。Safari已经以不同的名称实现了这些方法。由于这些方法在 Chrome 中可用,这意味着它们已经在V8中被实现,所以下一次V8更新时它们将在Node中可用。
为什么要使用静态方法?
你可能会想,为什么这个功能被实现为Object.groupBy
而不是Array.prototype.groupBy
。根据提案,有一个库曾经用不兼容的groupBy
方法对Array.prototype
进行了修改。在考虑为Web新增API时,向后兼容性非常重要。几年前,在尝试实现Array.prototype.flatten
时就出现了一个称为SmooshGate
的事件。
使用静态方法实际上对未来的可扩展性更好。当Records
和Tuples
提案实现时,可以添加一个Record.groupBy
方法,用于将数组分组为不可变记录。
简而言之,使用静态方法可以更好地保持向后兼容性,并提供更好的扩展性,以便在未来添加更多功能和数据结构。
JavaScript 正在填补这些空白,并使我们的开发更简单。目前,lodash.groupBy
每周的 npm 下载量在 150 万至 200 万次之间,当所有浏览器都支持该方法之后,就不再需要引入lodash.groupBy
库了!