dan/discord.js
1
0
Fork 0
mirror of https://github.com/danbulant/discord.js synced 2026-06-03 06:41:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

allow passing a function to shard eval (#2193)

Browse source
This commit is contained in:
Gus Caplan 2018-01-12 07:05:27 -06:00 committed by Crawl
parent 2249da464f
commit 8aafcd6cde
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/sharding/Shard.js
View file

@ -175,8 +175,8 @@ class Shard extends EventEmitter {
} }
/** /**
* Evaluates a script on the shard, in the context of the {@link Client}. * Evaluates a script or function on the shard, in the context of the {@link Client}.
* @param {string} script JavaScript to run on the shard * @param {string|Function} script JavaScript to run on the shard
* @returns {Promise<*>} Result of the script execution * @returns {Promise<*>} Result of the script execution
*/ */
eval(script) { eval(script) {
@ -191,7 +191,8 @@ class Shard extends EventEmitter {
}; };
this.process.on('message', listener); this.process.on('message', listener);
this.send({ _eval: script }).catch(err => { const _eval = typeof script === 'function' ? `(${script})(this)` : script;
this.send({ _eval }).catch(err => {
this.process.removeListener('message', listener); this.process.removeListener('message', listener);
this._evals.delete(script); this._evals.delete(script);
reject(err); reject(err);

1
src/sharding/ShardClientUtil.js
View file

@ -86,6 +86,7 @@ class ShardClientUtil {
*/ */
broadcastEval(script) { broadcastEval(script) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
script = typeof script !== 'function' ? `(${script})(this)` : script;
const listener = message => { const listener = message => {
if (!message || message._sEval !== script) return; if (!message || message._sEval !== script) return;
process.removeListener('message', listener); process.removeListener('message', listener);