mirror of
https://github.com/danbulant/discord.js
synced 2026-05-25 21:12:06 +00:00
36 lines
994 B
JavaScript
36 lines
994 B
JavaScript
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) {
|
|
super();
|
|
this.file = file;
|
|
if (!path.isAbsolute(file)) {
|
|
this.file = path.resolve(`${process.cwd()}${file}`);
|
|
}
|
|
this.totalShards = totalShards;
|
|
this.shards = new Collection();
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
spawn(amount) {
|
|
this.totalShards = amount;
|
|
this.createShard();
|
|
const interval = setInterval(() => {
|
|
if (this.shards.size === this.totalShards) {
|
|
return clearInterval(interval);
|
|
}
|
|
this.createShard();
|
|
}, 5500);
|
|
}
|
|
}
|
|
|
|
module.exports = ShardingManager;
|