daenamnu/discordeno
1
0
Fork 0
mirror of https://github.com/discordeno/discordeno.git synced 2026-06-02 17:00:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(gateway): add options.spreadShardsInRoundRobin to spread shards in a round-robin manner (#4204)

Browse Source 
* feat(gateway): add options.spreadShardsInRoundRobin to spread shards in a round-robin manner

* fix: forgor to add to gateway
This commit is contained in:
Awesome Stickz
2025-05-26 11:57:48 +05:30
committed by GitHub
parent a62fb5e17b
commit 94040813bb
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
packages/gateway/src/manager.ts
View File

@@ -47,6 +47,7 @@ export function createGatewayManager(options: CreateGatewayManagerOptions): Gate
totalWorkers: options.totalWorkers ?? 4,
shardsPerWorker: options.shardsPerWorker ?? 25,
spawnShardDelay: options.spawnShardDelay ?? 5300,
spreadShardsInRoundRobin: options.spreadShardsInRoundRobin ?? false,
preferSnakeCase: options.preferSnakeCase ?? false,
shards: new Map(),
buckets: new Map(),
@@ -247,7 +248,9 @@ export function createGatewayManager(options: CreateGatewayManagerOptions): Gate
)
},
calculateWorkerId(shardId) {
const workerId = Math.min(Math.floor(shardId / gateway.shardsPerWorker), gateway.totalWorkers - 1)
const workerId = options.spreadShardsInRoundRobin
? shardId % gateway.totalWorkers
: Math.min(Math.floor(shardId / gateway.shardsPerWorker), gateway.totalWorkers - 1)
gateway.logger.debug(
`[Gateway] Calculating workerId: Shard: ${shardId} -> Worker: ${workerId} -> Per Worker: ${gateway.shardsPerWorker} -> Total: ${gateway.totalWorkers}`,
)
@@ -607,6 +610,19 @@ export interface CreateGatewayManagerOptions {
* @default 4
*/
totalWorkers?: number
/**
* Whether to spread shards across workers in a round-robin manner.
*
* @remarks
* By default, shards are assigned to workers in contiguous blocks based on shardsPerWorker. If any shard is left over, it will be assigned to the last worker.
* This means that if you have 3 workers and 40 shards while shardsPerWorker is 10, the first worker will get shards 0-9, the second worker will get shards 10-19, and so on, the last worker will get shards 20-39.
*
* If this option is set to true, the shards will be assigned in a round-robin manner to spread more evenly across workers.
* For example, with 3 workers and 40 shards, the first shard will go to worker 0, the second shard to worker 1, the third shard to worker 2, the fourth shard to worker 0, and so on.
*
* @default false
*/
spreadShardsInRoundRobin?: boolean
/** Important data which is used by the manager to connect shards to the gateway. */
connection?: Camelize<DiscordGetGatewayBot>
/** Whether incoming payloads are compressed using zlib.