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

feat(ws): support for custom worker messaging (#10241)

Browse Source
This commit is contained in:
DD
2024-05-03 17:53:09 +03:00
committed by GitHub
parent 6cf094c282
commit 728164ed86
2 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packages/ws/src/strategies/sharding/WorkerShardingStrategy.ts
View File

@@ -66,6 +66,10 @@ export interface WorkerShardingStrategyOptions {
* Dictates how many shards should be spawned per worker thread.
*/
shardsPerWorker: number | 'all';
/**
* Handles a payload not recognized by the handler.
*/
unknownPayloadHandler?(payload: any): unknown;
/**
* Path to the worker file to use. The worker requires quite a bit of setup, it is recommended you leverage the {@link WorkerBootstrapper} class.
*/
@@ -225,7 +229,13 @@ export class WorkerShardingStrategy implements IShardingStrategy {
.on('messageerror', (err) => {
throw err;
})
.on('message', async (payload: WorkerReceivePayload) => this.onMessage(worker, payload));
.on('message', async (payload: any) => {
if ('op' in payload) {
await this.onMessage(worker, payload);
} else {
await this.options.unknownPayloadHandler?.(payload);
}
});
this.#workers.push(worker);
for (const shardId of workerData.shardIds) {
@@ -347,6 +357,11 @@ export class WorkerShardingStrategy implements IShardingStrategy {
break;
}
default: {
await this.options.unknownPayloadHandler?.(payload);
break;
}
}
}