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

Tidy up shards a bit

Browse Source
This commit is contained in:
Amish Shah
2016-08-30 19:07:05 +01:00
parent c70dfa83fb
commit cde3bcb3bd
4 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/sharding/ShardingManager.js
View File

@@ -1,6 +1,7 @@
const childProcess = require('child_process');
const path = require('path');
const EventEmitter = require('events').EventEmitter;
const Collection = require('../util/Collection');
class ShardingManager extends EventEmitter {
constructor(file, totalShards) {
@@ -10,23 +11,24 @@ class ShardingManager extends EventEmitter {
this.file = path.resolve(`${process.cwd()}${file}`);
}
this.totalShards = totalShards;
this.runningShards = 0;
this.createShards();
this.shards = new Collection();
}
createShard(id) {
const shard = childProcess.fork(path.resolve(this.file), [id, this.totalShards]);
createShard() {
const id = this.shards.size;
const shard = childProcess.fork(path.resolve(this.file), [id, this.totalShards], { silent: true });
this.shards.set(id, shard);
this.emit('launch', id, shard);
}
createShards() {
this.createShard(0);
spawn(amount) {
this.totalShards = amount;
this.createShard();
const interval = setInterval(() => {
this.runningShards++;
if (this.runningShards === this.totalShards) {
if (this.shards.size === this.totalShards) {
return clearInterval(interval);
}
this.createShard(this.runningShards);
this.createShard();
}, 5500);
}
}