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

feat(ws): custom workers (#9004)

Browse Source 
* feat(ws): custom workers

* chore: typo

* refactor(WebSocketShard): expose shard id

* chore: remove outdated readme comment

* chore: nits

* chore: remove unnecessary mutation

* feat: fancier resolution

* chore: remove unnecessary exports

* chore: apply suggestions

* refactor: use range errors

Co-authored-by: Aura Román <kyradiscord@gmail.com>
This commit is contained in:
DD
2023-01-10 19:31:56 +02:00
committed by GitHub
parent 39c4de2dbc
commit 828a13b526
11 changed files with 343 additions and 159 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
packages/ws/__tests__/strategy/WorkerShardingStrategy.test.ts
View File

@@ -53,45 +53,54 @@ vi.mock('node:worker_threads', async () => {
super();
mockConstructor(...args);
// need to delay this by an event loop cycle to allow the strategy to attach a listener
setImmediate(() => this.emit('online'));
setImmediate(() => {
this.emit('online');
// same deal here
setImmediate(() => {
const message = {
op: WorkerRecievePayloadOp.WorkerReady,
} satisfies WorkerRecievePayload;
this.emit('message', message);
});
});
}
public postMessage(message: WorkerSendPayload) {
switch (message.op) {
case WorkerSendPayloadOp.Connect: {
const response: WorkerRecievePayload = {
const response = {
op: WorkerRecievePayloadOp.Connected,
shardId: message.shardId,
};
} satisfies WorkerRecievePayload;
this.emit('message', response);
break;
}
case WorkerSendPayloadOp.Destroy: {
const response: WorkerRecievePayload = {
const response = {
op: WorkerRecievePayloadOp.Destroyed,
shardId: message.shardId,
};
} satisfies WorkerRecievePayload;
this.emit('message', response);
break;
}
case WorkerSendPayloadOp.Send: {
if (message.payload.op === GatewayOpcodes.RequestGuildMembers) {
const response: WorkerRecievePayload = {
const response = {
op: WorkerRecievePayloadOp.Event,
shardId: message.shardId,
event: WebSocketShardEvents.Dispatch,
data: memberChunkData,
};
} satisfies WorkerRecievePayload;
this.emit('message', response);
// Fetch session info
const sessionFetch: WorkerRecievePayload = {
const sessionFetch = {
op: WorkerRecievePayloadOp.RetrieveSessionInfo,
shardId: message.shardId,
nonce: Math.random(),
};
} satisfies WorkerRecievePayload;
this.emit('message', sessionFetch);
}
@@ -102,11 +111,11 @@ vi.mock('node:worker_threads', async () => {
case WorkerSendPayloadOp.SessionInfoResponse: {
message.session ??= sessionInfo;
const session: WorkerRecievePayload = {
const session = {
op: WorkerRecievePayloadOp.UpdateSessionInfo,
shardId: message.session.shardId,
session: { ...message.session, sequence: message.session.sequence + 1 },
};
} satisfies WorkerRecievePayload;
this.emit('message', session);
break;
}
@@ -186,7 +195,7 @@ test('spawn, connect, send a message, session info, and destroy', async () => {
await manager.connect();
expect(mockConstructor).toHaveBeenCalledWith(
expect.stringContaining('worker.js'),
expect.stringContaining('defaultWorker.js'),
expect.objectContaining({ workerData: expect.objectContaining({ shardIds: [0, 1] }) }),
);