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

refactor(sharding): use switch statement (#8928)

Browse Source
This commit is contained in:
Tristan Camejo
2022-12-15 19:07:28 +11:00
committed by GitHub
parent 627511d652
commit 6540914b4a
2 changed files with 60 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
packages/discord.js/src/sharding/Shard.js
View File

@@ -21,8 +21,14 @@ class Shard extends EventEmitter {
constructor(manager, id) { constructor(manager, id) {
super(); super();
if (manager.mode === 'process') childProcess = require('node:child_process'); switch (manager.mode) {
else if (manager.mode === 'worker') Worker = require('node:worker_threads').Worker; case 'process':
childProcess = require('node:child_process');
break;
case 'worker':
Worker = require('node:worker_threads').Worker;
break;
}
/** /**
* Manager that created the shard * Manager that created the shard
@@ -112,18 +118,21 @@ class Shard extends EventEmitter {
this._exitListener = this._handleExit.bind(this, undefined, timeout); this._exitListener = this._handleExit.bind(this, undefined, timeout);
if (this.manager.mode === 'process') { switch (this.manager.mode) {
this.process = childProcess case 'process':
.fork(path.resolve(this.manager.file), this.args, { this.process = childProcess
env: this.env, .fork(path.resolve(this.manager.file), this.args, {
execArgv: this.execArgv, env: this.env,
}) execArgv: this.execArgv,
.on('message', this._handleMessage.bind(this)) })
.on('exit', this._exitListener); .on('message', this._handleMessage.bind(this))
} else if (this.manager.mode === 'worker') { .on('exit', this._exitListener);
this.worker = new Worker(path.resolve(this.manager.file), { workerData: this.env }) break;
.on('message', this._handleMessage.bind(this)) case 'worker':
.on('exit', this._exitListener); this.worker = new Worker(path.resolve(this.manager.file), { workerData: this.env })
.on('message', this._handleMessage.bind(this))
.on('exit', this._exitListener);
break;
} }
this._evals.clear(); this._evals.clear();

68
packages/discord.js/src/sharding/ShardClientUtil.js
View File

@@ -29,29 +29,32 @@ class ShardClientUtil {
*/ */
this.parentPort = null; this.parentPort = null;
if (mode === 'process') { switch (mode) {
process.on('message', this._handleMessage.bind(this)); case 'process':
client.on('ready', () => { process.on('message', this._handleMessage.bind(this));
process.send({ _ready: true }); client.on('ready', () => {
}); process.send({ _ready: true });
client.on('disconnect', () => { });
process.send({ _disconnect: true }); client.on('disconnect', () => {
}); process.send({ _disconnect: true });
client.on('reconnecting', () => { });
process.send({ _reconnecting: true }); client.on('reconnecting', () => {
}); process.send({ _reconnecting: true });
} else if (mode === 'worker') { });
this.parentPort = require('node:worker_threads').parentPort; break;
this.parentPort.on('message', this._handleMessage.bind(this)); case 'worker':
client.on('ready', () => { this.parentPort = require('node:worker_threads').parentPort;
this.parentPort.postMessage({ _ready: true }); this.parentPort.on('message', this._handleMessage.bind(this));
}); client.on('ready', () => {
client.on('disconnect', () => { this.parentPort.postMessage({ _ready: true });
this.parentPort.postMessage({ _disconnect: true }); });
}); client.on('disconnect', () => {
client.on('reconnecting', () => { this.parentPort.postMessage({ _disconnect: true });
this.parentPort.postMessage({ _reconnecting: true }); });
}); client.on('reconnecting', () => {
this.parentPort.postMessage({ _reconnecting: true });
});
break;
} }
} }
@@ -81,14 +84,17 @@ class ShardClientUtil {
*/ */
send(message) { send(message) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
if (this.mode === 'process') { switch (this.mode) {
process.send(message, err => { case 'process':
if (err) reject(err); process.send(message, err => {
else resolve(); if (err) reject(err);
}); else resolve();
} else if (this.mode === 'worker') { });
this.parentPort.postMessage(message); break;
resolve(); case 'worker':
this.parentPort.postMessage(message);
resolve();
break;
} }
}); });
} }