Function updateStats

src/cluster/node.ts:25–87

(memOnly: boolean = false)

let totalMem = 0;

function updateStats(memOnly: boolean = false): Promise<void> {
  return new Promise((resolve, reject) => {
    if (!memOnly) {
      serverCount = 0;
      shardData = [];
    }
    responseCount = 0;
    totalMem = process.memoryUsage().heapUsed;
    clusterCount = processes.length;
    let timeout: ReturnType<typeof setTimeout> | undefined;
    const done: boolean[] = Array(clusterCount).fill(false);
    for (const [i, worker] of processes.entries()) {
      if (!worker.isConnected()) {
        clusterCount -= 1;
        if (responseCount >= clusterCount) {
          logger.debug(
            `Stats collected (clusterCount: ${clusterCount}, responseCount: ${responseCount}, serverCount: ${serverCount}, totalMem: ${totalMem})`,
          );
          resolve();
        }
        continue;
      }

      const listener = (packet: IncomingProcMessage) => {
        if (packet.data?.type === "serverCounts") {
          const countData = packet as ServerCountMessage;
          clearTimeout(timeout);
          if (done[i]) return;
          done[i] = true;
          if (!memOnly) {
            serverCount += countData.data.guilds;
            shardData = [...shardData, ...countData.data.shards].sort((a, b) => a.id - b.id);
          }
          responseCount += 1;
          totalMem += countData.data.mem;
          if (responseCount >= clusterCount) {
            worker.off("message", listener);
            logger.debug(
              `Stats collected (clusterCount: ${clusterCount}, responseCount: ${responseCount}, serverCount: ${serverCount}, totalMem: ${totalMem})`,
            );
            resolve();
          }
          timeout = setTimeout(() => {
            worker.off("message", listener);
            reject("Timed out while waiting for stats");
          }, 5000);
        }
      };

      timeout = setTimeout(() => {
        worker.off("message", listener);
        reject("Timed out while waiting for stats");
      }, 5000);

      worker.on("message", listener);
      worker.send({
        data: {
          type: "serverCounts",