formulas

packages/db/scripts/ch-update-sessions-with-revenue.ts

@@ -0,0 +1,96 @@
+import { TABLE_NAMES, ch } from '../src/clickhouse/client';
+import { clix } from '../src/clickhouse/query-builder';
+
+const START_DATE = new Date('2025-11-10T00:00:00Z');
+const END_DATE = new Date('2025-11-20T23:00:00Z');
+const SESSIONS_PER_HOUR = 2;
+
+// Revenue between $10 (1000 cents) and $200 (20000 cents)
+const MIN_REVENUE = 1000;
+const MAX_REVENUE = 20000;
+
+function getRandomRevenue() {
+  return (
+    Math.floor(Math.random() * (MAX_REVENUE - MIN_REVENUE + 1)) + MIN_REVENUE
+  );
+}
+
+async function main() {
+  console.log(
+    `Starting revenue update for sessions between ${START_DATE.toISOString()} and ${END_DATE.toISOString()}`,
+  );
+
+  let currentDate = new Date(START_DATE);
+
+  while (currentDate < END_DATE) {
+    const nextHour = new Date(currentDate.getTime() + 60 * 60 * 1000);
+    console.log(`Processing hour: ${currentDate.toISOString()}`);
+
+    // 1. Pick random sessions for this hour
+    const sessions = await clix(ch)
+      .from(TABLE_NAMES.sessions)
+      .select(['id'])
+      .where('created_at', '>=', currentDate)
+      .andWhere('created_at', '<', nextHour)
+      .where('project_id', '=', 'public-web')
+      .limit(SESSIONS_PER_HOUR)
+      .execute();
+
+    if (sessions.length === 0) {
+      console.log(`No sessions found for ${currentDate.toISOString()}`);
+      currentDate = nextHour;
+      continue;
+    }
+
+    const sessionIds = sessions.map((s: any) => s.id);
+    console.log(
+      `Found ${sessionIds.length} sessions to update: ${sessionIds.join(', ')}`,
+    );
+
+    // 2. Construct update query
+    // We want to assign a DIFFERENT random revenue to each session
+    // Query: ALTER TABLE sessions UPDATE revenue = if(id='id1', rev1, if(id='id2', rev2, ...)) WHERE id IN ('id1', 'id2', ...)
+
+    const updates: { id: string; revenue: number }[] = [];
+
+    for (const id of sessionIds) {
+      const revenue = getRandomRevenue();
+      updates.push({ id, revenue });
+    }
+
+    // Build nested if() for the update expression
+    // ClickHouse doesn't have CASE WHEN in UPDATE expression in the same way, but if() works.
+    // Actually multiIf is cleaner: multiIf(id='id1', rev1, id='id2', rev2, revenue)
+
+    const conditions = updates
+      .map((u) => `id = '${u.id}', ${u.revenue}`)
+      .join(', ');
+    const updateExpr = `multiIf(${conditions}, revenue)`;
+
+    const idsStr = sessionIds.map((id: string) => `'${id}'`).join(', ');
+    const query = `ALTER TABLE ${TABLE_NAMES.sessions} UPDATE revenue = ${updateExpr} WHERE id IN (${idsStr})`;
+
+    console.log(`Executing update: ${query}`);
+
+    try {
+      await ch.command({
+        query,
+      });
+      console.log('Update command sent.');
+
+      // Wait a bit to not overload mutations if running on a large range
+      await new Promise((resolve) => setTimeout(resolve, 500));
+    } catch (error) {
+      console.error('Failed to update sessions:', error);
+    }
+
+    currentDate = nextHour;
+  }
+
+  console.log('Done!');
+}
+
+main().catch((error) => {
+  console.error('Script failed:', error);
+  process.exit(1);
+});