Skip to content

Commit

Permalink
fix(GuildMemberManager): refactor fetchMany (#6328)
Browse files Browse the repository at this point in the history
  • Loading branch information
iCrawl committed Aug 7, 2021
1 parent ceae596 commit 909c873
Showing 1 changed file with 3 additions and 9 deletions.
12 changes: 3 additions & 9 deletions src/managers/GuildMemberManager.js
Expand Up @@ -378,13 +378,8 @@ class GuildMemberManager extends CachedManager {
query,
time = 120e3,
nonce = SnowflakeUtil.generate(),
force = false,
} = {}) {
return new Promise((resolve, reject) => {
if (!query && !limit && !presences && !user_ids && !force) {
resolve(this.cache);
return;
}
if (!query && !user_ids) query = '';
if (nonce.length > 32) throw new RangeError('MEMBER_FETCH_NONCE_LENGTH');
this.guild.shard.send({
Expand All @@ -399,20 +394,19 @@ class GuildMemberManager extends CachedManager {
},
});
const fetchedMembers = new Collection();
const option = Boolean(query || limit || presences || user_ids);
let i = 0;
const handler = (members, _, chunk) => {
timeout.refresh();
if (chunk.nonce !== nonce) return;
i++;
for (const member of members.values()) {
if (option || force) fetchedMembers.set(member.id, member);
fetchedMembers.set(member.id, member);
}
if ((option && members.size < 1000) || (limit && fetchedMembers.size >= limit) || i === chunk.count) {
if (members.size < 1000 || (limit && fetchedMembers.size >= limit) || i === chunk.count) {
clearTimeout(timeout);
this.client.removeListener(Events.GUILD_MEMBERS_CHUNK, handler);
this.client.decrementMaxListeners();
let fetched = option || force ? fetchedMembers : this.cache;
let fetched = fetchedMembers;
if (user_ids && !Array.isArray(user_ids) && fetched.size) fetched = fetched.first();
resolve(fetched);
}
Expand Down

0 comments on commit 909c873

Please sign in to comment.