@discordjs/ws 0.1.0

package/dist/strategies/context/WorkerContextFetchingStrategy.mjs

@@ -0,0 +1,42 @@
+import { isMainThread, parentPort } from 'node:worker_threads';
+import { Collection } from '@discordjs/collection';
+import { WorkerSendPayloadOp, WorkerRecievePayloadOp } from '../sharding/WorkerShardingStrategy.mjs';
+
+class WorkerContextFetchingStrategy {
+  constructor(options) {
+    this.options = options;
+    this.sessionPromises = new Collection();
+    if (isMainThread) {
+      throw new Error("Cannot instantiate WorkerContextFetchingStrategy on the main thread");
+    }
+    parentPort.on("message", (payload) => {
+      if (payload.op === WorkerSendPayloadOp.SessionInfoResponse) {
+        const resolve = this.sessionPromises.get(payload.nonce);
+        resolve?.(payload.session);
+        this.sessionPromises.delete(payload.nonce);
+      }
+    });
+  }
+  async retrieveSessionInfo(shardId) {
+    const nonce = Math.random();
+    const payload = {
+      op: WorkerRecievePayloadOp.RetrieveSessionInfo,
+      shardId,
+      nonce
+    };
+    const promise = new Promise((resolve) => this.sessionPromises.set(nonce, resolve));
+    parentPort.postMessage(payload);
+    return promise;
+  }
+  updateSessionInfo(shardId, sessionInfo) {
+    const payload = {
+      op: WorkerRecievePayloadOp.UpdateSessionInfo,
+      shardId,
+      session: sessionInfo
+    };
+    parentPort.postMessage(payload);
+  }
+}
+
+export { WorkerContextFetchingStrategy };
+//# sourceMappingURL=WorkerContextFetchingStrategy.mjs.map

package/dist/strategies/context/WorkerContextFetchingStrategy.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerContextFetchingStrategy.mjs","sources":["../../../src/strategies/context/WorkerContextFetchingStrategy.ts"],"sourcesContent":["import { isMainThread, parentPort } from 'node:worker_threads';\nimport { Collection } from '@discordjs/collection';\nimport type { FetchingStrategyOptions, IContextFetchingStrategy } from './IContextFetchingStrategy';\nimport type { SessionInfo } from '../../ws/WebSocketManager';\nimport {\n\tWorkerRecievePayload,\n\tWorkerRecievePayloadOp,\n\tWorkerSendPayload,\n\tWorkerSendPayloadOp,\n} from '../sharding/WorkerShardingStrategy';\n\nexport class WorkerContextFetchingStrategy implements IContextFetchingStrategy {\n\tprivate readonly sessionPromises = new Collection<number, (session: SessionInfo | null) => void>();\n\n\tpublic constructor(public readonly options: FetchingStrategyOptions) {\n\t\tif (isMainThread) {\n\t\t\tthrow new Error('Cannot instantiate WorkerContextFetchingStrategy on the main thread');\n\t\t}\n\n\t\tparentPort!.on('message', (payload: WorkerSendPayload) => {\n\t\t\tif (payload.op === WorkerSendPayloadOp.SessionInfoResponse) {\n\t\t\t\tconst resolve = this.sessionPromises.get(payload.nonce);\n\t\t\t\tresolve?.(payload.session);\n\t\t\t\tthis.sessionPromises.delete(payload.nonce);\n\t\t\t}\n\t\t});\n\t}\n\n\tpublic async retrieveSessionInfo(shardId: number): Promise<SessionInfo | null> {\n\t\tconst nonce = Math.random();\n\t\tconst payload: WorkerRecievePayload = {\n\t\t\top: WorkerRecievePayloadOp.RetrieveSessionInfo,\n\t\t\tshardId,\n\t\t\tnonce,\n\t\t};\n\t\tconst promise = new Promise<SessionInfo | null>((resolve) => this.sessionPromises.set(nonce, resolve));\n\t\tparentPort!.postMessage(payload);\n\t\treturn promise;\n\t}\n\n\tpublic updateSessionInfo(shardId: number, sessionInfo: SessionInfo | null) {\n\t\tconst payload: WorkerRecievePayload = {\n\t\t\top: WorkerRecievePayloadOp.UpdateSessionInfo,\n\t\t\tshardId,\n\t\t\tsession: sessionInfo,\n\t\t};\n\t\tparentPort!.postMessage(payload);\n\t}\n}\n"],"names":[],"mappings":";;;;AAMO,MAAM,6BAA6B,CAAC;AAC3C,EAAE,WAAW,CAAC,OAAO,EAAE;AACvB,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,IAAI,IAAI,CAAC,eAAe,GAAG,IAAI,UAAU,EAAE,CAAC;AAC5C,IAAI,IAAI,YAAY,EAAE;AACtB,MAAM,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;AAC7F,KAAK;AACL,IAAI,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,KAAK;AAC1C,MAAM,IAAI,OAAO,CAAC,EAAE,KAAK,mBAAmB,CAAC,mBAAmB,EAAE;AAClE,QAAQ,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChE,QAAQ,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;AACnC,QAAQ,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AACnD,OAAO;AACP,KAAK,CAAC,CAAC;AACP,GAAG;AACH,EAAE,MAAM,mBAAmB,CAAC,OAAO,EAAE;AACrC,IAAI,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;AAChC,IAAI,MAAM,OAAO,GAAG;AACpB,MAAM,EAAE,EAAE,sBAAsB,CAAC,mBAAmB;AACpD,MAAM,OAAO;AACb,MAAM,KAAK;AACX,KAAK,CAAC;AACN,IAAI,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;AACvF,IAAI,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AACpC,IAAI,OAAO,OAAO,CAAC;AACnB,GAAG;AACH,EAAE,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE;AAC1C,IAAI,MAAM,OAAO,GAAG;AACpB,MAAM,EAAE,EAAE,sBAAsB,CAAC,iBAAiB;AAClD,MAAM,OAAO;AACb,MAAM,OAAO,EAAE,WAAW;AAC1B,KAAK,CAAC;AACN,IAAI,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AACpC,GAAG;AACH;;;;"}

package/dist/strategies/sharding/IShardingStrategy.d.ts

@@ -0,0 +1,25 @@
+import type { GatewaySendPayload } from 'discord-api-types/v10';
+import type { Awaitable } from '../../utils/utils';
+import type { WebSocketShardDestroyOptions } from '../../ws/WebSocketShard';
+/**
+ * Strategies responsible for spawning, initializing connections, destroying shards, and relaying events
+ */
+export interface IShardingStrategy {
+    /**
+     * Spawns all the shards
+     */
+    spawn: (shardIds: number[]) => Awaitable<void>;
+    /**
+     * Initializes all the shards
+     */
+    connect: () => Awaitable<void>;
+    /**
+     * Destroys all the shards
+     */
+    destroy: (options?: Omit<WebSocketShardDestroyOptions, 'recover'>) => Awaitable<void>;
+    /**
+     * Sends a payload to a shard
+     */
+    send: (shardId: number, payload: GatewaySendPayload) => Awaitable<void>;
+}
+//# sourceMappingURL=IShardingStrategy.d.ts.map

package/dist/strategies/sharding/IShardingStrategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"IShardingStrategy.d.ts","sourceRoot":"","sources":["../../../src/strategies/sharding/IShardingStrategy.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,KAAK,EAAE,4BAA4B,EAAE,MAAM,yBAAyB,CAAC;AAE5E;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;IAC/C;;OAEG;IACH,OAAO,EAAE,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;IAC/B;;OAEG;IACH,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,4BAA4B,EAAE,SAAS,CAAC,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;IACtF;;OAEG;IACH,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,kBAAkB,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;CACxE"}

package/dist/strategies/sharding/SimpleShardingStrategy.cjs

@@ -0,0 +1,53 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+const collection = require('@discordjs/collection');
+const IdentifyThrottler = require('../../utils/IdentifyThrottler.cjs');
+const WebSocketShard = require('../../ws/WebSocketShard.cjs');
+const IContextFetchingStrategy = require('../context/IContextFetchingStrategy.cjs');
+const SimpleContextFetchingStrategy = require('../context/SimpleContextFetchingStrategy.cjs');
+
+class SimpleShardingStrategy {
+  constructor(manager) {
+    this.shards = new collection.Collection();
+    this.manager = manager;
+    this.throttler = new IdentifyThrottler.IdentifyThrottler(manager);
+  }
+  async spawn(shardIds) {
+    const strategyOptions = await IContextFetchingStrategy.managerToFetchingStrategyOptions(this.manager);
+    for (const shardId of shardIds) {
+      const strategy = new SimpleContextFetchingStrategy.SimpleContextFetchingStrategy(this.manager, strategyOptions);
+      const shard = new WebSocketShard.WebSocketShard(strategy, shardId);
+      for (const event of Object.values(WebSocketShard.WebSocketShardEvents)) {
+        shard.on(event, (payload) => this.manager.emit(event, { ...payload, shardId }));
+      }
+      this.shards.set(shardId, shard);
+    }
+  }
+  async connect() {
+    const promises = [];
+    for (const shard of this.shards.values()) {
+      await this.throttler.waitForIdentify();
+      promises.push(shard.connect());
+    }
+    await Promise.all(promises);
+  }
+  async destroy(options) {
+    const promises = [];
+    for (const shard of this.shards.values()) {
+      promises.push(shard.destroy(options));
+    }
+    await Promise.all(promises);
+    this.shards.clear();
+  }
+  send(shardId, payload) {
+    const shard = this.shards.get(shardId);
+    if (!shard)
+      throw new Error(`Shard ${shardId} not found`);
+    return shard.send(payload);
+  }
+}
+
+exports.SimpleShardingStrategy = SimpleShardingStrategy;
+//# sourceMappingURL=SimpleShardingStrategy.cjs.map

package/dist/strategies/sharding/SimpleShardingStrategy.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"SimpleShardingStrategy.cjs","sources":["../../../src/strategies/sharding/SimpleShardingStrategy.ts"],"sourcesContent":["import { Collection } from '@discordjs/collection';\nimport type { GatewaySendPayload } from 'discord-api-types/v10';\nimport type { IShardingStrategy } from './IShardingStrategy';\nimport { IdentifyThrottler } from '../../utils/IdentifyThrottler';\nimport type { WebSocketManager } from '../../ws/WebSocketManager';\nimport { WebSocketShard, WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';\nimport { managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy';\nimport { SimpleContextFetchingStrategy } from '../context/SimpleContextFetchingStrategy';\n\n/**\n * Simple strategy that just spawns shards in the current process\n */\nexport class SimpleShardingStrategy implements IShardingStrategy {\n\tprivate readonly manager: WebSocketManager;\n\tprivate readonly shards = new Collection<number, WebSocketShard>();\n\n\tprivate readonly throttler: IdentifyThrottler;\n\n\tpublic constructor(manager: WebSocketManager) {\n\t\tthis.manager = manager;\n\t\tthis.throttler = new IdentifyThrottler(manager);\n\t}\n\n\tpublic async spawn(shardIds: number[]) {\n\t\tconst strategyOptions = await managerToFetchingStrategyOptions(this.manager);\n\t\tfor (const shardId of shardIds) {\n\t\t\tconst strategy = new SimpleContextFetchingStrategy(this.manager, strategyOptions);\n\t\t\tconst shard = new WebSocketShard(strategy, shardId);\n\t\t\tfor (const event of Object.values(WebSocketShardEvents)) {\n\t\t\t\t// @ts-expect-error\n\t\t\t\tshard.on(event, (payload) => this.manager.emit(event, { ...payload, shardId }));\n\t\t\t}\n\t\t\tthis.shards.set(shardId, shard);\n\t\t}\n\t}\n\n\tpublic async connect() {\n\t\tconst promises = [];\n\n\t\tfor (const shard of this.shards.values()) {\n\t\t\tawait this.throttler.waitForIdentify();\n\t\t\tpromises.push(shard.connect());\n\t\t}\n\n\t\tawait Promise.all(promises);\n\t}\n\n\tpublic async destroy(options?: Omit<WebSocketShardDestroyOptions, 'recover'>) {\n\t\tconst promises = [];\n\n\t\tfor (const shard of this.shards.values()) {\n\t\t\tpromises.push(shard.destroy(options));\n\t\t}\n\n\t\tawait Promise.all(promises);\n\t\tthis.shards.clear();\n\t}\n\n\tpublic send(shardId: number, payload: GatewaySendPayload) {\n\t\tconst shard = this.shards.get(shardId);\n\t\tif (!shard) throw new Error(`Shard ${shardId} not found`);\n\t\treturn shard.send(payload);\n\t}\n}\n"],"names":["Collection","IdentifyThrottler","managerToFetchingStrategyOptions","SimpleContextFetchingStrategy","WebSocketShard","WebSocketShardEvents"],"mappings":";;;;;;;;;;AAKO,MAAM,sBAAsB,CAAC;AACpC,EAAE,WAAW,CAAC,OAAO,EAAE;AACvB,IAAI,IAAI,CAAC,MAAM,GAAG,IAAIA,qBAAU,EAAE,CAAC;AACnC,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,IAAI,IAAI,CAAC,SAAS,GAAG,IAAIC,mCAAiB,CAAC,OAAO,CAAC,CAAC;AACpD,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,QAAQ,EAAE;AACxB,IAAI,MAAM,eAAe,GAAG,MAAMC,yDAAgC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AACjF,IAAI,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;AACpC,MAAM,MAAM,QAAQ,GAAG,IAAIC,2DAA6B,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;AACxF,MAAM,MAAM,KAAK,GAAG,IAAIC,6BAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AAC1D,MAAM,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAACC,mCAAoB,CAAC,EAAE;AAC/D,QAAQ,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;AACxF,OAAO;AACP,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AACtC,KAAK;AACL,GAAG;AACH,EAAE,MAAM,OAAO,GAAG;AAClB,IAAI,MAAM,QAAQ,GAAG,EAAE,CAAC;AACxB,IAAI,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE;AAC9C,MAAM,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;AAC7C,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;AACrC,KAAK;AACL,IAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAChC,GAAG;AACH,EAAE,MAAM,OAAO,CAAC,OAAO,EAAE;AACzB,IAAI,MAAM,QAAQ,GAAG,EAAE,CAAC;AACxB,IAAI,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE;AAC9C,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;AAC5C,KAAK;AACL,IAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAChC,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;AACxB,GAAG;AACH,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE;AACzB,IAAI,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AAC3C,IAAI,IAAI,CAAC,KAAK;AACd,MAAM,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;AACpD,IAAI,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AAC/B,GAAG;AACH;;;;"}

package/dist/strategies/sharding/SimpleShardingStrategy.d.ts

@@ -0,0 +1,18 @@
+import type { GatewaySendPayload } from 'discord-api-types/v10';
+import type { IShardingStrategy } from './IShardingStrategy';
+import type { WebSocketManager } from '../../ws/WebSocketManager';
+import { WebSocketShardDestroyOptions } from '../../ws/WebSocketShard';
+/**
+ * Simple strategy that just spawns shards in the current process
+ */
+export declare class SimpleShardingStrategy implements IShardingStrategy {
+    private readonly manager;
+    private readonly shards;
+    private readonly throttler;
+    constructor(manager: WebSocketManager);
+    spawn(shardIds: number[]): Promise<void>;
+    connect(): Promise<void>;
+    destroy(options?: Omit<WebSocketShardDestroyOptions, 'recover'>): Promise<void>;
+    send(shardId: number, payload: GatewaySendPayload): Promise<void>;
+}
+//# sourceMappingURL=SimpleShardingStrategy.d.ts.map

package/dist/strategies/sharding/SimpleShardingStrategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SimpleShardingStrategy.d.ts","sourceRoot":"","sources":["../../../src/strategies/sharding/SimpleShardingStrategy.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAE7D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAkB,4BAA4B,EAAwB,MAAM,yBAAyB,CAAC;AAI7G;;GAEG;AACH,qBAAa,sBAAuB,YAAW,iBAAiB;IAC/D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA4C;IAEnE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAoB;gBAE3B,OAAO,EAAE,gBAAgB;IAK/B,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE;IAaxB,OAAO;IAWP,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,4BAA4B,EAAE,SAAS,CAAC;IAWrE,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,kBAAkB;CAKxD"}

package/dist/strategies/sharding/SimpleShardingStrategy.mjs

@@ -0,0 +1,49 @@
+import { Collection } from '@discordjs/collection';
+import { IdentifyThrottler } from '../../utils/IdentifyThrottler.mjs';
+import { WebSocketShard, WebSocketShardEvents } from '../../ws/WebSocketShard.mjs';
+import { managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy.mjs';
+import { SimpleContextFetchingStrategy } from '../context/SimpleContextFetchingStrategy.mjs';
+
+class SimpleShardingStrategy {
+  constructor(manager) {
+    this.shards = new Collection();
+    this.manager = manager;
+    this.throttler = new IdentifyThrottler(manager);
+  }
+  async spawn(shardIds) {
+    const strategyOptions = await managerToFetchingStrategyOptions(this.manager);
+    for (const shardId of shardIds) {
+      const strategy = new SimpleContextFetchingStrategy(this.manager, strategyOptions);
+      const shard = new WebSocketShard(strategy, shardId);
+      for (const event of Object.values(WebSocketShardEvents)) {
+        shard.on(event, (payload) => this.manager.emit(event, { ...payload, shardId }));
+      }
+      this.shards.set(shardId, shard);
+    }
+  }
+  async connect() {
+    const promises = [];
+    for (const shard of this.shards.values()) {
+      await this.throttler.waitForIdentify();
+      promises.push(shard.connect());
+    }
+    await Promise.all(promises);
+  }
+  async destroy(options) {
+    const promises = [];
+    for (const shard of this.shards.values()) {
+      promises.push(shard.destroy(options));
+    }
+    await Promise.all(promises);
+    this.shards.clear();
+  }
+  send(shardId, payload) {
+    const shard = this.shards.get(shardId);
+    if (!shard)
+      throw new Error(`Shard ${shardId} not found`);
+    return shard.send(payload);
+  }
+}
+
+export { SimpleShardingStrategy };
+//# sourceMappingURL=SimpleShardingStrategy.mjs.map

package/dist/strategies/sharding/SimpleShardingStrategy.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"SimpleShardingStrategy.mjs","sources":["../../../src/strategies/sharding/SimpleShardingStrategy.ts"],"sourcesContent":["import { Collection } from '@discordjs/collection';\nimport type { GatewaySendPayload } from 'discord-api-types/v10';\nimport type { IShardingStrategy } from './IShardingStrategy';\nimport { IdentifyThrottler } from '../../utils/IdentifyThrottler';\nimport type { WebSocketManager } from '../../ws/WebSocketManager';\nimport { WebSocketShard, WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';\nimport { managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy';\nimport { SimpleContextFetchingStrategy } from '../context/SimpleContextFetchingStrategy';\n\n/**\n * Simple strategy that just spawns shards in the current process\n */\nexport class SimpleShardingStrategy implements IShardingStrategy {\n\tprivate readonly manager: WebSocketManager;\n\tprivate readonly shards = new Collection<number, WebSocketShard>();\n\n\tprivate readonly throttler: IdentifyThrottler;\n\n\tpublic constructor(manager: WebSocketManager) {\n\t\tthis.manager = manager;\n\t\tthis.throttler = new IdentifyThrottler(manager);\n\t}\n\n\tpublic async spawn(shardIds: number[]) {\n\t\tconst strategyOptions = await managerToFetchingStrategyOptions(this.manager);\n\t\tfor (const shardId of shardIds) {\n\t\t\tconst strategy = new SimpleContextFetchingStrategy(this.manager, strategyOptions);\n\t\t\tconst shard = new WebSocketShard(strategy, shardId);\n\t\t\tfor (const event of Object.values(WebSocketShardEvents)) {\n\t\t\t\t// @ts-expect-error\n\t\t\t\tshard.on(event, (payload) => this.manager.emit(event, { ...payload, shardId }));\n\t\t\t}\n\t\t\tthis.shards.set(shardId, shard);\n\t\t}\n\t}\n\n\tpublic async connect() {\n\t\tconst promises = [];\n\n\t\tfor (const shard of this.shards.values()) {\n\t\t\tawait this.throttler.waitForIdentify();\n\t\t\tpromises.push(shard.connect());\n\t\t}\n\n\t\tawait Promise.all(promises);\n\t}\n\n\tpublic async destroy(options?: Omit<WebSocketShardDestroyOptions, 'recover'>) {\n\t\tconst promises = [];\n\n\t\tfor (const shard of this.shards.values()) {\n\t\t\tpromises.push(shard.destroy(options));\n\t\t}\n\n\t\tawait Promise.all(promises);\n\t\tthis.shards.clear();\n\t}\n\n\tpublic send(shardId: number, payload: GatewaySendPayload) {\n\t\tconst shard = this.shards.get(shardId);\n\t\tif (!shard) throw new Error(`Shard ${shardId} not found`);\n\t\treturn shard.send(payload);\n\t}\n}\n"],"names":[],"mappings":";;;;;;AAKO,MAAM,sBAAsB,CAAC;AACpC,EAAE,WAAW,CAAC,OAAO,EAAE;AACvB,IAAI,IAAI,CAAC,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;AACnC,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,IAAI,IAAI,CAAC,SAAS,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;AACpD,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,QAAQ,EAAE;AACxB,IAAI,MAAM,eAAe,GAAG,MAAM,gCAAgC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AACjF,IAAI,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;AACpC,MAAM,MAAM,QAAQ,GAAG,IAAI,6BAA6B,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;AACxF,MAAM,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AAC1D,MAAM,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE;AAC/D,QAAQ,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;AACxF,OAAO;AACP,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AACtC,KAAK;AACL,GAAG;AACH,EAAE,MAAM,OAAO,GAAG;AAClB,IAAI,MAAM,QAAQ,GAAG,EAAE,CAAC;AACxB,IAAI,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE;AAC9C,MAAM,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;AAC7C,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;AACrC,KAAK;AACL,IAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAChC,GAAG;AACH,EAAE,MAAM,OAAO,CAAC,OAAO,EAAE;AACzB,IAAI,MAAM,QAAQ,GAAG,EAAE,CAAC;AACxB,IAAI,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE;AAC9C,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;AAC5C,KAAK;AACL,IAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAChC,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;AACxB,GAAG;AACH,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE;AACzB,IAAI,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AAC3C,IAAI,IAAI,CAAC,KAAK;AACd,MAAM,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;AACpD,IAAI,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AAC/B,GAAG;AACH;;;;"}

package/dist/strategies/sharding/WorkerShardingStrategy.cjs

@@ -0,0 +1,162 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+const node_events = require('node:events');
+const node_path = require('node:path');
+const node_worker_threads = require('node:worker_threads');
+const collection = require('@discordjs/collection');
+const IdentifyThrottler = require('../../utils/IdentifyThrottler.cjs');
+const IContextFetchingStrategy = require('../context/IContextFetchingStrategy.cjs');
+
+var __accessCheck = (obj, member, msg) => {
+  if (!member.has(obj))
+    throw TypeError("Cannot " + msg);
+};
+var __privateGet = (obj, member, getter) => {
+  __accessCheck(obj, member, "read from private field");
+  return getter ? getter.call(obj) : member.get(obj);
+};
+var __privateAdd = (obj, member, value) => {
+  if (member.has(obj))
+    throw TypeError("Cannot add the same private member more than once");
+  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
+};
+var __privateSet = (obj, member, value, setter) => {
+  __accessCheck(obj, member, "write to private field");
+  setter ? setter.call(obj, value) : member.set(obj, value);
+  return value;
+};
+var _workers, _workerByShardId;
+var WorkerSendPayloadOp = /* @__PURE__ */ ((WorkerSendPayloadOp2) => {
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["Connect"] = 0] = "Connect";
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["Destroy"] = 1] = "Destroy";
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["Send"] = 2] = "Send";
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["SessionInfoResponse"] = 3] = "SessionInfoResponse";
+  return WorkerSendPayloadOp2;
+})(WorkerSendPayloadOp || {});
+var WorkerRecievePayloadOp = /* @__PURE__ */ ((WorkerRecievePayloadOp2) => {
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["Connected"] = 0] = "Connected";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["Destroyed"] = 1] = "Destroyed";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["Event"] = 2] = "Event";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["RetrieveSessionInfo"] = 3] = "RetrieveSessionInfo";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["UpdateSessionInfo"] = 4] = "UpdateSessionInfo";
+  return WorkerRecievePayloadOp2;
+})(WorkerRecievePayloadOp || {});
+class WorkerShardingStrategy {
+  constructor(manager, options) {
+    __privateAdd(this, _workers, []);
+    __privateAdd(this, _workerByShardId, new collection.Collection());
+    this.connectPromises = new collection.Collection();
+    this.destroyPromises = new collection.Collection();
+    this.manager = manager;
+    this.throttler = new IdentifyThrottler.IdentifyThrottler(manager);
+    this.options = options;
+  }
+  async spawn(shardIds) {
+    const shardsPerWorker = this.options.shardsPerWorker === "all" ? shardIds.length : this.options.shardsPerWorker;
+    const strategyOptions = await IContextFetchingStrategy.managerToFetchingStrategyOptions(this.manager);
+    let shards = 0;
+    while (shards !== shardIds.length) {
+      const slice = shardIds.slice(shards, shardsPerWorker + shards);
+      const workerData = {
+        ...strategyOptions,
+        shardIds: slice
+      };
+      const worker = new node_worker_threads.Worker(node_path.join(__dirname, "worker.cjs"), { workerData });
+      await node_events.once(worker, "online");
+      worker.on("error", (err) => {
+        throw err;
+      }).on("messageerror", (err) => {
+        throw err;
+      }).on("message", (payload) => this.onMessage(worker, payload));
+      __privateGet(this, _workers).push(worker);
+      for (const shardId of slice) {
+        __privateGet(this, _workerByShardId).set(shardId, worker);
+      }
+      shards += slice.length;
+    }
+  }
+  async connect() {
+    const promises = [];
+    for (const [shardId, worker] of __privateGet(this, _workerByShardId).entries()) {
+      await this.throttler.waitForIdentify();
+      const payload = {
+        op: 0 /* Connect */,
+        shardId
+      };
+      const promise = new Promise((resolve) => this.connectPromises.set(shardId, resolve));
+      worker.postMessage(payload);
+      promises.push(promise);
+    }
+    await Promise.all(promises);
+  }
+  async destroy(options = {}) {
+    const promises = [];
+    for (const [shardId, worker] of __privateGet(this, _workerByShardId).entries()) {
+      const payload = {
+        op: 1 /* Destroy */,
+        shardId,
+        options
+      };
+      promises.push(new Promise((resolve) => this.destroyPromises.set(shardId, resolve)).then(() => worker.terminate()));
+      worker.postMessage(payload);
+    }
+    __privateSet(this, _workers, []);
+    __privateGet(this, _workerByShardId).clear();
+    await Promise.all(promises);
+  }
+  send(shardId, data) {
+    const worker = __privateGet(this, _workerByShardId).get(shardId);
+    if (!worker) {
+      throw new Error(`No worker found for shard ${shardId}`);
+    }
+    const payload = {
+      op: 2 /* Send */,
+      shardId,
+      payload: data
+    };
+    worker.postMessage(payload);
+  }
+  async onMessage(worker, payload) {
+    switch (payload.op) {
+      case 0 /* Connected */: {
+        const resolve = this.connectPromises.get(payload.shardId);
+        resolve();
+        this.connectPromises.delete(payload.shardId);
+        break;
+      }
+      case 1 /* Destroyed */: {
+        const resolve = this.destroyPromises.get(payload.shardId);
+        resolve();
+        this.destroyPromises.delete(payload.shardId);
+        break;
+      }
+      case 2 /* Event */: {
+        this.manager.emit(payload.event, { ...payload.data, shardId: payload.shardId });
+        break;
+      }
+      case 3 /* RetrieveSessionInfo */: {
+        const session = await this.manager.options.retrieveSessionInfo(payload.shardId);
+        const response = {
+          op: 3 /* SessionInfoResponse */,
+          nonce: payload.nonce,
+          session
+        };
+        worker.postMessage(response);
+        break;
+      }
+      case 4 /* UpdateSessionInfo */: {
+        await this.manager.options.updateSessionInfo(payload.shardId, payload.session);
+        break;
+      }
+    }
+  }
+}
+_workers = new WeakMap();
+_workerByShardId = new WeakMap();
+
+exports.WorkerRecievePayloadOp = WorkerRecievePayloadOp;
+exports.WorkerSendPayloadOp = WorkerSendPayloadOp;
+exports.WorkerShardingStrategy = WorkerShardingStrategy;
+//# sourceMappingURL=WorkerShardingStrategy.cjs.map

package/dist/strategies/sharding/WorkerShardingStrategy.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerShardingStrategy.cjs","sources":["../../../src/strategies/sharding/WorkerShardingStrategy.ts"],"sourcesContent":["import { once } from 'node:events';\nimport { join } from 'node:path';\nimport { Worker } from 'node:worker_threads';\nimport { Collection } from '@discordjs/collection';\nimport type { GatewaySendPayload } from 'discord-api-types/v10';\nimport type { IShardingStrategy } from './IShardingStrategy';\nimport { IdentifyThrottler } from '../../utils/IdentifyThrottler';\nimport type { SessionInfo, WebSocketManager } from '../../ws/WebSocketManager';\nimport type { WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';\nimport { FetchingStrategyOptions, managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy';\n\nexport interface WorkerData extends FetchingStrategyOptions {\n\tshardIds: number[];\n}\n\nexport enum WorkerSendPayloadOp {\n\tConnect,\n\tDestroy,\n\tSend,\n\tSessionInfoResponse,\n}\n\nexport type WorkerSendPayload =\n\t| { op: WorkerSendPayloadOp.Connect; shardId: number }\n\t| { op: WorkerSendPayloadOp.Destroy; shardId: number; options?: WebSocketShardDestroyOptions }\n\t| { op: WorkerSendPayloadOp.Send; shardId: number; payload: GatewaySendPayload }\n\t| { op: WorkerSendPayloadOp.SessionInfoResponse; nonce: number; session: SessionInfo | null };\n\nexport enum WorkerRecievePayloadOp {\n\tConnected,\n\tDestroyed,\n\tEvent,\n\tRetrieveSessionInfo,\n\tUpdateSessionInfo,\n}\n\nexport type WorkerRecievePayload =\n\t| { op: WorkerRecievePayloadOp.Connected; shardId: number }\n\t| { op: WorkerRecievePayloadOp.Destroyed; shardId: number }\n\t// Can't seem to get a type-safe union based off of the event, so I'm sadly leaving data as any for now\n\t| { op: WorkerRecievePayloadOp.Event; shardId: number; event: WebSocketShardEvents; data: any }\n\t| { op: WorkerRecievePayloadOp.RetrieveSessionInfo; shardId: number; nonce: number }\n\t| { op: WorkerRecievePayloadOp.UpdateSessionInfo; shardId: number; session: SessionInfo | null };\n\n/**\n * Options for a {@link WorkerShardingStrategy}\n */\nexport interface WorkerShardingStrategyOptions {\n\t/**\n\t * Dictates how many shards should be spawned per worker thread.\n\t */\n\tshardsPerWorker: number | 'all';\n}\n\n/**\n * Strategy used to spawn threads in worker_threads\n */\nexport class WorkerShardingStrategy implements IShardingStrategy {\n\tprivate readonly manager: WebSocketManager;\n\tprivate readonly options: WorkerShardingStrategyOptions;\n\n\t#workers: Worker[] = [];\n\treadonly #workerByShardId = new Collection<number, Worker>();\n\n\tprivate readonly connectPromises = new Collection<number, () => void>();\n\tprivate readonly destroyPromises = new Collection<number, () => void>();\n\n\tprivate readonly throttler: IdentifyThrottler;\n\n\tpublic constructor(manager: WebSocketManager, options: WorkerShardingStrategyOptions) {\n\t\tthis.manager = manager;\n\t\tthis.throttler = new IdentifyThrottler(manager);\n\t\tthis.options = options;\n\t}\n\n\tpublic async spawn(shardIds: number[]) {\n\t\tconst shardsPerWorker = this.options.shardsPerWorker === 'all' ? shardIds.length : this.options.shardsPerWorker;\n\t\tconst strategyOptions = await managerToFetchingStrategyOptions(this.manager);\n\n\t\tlet shards = 0;\n\t\twhile (shards !== shardIds.length) {\n\t\t\tconst slice = shardIds.slice(shards, shardsPerWorker + shards);\n\t\t\tconst workerData: WorkerData = {\n\t\t\t\t...strategyOptions,\n\t\t\t\tshardIds: slice,\n\t\t\t};\n\n\t\t\tconst worker = new Worker(join(__dirname, 'worker.cjs'), { workerData });\n\t\t\tawait once(worker, 'online');\n\t\t\tworker\n\t\t\t\t.on('error', (err) => {\n\t\t\t\t\tthrow err;\n\t\t\t\t})\n\t\t\t\t.on('messageerror', (err) => {\n\t\t\t\t\tthrow err;\n\t\t\t\t})\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-misused-promises\n\t\t\t\t.on('message', (payload: WorkerRecievePayload) => this.onMessage(worker, payload));\n\n\t\t\tthis.#workers.push(worker);\n\t\t\tfor (const shardId of slice) {\n\t\t\t\tthis.#workerByShardId.set(shardId, worker);\n\t\t\t}\n\n\t\t\tshards += slice.length;\n\t\t}\n\t}\n\n\tpublic async connect() {\n\t\tconst promises = [];\n\n\t\tfor (const [shardId, worker] of this.#workerByShardId.entries()) {\n\t\t\tawait this.throttler.waitForIdentify();\n\n\t\t\tconst payload: WorkerSendPayload = {\n\t\t\t\top: WorkerSendPayloadOp.Connect,\n\t\t\t\tshardId,\n\t\t\t};\n\n\t\t\tconst promise = new Promise<void>((resolve) => this.connectPromises.set(shardId, resolve));\n\t\t\tworker.postMessage(payload);\n\t\t\tpromises.push(promise);\n\t\t}\n\n\t\tawait Promise.all(promises);\n\t}\n\n\tpublic async destroy(options: Omit<WebSocketShardDestroyOptions, 'recover'> = {}) {\n\t\tconst promises = [];\n\n\t\tfor (const [shardId, worker] of this.#workerByShardId.entries()) {\n\t\t\tconst payload: WorkerSendPayload = {\n\t\t\t\top: WorkerSendPayloadOp.Destroy,\n\t\t\t\tshardId,\n\t\t\t\toptions,\n\t\t\t};\n\n\t\t\tpromises.push(\n\t\t\t\tnew Promise<void>((resolve) => this.destroyPromises.set(shardId, resolve)).then(() => worker.terminate()),\n\t\t\t);\n\t\t\tworker.postMessage(payload);\n\t\t}\n\n\t\tthis.#workers = [];\n\t\tthis.#workerByShardId.clear();\n\n\t\tawait Promise.all(promises);\n\t}\n\n\tpublic send(shardId: number, data: GatewaySendPayload) {\n\t\tconst worker = this.#workerByShardId.get(shardId);\n\t\tif (!worker) {\n\t\t\tthrow new Error(`No worker found for shard ${shardId}`);\n\t\t}\n\n\t\tconst payload: WorkerSendPayload = {\n\t\t\top: WorkerSendPayloadOp.Send,\n\t\t\tshardId,\n\t\t\tpayload: data,\n\t\t};\n\t\tworker.postMessage(payload);\n\t}\n\n\tprivate async onMessage(worker: Worker, payload: WorkerRecievePayload) {\n\t\tswitch (payload.op) {\n\t\t\tcase WorkerRecievePayloadOp.Connected: {\n\t\t\t\tconst resolve = this.connectPromises.get(payload.shardId)!;\n\t\t\t\tresolve();\n\t\t\t\tthis.connectPromises.delete(payload.shardId);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.Destroyed: {\n\t\t\t\tconst resolve = this.destroyPromises.get(payload.shardId)!;\n\t\t\t\tresolve();\n\t\t\t\tthis.destroyPromises.delete(payload.shardId);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.Event: {\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-argument\n\t\t\t\tthis.manager.emit(payload.event, { ...payload.data, shardId: payload.shardId });\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.RetrieveSessionInfo: {\n\t\t\t\tconst session = await this.manager.options.retrieveSessionInfo(payload.shardId);\n\t\t\t\tconst response: WorkerSendPayload = {\n\t\t\t\t\top: WorkerSendPayloadOp.SessionInfoResponse,\n\t\t\t\t\tnonce: payload.nonce,\n\t\t\t\t\tsession,\n\t\t\t\t};\n\t\t\t\tworker.postMessage(response);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.UpdateSessionInfo: {\n\t\t\t\tawait this.manager.options.updateSessionInfo(payload.shardId, payload.session);\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t}\n}\n"],"names":["Collection","IdentifyThrottler","managerToFetchingStrategyOptions","Worker","join","once"],"mappings":";;;;;;;;;;;AAAA,IAAI,aAAa,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,KAAK;AAC1C,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;AACtB,IAAI,MAAM,SAAS,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;AACrC,CAAC,CAAC;AACF,IAAI,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,KAAK;AAC5C,EAAE,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,yBAAyB,CAAC,CAAC;AACxD,EAAE,OAAO,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACrD,CAAC,CAAC;AACF,IAAI,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,KAAK;AAC3C,EAAE,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;AACrB,IAAI,MAAM,SAAS,CAAC,mDAAmD,CAAC,CAAC;AACzE,EAAE,MAAM,YAAY,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;AACvE,CAAC,CAAC;AACF,IAAI,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK;AACnD,EAAE,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,wBAAwB,CAAC,CAAC;AACvD,EAAE,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;AAC5D,EAAE,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AACF,IAAI,QAAQ,EAAE,gBAAgB,CAAC;AAOrB,IAAC,mBAAmB,mBAAmB,CAAC,CAAC,oBAAoB,KAAK;AAC5E,EAAE,oBAAoB,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC;AACxE,EAAE,oBAAoB,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC;AACxE,EAAE,oBAAoB,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC;AAClE,EAAE,oBAAoB,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,GAAG,qBAAqB,CAAC;AAChG,EAAE,OAAO,oBAAoB,CAAC;AAC9B,CAAC,EAAE,mBAAmB,IAAI,EAAE,EAAE;AACpB,IAAC,sBAAsB,mBAAmB,CAAC,CAAC,uBAAuB,KAAK;AAClF,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC;AAClF,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC;AAClF,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC;AAC1E,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,GAAG,qBAAqB,CAAC;AACtG,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,GAAG,mBAAmB,CAAC;AAClG,EAAE,OAAO,uBAAuB,CAAC;AACjC,CAAC,EAAE,sBAAsB,IAAI,EAAE,EAAE;AAC1B,MAAM,sBAAsB,CAAC;AACpC,EAAE,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE;AAChC,IAAI,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;AACrC,IAAI,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,IAAIA,qBAAU,EAAE,CAAC,CAAC;AAC3D,IAAI,IAAI,CAAC,eAAe,GAAG,IAAIA,qBAAU,EAAE,CAAC;AAC5C,IAAI,IAAI,CAAC,eAAe,GAAG,IAAIA,qBAAU,EAAE,CAAC;AAC5C,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,IAAI,IAAI,CAAC,SAAS,GAAG,IAAIC,mCAAiB,CAAC,OAAO,CAAC,CAAC;AACpD,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,QAAQ,EAAE;AACxB,IAAI,MAAM,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,KAAK,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;AACpH,IAAI,MAAM,eAAe,GAAG,MAAMC,yDAAgC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AACjF,IAAI,IAAI,MAAM,GAAG,CAAC,CAAC;AACnB,IAAI,OAAO,MAAM,KAAK,QAAQ,CAAC,MAAM,EAAE;AACvC,MAAM,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,GAAG,MAAM,CAAC,CAAC;AACrE,MAAM,MAAM,UAAU,GAAG;AACzB,QAAQ,GAAG,eAAe;AAC1B,QAAQ,QAAQ,EAAE,KAAK;AACvB,OAAO,CAAC;AACR,MAAM,MAAM,MAAM,GAAG,IAAIC,0BAAM,CAACC,cAAI,CAAC,SAAS,EAAE,YAAY,CAAC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;AAC/E,MAAM,MAAMC,gBAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;AACnC,MAAM,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,KAAK;AAClC,QAAQ,MAAM,GAAG,CAAC;AAClB,OAAO,CAAC,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,GAAG,KAAK;AACrC,QAAQ,MAAM,GAAG,CAAC;AAClB,OAAO,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AACrE,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChD,MAAM,KAAK,MAAM,OAAO,IAAI,KAAK,EAAE;AACnC,QAAQ,YAAY,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;AAClE,OAAO;AACP,MAAM,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;AAC7B,KAAK;AACL,GAAG;AACH,EAAE,MAAM,OAAO,GAAG;AAClB,IAAI,MAAM,QAAQ,GAAG,EAAE,CAAC;AACxB,IAAI,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,OAAO,EAAE,EAAE;AACpF,MAAM,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;AAC7C,MAAM,MAAM,OAAO,GAAG;AACtB,QAAQ,EAAE,EAAE,CAAC;AACb,QAAQ,OAAO;AACf,OAAO,CAAC;AACR,MAAM,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;AAC3F,MAAM,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AAClC,MAAM,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AAC7B,KAAK;AACL,IAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAChC,GAAG;AACH,EAAE,MAAM,OAAO,CAAC,OAAO,GAAG,EAAE,EAAE;AAC9B,IAAI,MAAM,QAAQ,GAAG,EAAE,CAAC;AACxB,IAAI,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,OAAO,EAAE,EAAE;AACpF,MAAM,MAAM,OAAO,GAAG;AACtB,QAAQ,EAAE,EAAE,CAAC;AACb,QAAQ,OAAO;AACf,QAAQ,OAAO;AACf,OAAO,CAAC;AACR,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;AACzH,MAAM,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AAClC,KAAK;AACL,IAAI,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;AACrC,IAAI,YAAY,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,KAAK,EAAE,CAAC;AACjD,IAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAChC,GAAG;AACH,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE;AACtB,IAAI,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACrE,IAAI,IAAI,CAAC,MAAM,EAAE;AACjB,MAAM,MAAM,IAAI,KAAK,CAAC,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;AAC9D,KAAK;AACL,IAAI,MAAM,OAAO,GAAG;AACpB,MAAM,EAAE,EAAE,CAAC;AACX,MAAM,OAAO;AACb,MAAM,OAAO,EAAE,IAAI;AACnB,KAAK,CAAC;AACN,IAAI,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AAChC,GAAG;AACH,EAAE,MAAM,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE;AACnC,IAAI,QAAQ,OAAO,CAAC,EAAE;AACtB,MAAM,KAAK,CAAC,kBAAkB;AAC9B,QAAQ,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAClE,QAAQ,OAAO,EAAE,CAAC;AAClB,QAAQ,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACrD,QAAQ,MAAM;AACd,OAAO;AACP,MAAM,KAAK,CAAC,kBAAkB;AAC9B,QAAQ,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAClE,QAAQ,OAAO,EAAE,CAAC;AAClB,QAAQ,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACrD,QAAQ,MAAM;AACd,OAAO;AACP,MAAM,KAAK,CAAC,cAAc;AAC1B,QAAQ,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;AACxF,QAAQ,MAAM;AACd,OAAO;AACP,MAAM,KAAK,CAAC,4BAA4B;AACxC,QAAQ,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACxF,QAAQ,MAAM,QAAQ,GAAG;AACzB,UAAU,EAAE,EAAE,CAAC;AACf,UAAU,KAAK,EAAE,OAAO,CAAC,KAAK;AAC9B,UAAU,OAAO;AACjB,SAAS,CAAC;AACV,QAAQ,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;AACrC,QAAQ,MAAM;AACd,OAAO;AACP,MAAM,KAAK,CAAC,0BAA0B;AACtC,QAAQ,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;AACvF,QAAQ,MAAM;AACd,OAAO;AACP,KAAK;AACL,GAAG;AACH,CAAC;AACD,QAAQ,GAAG,IAAI,OAAO,EAAE,CAAC;AACzB,gBAAgB,GAAG,IAAI,OAAO,EAAE;;;;;;"}

package/dist/strategies/sharding/WorkerShardingStrategy.d.ts

@@ -0,0 +1,84 @@
+import type { GatewaySendPayload } from 'discord-api-types/v10';
+import type { IShardingStrategy } from './IShardingStrategy';
+import type { SessionInfo, WebSocketManager } from '../../ws/WebSocketManager';
+import type { WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';
+import { FetchingStrategyOptions } from '../context/IContextFetchingStrategy';
+export interface WorkerData extends FetchingStrategyOptions {
+    shardIds: number[];
+}
+export declare enum WorkerSendPayloadOp {
+    Connect = 0,
+    Destroy = 1,
+    Send = 2,
+    SessionInfoResponse = 3
+}
+export declare type WorkerSendPayload = {
+    op: WorkerSendPayloadOp.Connect;
+    shardId: number;
+} | {
+    op: WorkerSendPayloadOp.Destroy;
+    shardId: number;
+    options?: WebSocketShardDestroyOptions;
+} | {
+    op: WorkerSendPayloadOp.Send;
+    shardId: number;
+    payload: GatewaySendPayload;
+} | {
+    op: WorkerSendPayloadOp.SessionInfoResponse;
+    nonce: number;
+    session: SessionInfo | null;
+};
+export declare enum WorkerRecievePayloadOp {
+    Connected = 0,
+    Destroyed = 1,
+    Event = 2,
+    RetrieveSessionInfo = 3,
+    UpdateSessionInfo = 4
+}
+export declare type WorkerRecievePayload = {
+    op: WorkerRecievePayloadOp.Connected;
+    shardId: number;
+} | {
+    op: WorkerRecievePayloadOp.Destroyed;
+    shardId: number;
+} | {
+    op: WorkerRecievePayloadOp.Event;
+    shardId: number;
+    event: WebSocketShardEvents;
+    data: any;
+} | {
+    op: WorkerRecievePayloadOp.RetrieveSessionInfo;
+    shardId: number;
+    nonce: number;
+} | {
+    op: WorkerRecievePayloadOp.UpdateSessionInfo;
+    shardId: number;
+    session: SessionInfo | null;
+};
+/**
+ * Options for a {@link WorkerShardingStrategy}
+ */
+export interface WorkerShardingStrategyOptions {
+    /**
+     * Dictates how many shards should be spawned per worker thread.
+     */
+    shardsPerWorker: number | 'all';
+}
+/**
+ * Strategy used to spawn threads in worker_threads
+ */
+export declare class WorkerShardingStrategy implements IShardingStrategy {
+    #private;
+    private readonly manager;
+    private readonly options;
+    private readonly connectPromises;
+    private readonly destroyPromises;
+    private readonly throttler;
+    constructor(manager: WebSocketManager, options: WorkerShardingStrategyOptions);
+    spawn(shardIds: number[]): Promise<void>;
+    connect(): Promise<void>;
+    destroy(options?: Omit<WebSocketShardDestroyOptions, 'recover'>): Promise<void>;
+    send(shardId: number, data: GatewaySendPayload): void;
+    private onMessage;
+}
+//# sourceMappingURL=WorkerShardingStrategy.d.ts.map

package/dist/strategies/sharding/WorkerShardingStrategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerShardingStrategy.d.ts","sourceRoot":"","sources":["../../../src/strategies/sharding/WorkerShardingStrategy.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAE7D,OAAO,KAAK,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC/E,OAAO,KAAK,EAAE,4BAA4B,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAClG,OAAO,EAAE,uBAAuB,EAAoC,MAAM,qCAAqC,CAAC;AAEhH,MAAM,WAAW,UAAW,SAAQ,uBAAuB;IAC1D,QAAQ,EAAE,MAAM,EAAE,CAAC;CACnB;AAED,oBAAY,mBAAmB;IAC9B,OAAO,IAAA;IACP,OAAO,IAAA;IACP,IAAI,IAAA;IACJ,mBAAmB,IAAA;CACnB;AAED,oBAAY,iBAAiB,GAC1B;IAAE,EAAE,EAAE,mBAAmB,CAAC,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACpD;IAAE,EAAE,EAAE,mBAAmB,CAAC,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,4BAA4B,CAAA;CAAE,GAC5F;IAAE,EAAE,EAAE,mBAAmB,CAAC,IAAI,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,kBAAkB,CAAA;CAAE,GAC9E;IAAE,EAAE,EAAE,mBAAmB,CAAC,mBAAmB,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,WAAW,GAAG,IAAI,CAAA;CAAE,CAAC;AAE/F,oBAAY,sBAAsB;IACjC,SAAS,IAAA;IACT,SAAS,IAAA;IACT,KAAK,IAAA;IACL,mBAAmB,IAAA;IACnB,iBAAiB,IAAA;CACjB;AAED,oBAAY,oBAAoB,GAC7B;IAAE,EAAE,EAAE,sBAAsB,CAAC,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACzD;IAAE,EAAE,EAAE,sBAAsB,CAAC,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAEzD;IAAE,EAAE,EAAE,sBAAsB,CAAC,KAAK,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,oBAAoB,CAAC;IAAC,IAAI,EAAE,GAAG,CAAA;CAAE,GAC7F;IAAE,EAAE,EAAE,sBAAsB,CAAC,mBAAmB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAClF;IAAE,EAAE,EAAE,sBAAsB,CAAC,iBAAiB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,WAAW,GAAG,IAAI,CAAA;CAAE,CAAC;AAElG;;GAEG;AACH,MAAM,WAAW,6BAA6B;IAC7C;;OAEG;IACH,eAAe,EAAE,MAAM,GAAG,KAAK,CAAC;CAChC;AAED;;GAEG;AACH,qBAAa,sBAAuB,YAAW,iBAAiB;;IAC/D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAgC;IAKxD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAwC;IACxE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAwC;IAExE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAoB;gBAE3B,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,6BAA6B;IAMvE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE;IAiCxB,OAAO;IAmBP,OAAO,CAAC,OAAO,GAAE,IAAI,CAAC,4BAA4B,EAAE,SAAS,CAAM;IAsBzE,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB;YAcvC,SAAS;CAuCvB"}