Nomchy/discord.js
1
0
Fork 0
mirror of https://github.com/discordjs/discord.js.git synced 2026-03-18 12:33:30 +01:00
Code Issues Packages Projects Releases Wiki Activity

add option for guilds per shard, and fix type (#1053)

Browse Source
This commit is contained in:
Gus Caplan
2016-12-30 20:03:23 -06:00
committed by Schuyler Cebulskie
parent 2156ce1e15
commit 2b25641270
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/util/FetchRecommendedShards.js
View File

@@ -3,17 +3,18 @@ const botGateway = require('./Constants').Endpoints.botGateway;
/** /**
* Gets the recommended shard count from Discord * Gets the recommended shard count from Discord
* @param {number} token Discord auth token * @param {string} token Discord auth token
* @param {number} [guildsPerShard=1000] Number of guilds per shard
* @returns {Promise<number>} the recommended number of shards * @returns {Promise<number>} the recommended number of shards
*/ */
module.exports = function fetchRecommendedShards(token) { module.exports = function fetchRecommendedShards(token, guildsPerShard = 1000) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
if (!token) throw new Error('A token must be provided.'); if (!token) throw new Error('A token must be provided.');
superagent.get(botGateway) superagent.get(botGateway)
.set('Authorization', `Bot ${token.replace(/^Bot\s*/i, '')}`) .set('Authorization', `Bot ${token.replace(/^Bot\s*/i, '')}`)
.end((err, res) => { .end((err, res) => {
if (err) reject(err); if (err) reject(err);
resolve(res.body.shards); resolve(res.body.shards * (1000 / guildsPerShard));
}); });
}); });
}; };