seed-1k-load.mjs

seed-1k-load.mjstests/performance/helpers/seed-1k-load.mjsraw ← helpers5,864 B← Back to the app
1#!/usr/bin/env node
2/**
3 * AI Software Factory — 1,000-connection load fixture seeder (TASK-0091).
4 *
5 * Creates the dedicated perf workspace + 1,000 load users, each with:
6 *   - a WorkspaceMember (MEMBER) row, and
7 *   - a workspace-scoped Permission grant (granteeType "user", level
8 *     EDIT) — the resolver in apps/realtime/src/prisma-store.ts assembles
9 *     role baseline (MEMBER → view) + grants → EDIT, so every load user
10 *     gets canEdit=true on the load pages.
11 *
12 * Why the load users need their OWN identities (recorded in
13 * realtime-1k.js): the server caps live sockets at 100/user and control
14 * sockets at 10/user — one shared token can never reach 1,000
15 * connections. The k6 VUs mint their own JWTs (k6/crypto) for these ids;
16 * this seeder only creates the DB rows.
17 *
18 * Re-running is a no-op (it checks the existing user count).
19 *
20 * Usage (repo root, Prisma generated + DATABASE_URL pointing at the load
21 * Postgres):
22 *
23 *   K6_WORKSPACE_ID=load-ws node tests/performance/helpers/seed-1k-load.mjs
24 *
25 * Env:
26 *   K6_WORKSPACE_ID  workspace id to create/reuse (default "load-ws")
27 *   K6_USERS         load user count (default 1000)
28 *   K6_PAGES         comma-separated load page ids to create (default:
29 *                    10 pages load-p-000 … load-p-009 — 100 peers/room)
30 *   DATABASE_URL     Prisma DSN (default postgresql://factory:factory@localhost:5432/factory)
31 */
32import { PrismaClient } from "@prisma/client";
33 
34const WORKSPACE_ID = process.env.K6_WORKSPACE_ID || "load-ws";
35const USER_COUNT = parseInt(process.env.K6_USERS || "1000", 10);
36const PAGE_IDS = String(process.env.K6_PAGES || "")
37  .split(",")
38  .filter(Boolean);
39const pages =
40  PAGE_IDS.length > 0
41    ? PAGE_IDS
42    : Array.from({ length: 10 }, (_, i) => `load-p-${String(i).padStart(3, "0")}`);
43 
44const prisma = new PrismaClient();
45 
46function pad4(n) {
47  return String(n).padStart(4, "0");
48}
49 
50async function main() {
51  // 0-BASED to match the k6 script, which mints VU __VU (1-based) as
52  // k6-load-u-XXXX for XXXX = __VU - 1 → k6-load-u-0000 … k6-load-u-0999.
53  // (An earlier 1-based seeder left VU 1 pointing at a non-existent user →
54  // a 403 join-failure on the first 1k run — TASK-0095 S14 fix.)
55  const membership = await prisma.workspaceMember.count({ where: { workspaceId: WORKSPACE_ID } });
56  if (membership >= USER_COUNT) {
57    console.log(`already seeded (${membership} memberships in ${WORKSPACE_ID} ≥ ${USER_COUNT})`);
58    console.log(`workspace=${WORKSPACE_ID} pages=${pages.join(",")}`);
59    return;
60  }
61 
62  // 1. The perf workspace (reused if it exists from a prior partial run).
63  const ws = await prisma.workspace.upsert({
64    where: { id: WORKSPACE_ID },
65    update: {},
66    create: {
67      id: WORKSPACE_ID,
68      name: "k6 Load (TASK-0091)",
69      slug: `k6-load-${WORKSPACE_ID}`,
70      settings: {},
71    },
72  });
73  if (!ws) throw new Error("workspace upsert failed");
74 
75  // 2. The load pages (the rooms the VUs join, 100 peers each by default).
76  for (const pageId of pages) {
77    await prisma.page.upsert({
78      where: { id: pageId },
79      update: {},
80      create: { id: pageId, workspaceId: WORKSPACE_ID, title: `k6 load page ${pageId}` },
81    });
82  }
83 
84  // 3. The 1,000 load users + memberships + EDIT grants (batched).
85  const BATCH = 500;
86  let created = 0;
87  for (let start = 0; start < USER_COUNT; start += BATCH) {
88    const ids = [];
89    const users = [];
90    const members = [];
91    const grants = [];
92    for (let i = start; i < Math.min(start + BATCH, USER_COUNT); i += 1) {
93      // 0-based (see the header note): k6-load-u-0000 … k6-load-u-0999.
94      const id = `k6-load-u-${pad4(i)}`;
95      ids.push(id);
96      users.push({ id, email: `${id}@factory.local`, name: `k6 Load ${pad4(i + 1)}` });
97      members.push({ workspaceId: WORKSPACE_ID, userId: id, role: "MEMBER" });
98      // Workspace-scoped EDIT: the resolver's applicable-grant filter keeps
99      // `resourceType === "workspace" && resourceId === workspaceId`, and
100      // the shared engine's max(role baseline view, grant edit) = edit.
101      grants.push({
102        workspaceId: WORKSPACE_ID,
103        granteeType: "user",
104        granteeId: id,
105        resourceType: "workspace",
106        resourceId: WORKSPACE_ID,
107        level: "EDIT",
108      });
109    }
110    await prisma.$transaction([
111      ...users.map((u) => prisma.user.upsert({ where: { id: u.id }, update: {}, create: u })),
112      ...members.map((m) =>
113        // The compound-unique `where` takes ONLY the two key fields (workspaceId,
114        // userId) — passing the whole member object (with `role`) is rejected by
115        // Prisma as an unknown `where` argument (found during TASK-0135).
116        prisma.workspaceMember.upsert({
117          where: { workspaceId_userId: { workspaceId: m.workspaceId, userId: m.userId } },
118          update: {},
119          create: m,
120        }),
121      ),
122      ...grants.map((g) =>
123        prisma.permission.upsert({
124          where: {
125            workspaceId_granteeType_granteeId_resourceType_resourceId: {
126              workspaceId: g.workspaceId,
127              granteeType: g.granteeType,
128              granteeId: g.granteeId,
129              resourceType: g.resourceType,
130              resourceId: g.resourceId,
131            },
132          },
133          update: {},
134          create: g,
135        }),
136      ),
137    ]);
138    created += ids.length;
139    process.stdout.write(`  ${created}/${USER_COUNT}\r`);
140  }
141  console.log(`\nseeded ${created} load users in workspace ${WORKSPACE_ID}`);
142  console.log(`workspace=${WORKSPACE_ID} pages=${pages.join(",")}`);
143  console.log("run the suite with:");
144  console.log(`  K6_WORKSPACE_ID=${WORKSPACE_ID} K6_PAGE_IDS=${pages.join(",")} pnpm --dir tests/performance realtime:1k`);
145}
146 
147main()
148  .catch((err) => {
149    console.error(err);
150    process.exitCode = 1;
151  })
152  .finally(() => prisma.$disconnect());
153