// 子进程:对同一 key 连续扣 N 个令牌,每次成功输出一行时间戳(毫秒)。由 rate-limit.test.ts 以 Deno 子进程启动。
import { Kysely, PostgresDialect } from "kysely";
import pg from "pg";
import { PgTokenBucket } from "../../src/rate-limit.ts";

const [url, schema, key, n, capacity, rate] = Deno.args as [string, string, string, string, string, string];
const db = new Kysely<unknown>({
  dialect: new PostgresDialect({
    pool: new pg.Pool({ connectionString: url, max: 1, options: `-c search_path=${schema},public` }),
  }),
});
const bucket = new PgTokenBucket(db);
try {
  for (let i = 0; i < Number(n); i++) {
    const r = await bucket.acquire(key, 1, { capacity: Number(capacity), refillPerSec: Number(rate), waitMs: 30_000 });
    if (!r.ok) throw new Error("等待超时");
    console.log(Date.now());
  }
} finally {
  await db.destroy();
}