Function main

scripts/daemon-memory-check.ts:497–619

()

}

async function main() {
  const options = parseArgs(process.argv.slice(2));
  const port = await reserveLoopbackPort();
  const scratch = mkdtempSync(join(tmpdir(), "hunk-daemon-memory-"));
  const child = Bun.spawn([process.execPath, "src/main.tsx", "daemon", "serve"], {
    cwd: process.cwd(),
    env: {
      ...process.env,
      HUNK_MCP_HOST: "127.0.0.1",
      HUNK_MCP_PORT: String(port),
      HUNK_CONFIG_HOME: scratch,
      HUNK_NO_UPDATE_CHECK: "1",
    },
    stdout: "pipe",
    stderr: "pipe",
  });

  const samples: MemorySample[] = [];

  try {
    const health = await waitUntil("daemon health", () => readHealth(port), 10_000);
    const pid = health.pid;
    console.log(`daemon pid=${pid} port=${port}`);
    samples.push(await sample("startup", 0, pid, port));

    for (let cycle = 1; cycle <= options.cycles; cycle += 1) {
      const sockets: WebSocket[] = [];
      const sessionIds = Array.from(
        { length: options.sessionsPerCycle },
        (_, sessionIndex) => `memory-${cycle}-${sessionIndex}`,
      );

      for (let sessionIndex = 0; sessionIndex < sessionIds.length; sessionIndex += 1) {
        sockets.push(
          await openRegisteredSocket(port, sessionIds[sessionIndex]!, cycle, sessionIndex, options),
        );
      }

      await waitUntil("session registration", async () => {
        const nextHealth = await readHealth(port);
        return nextHealth?.sessions === options.sessionsPerCycle ? nextHealth : null;
      });
      await sendSnapshotUpdates(sockets, sessionIds, options.snapshotUpdatesPerCycle, options);
      await exerciseSessionApi(port, sessionIds, options.apiRequestsPerCycle);
      samples.push(await sample("live", cycle, pid, port));

      await closeSockets(sockets);
      await waitUntil("session cleanup", async () => {
        const nextHealth = await readHealth(port);
        return nextHealth?.sessions === 0 ? nextHealth : null;
      });
      await Bun.sleep(options.settleMs);
      const cleanupSample = await sample("cleanup", cycle, pid, port);
      samples.push(cleanupSample);

      console.log(
        `cycle=${cycle.toString().padStart(3)} cleanup_rss=${formatBytes(cleanupSample.rssBytes)} ` +
          `heap=${formatBytes(cleanupSample.heapBytes)} hwm=${formatBytes(cleanupSample.highWaterRssBytes)}`,