Nomchy/discord.js
1
0
Fork 0
mirror of https://github.com/discordjs/discord.js.git synced 2026-03-09 16:13:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix(Collection): use Symbol.species for creating derived collections (#2715)

Browse Source
This commit is contained in:
Alex Hîncu
2018-08-10 11:57:33 +03:00
committed by Crawl
parent 0a21280990
commit ce8a679a14
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/stores/DataStore.js
View File

@@ -46,6 +46,10 @@ class DataStore extends Collection {
if (typeof idOrInstance === 'string') return idOrInstance;
return null;
}
static get [Symbol.species]() {
return Collection;
}
}
module.exports = DataStore;

9
src/util/Collection.js
View File

@@ -220,7 +220,7 @@ class Collection extends Map {
*/
filter(fn, thisArg) {
if (typeof thisArg !== 'undefined') fn = fn.bind(thisArg);
const results = new Collection();
const results = new this.constructor[Symbol.species]();
for (const [key, val] of this) {
if (fn(val, key, this)) results.set(key, val);
}
@@ -237,7 +237,7 @@ class Collection extends Map {
*/
partition(fn, thisArg) {
if (typeof thisArg !== 'undefined') fn = fn.bind(thisArg);
const results = [new Collection(), new Collection()];
const results = [new this.constructor[Symbol.species](), new this.constructor[Symbol.species]()];
for (const [key, val] of this) {
if (fn(val, key, this)) {
results[0].set(key, val);
@@ -365,7 +365,7 @@ class Collection extends Map {
* @example const newColl = someColl.clone();
*/
clone() {
return new this.constructor(this);
return new this.constructor[Symbol.species](this);
}
/**
@@ -409,7 +409,8 @@ class Collection extends Map {
* @example collection.sort((userA, userB) => userA.createdTimestamp - userB.createdTimestamp);
*/
sort(compareFunction = (x, y) => +(x > y) || +(x === y) - 1) {
return new Collection([...this.entries()].sort((a, b) => compareFunction(a[1], b[1], a[0], b[0])));
return new this.constructor[Symbol.species]([...this.entries()]
.sort((a, b) => compareFunction(a[1], b[1], a[0], b[0])));
}
toJSON() {