Skip to content

Commit

Permalink
feat: Add message to send resume event to shard (#9626)
Browse files Browse the repository at this point in the history
* fix(WebSocketManager): Fix not ready after reconnect

* fix(WebSocketManager): Move fixes from Ready to Resume

* fix: undo changes

* feat: Add message to send resume event to shard

---------

Co-authored-by: GrapeColor <grapecolor@users.noreply.github.com>
Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>
  • Loading branch information
3 people committed Jun 14, 2023
1 parent db925fa commit a873ec1
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 0 deletions.
6 changes: 6 additions & 0 deletions packages/discord.js/src/sharding/Shard.js
Original file line number Diff line number Diff line change
Expand Up @@ -367,6 +367,12 @@ class Shard extends EventEmitter {
return;
}

// Shard has resumed
if (message._resume) {
this.ready = true;
return;
}

// Shard is requesting a property fetch
if (message._sFetchProp) {
const resp = { _sFetchProp: message._sFetchProp, _sFetchPropShard: message._sFetchPropShard };
Expand Down
6 changes: 6 additions & 0 deletions packages/discord.js/src/sharding/ShardClientUtil.js
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,9 @@ class ShardClientUtil {
client.on(Events.ShardReconnecting, () => {
process.send({ _reconnecting: true });
});
client.on(Events.ShardResume, () => {
process.send({ _resume: true });
});
break;
case 'worker':
this.parentPort = require('node:worker_threads').parentPort;
Expand All @@ -55,6 +58,9 @@ class ShardClientUtil {
client.on(Events.ShardReconnecting, () => {
this.parentPort.postMessage({ _reconnecting: true });
});
client.on(Events.ShardResume, () => {
this.parentPort.postMessage({ _resume: true });
});
break;
}
}
Expand Down

0 comments on commit a873ec1

Please sign in to comment.