swc/crates/swc_bundler/tests/.cache/deno/c53d5a1a462c7cf4f18b3a1d58bb97d9b5c2cf36.ts
2021-11-09 20:42:49 +09:00

65 lines
1.7 KiB
TypeScript

// Loaded from https://deno.land/x/discordeno@11.0.0-rc.2/src/ws/process_queue.ts
import { loopObject } from "../util/loop_object.ts";
import { delay } from "../util/utils.ts";
import { ws } from "./ws.ts";
export async function processQueue(id: number) {
const shard = ws.shards.get(id);
// If no items or its already processing then exit
if (!shard?.queue.length || shard.processingQueue) return;
shard.processingQueue = true;
while (shard.queue.length) {
if (shard.ws.readyState !== WebSocket.OPEN) {
shard.processingQueue = false;
return;
}
const now = Date.now();
if (now - shard.queueStartedAt >= 60000) {
shard.queueStartedAt = now;
shard.queueCounter = 0;
}
// Send a request that is next in line
const request = shard.queue.shift();
if (!request) return;
if (request?.d) {
request.d = loopObject(
request.d as Record<string, unknown>,
(value) =>
typeof value === "bigint"
? value.toString()
: Array.isArray(value)
? value.map((v) => (typeof v === "bigint" ? v.toString() : v))
: value,
`Running forEach loop in ws.processQueue function for changing bigints to strings.`
);
}
ws.log("RAW_SEND", shard.id, request);
shard.ws.send(JSON.stringify(request));
// Counter is useful for preventing 120/m requests.
shard.queueCounter++;
// Handle if the requests have been maxed
if (shard.queueCounter >= 118) {
ws.log("DEBUG", {
message: "Max gateway requests per minute reached setting timeout for one minute",
shardId: shard.id,
});
await delay(60000);
shard.queueCounter = 0;
continue;
}
}
shard.processingQueue = false;
}