Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
1234567891011121314151617181920212223242526272829303132333435363738394041424344import { and, Database, inArray } from "@opencode-ai/console-core/drizzle/index.js"import { ModelTpsRateLimitTable } from "@opencode-ai/console-core/schema/ip.sql.js"
type Result = Record<string, { interval: number; qualify: number; unqualify: number }[]>
export default { async fetch(request: Request) { if (request.method !== "POST") return new Response("Method Not Allowed", { status: 405 })
const body = (await request.json()) as { ids: string[] } const ids = body.ids if (ids.length === 0) return Response.json({} satisfies Result)
const toInterval = (date: Date) => parseInt( date .toISOString() .replace(/[^0-9]/g, "") .substring(0, 12), ) const now = Date.now() const intervals = Array.from({ length: 30 }, (_, i) => toInterval(new Date(now - i * 60 * 1000)))
const rows = await Database.use((tx) => tx .select() .from(ModelTpsRateLimitTable) .where(and(inArray(ModelTpsRateLimitTable.id, ids), inArray(ModelTpsRateLimitTable.interval, intervals))), )
const rowsByKey = new Map(rows.map((row) => [`${row.id}:${row.interval}`, row])) const result: Result = Object.fromEntries( ids.map((id) => [ id, intervals.map((interval) => { const row = rowsByKey.get(`${id}:${interval}`) return { interval, qualify: row?.qualify ?? 0, unqualify: row?.unqualify ?? 0 } }), ]), ) return Response.json(result) },}