stan/esports
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix sql

Browse Source
This commit is contained in:
Stan Runge
2026-03-13 16:22:47 +01:00
parent 4e1834c2a5
commit 703c602905
1 changed files with 15 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
scripts/cleanup-seed-from-prod.ts
View File

@@ -7,34 +7,27 @@
* DATABASE_URL="postgresql://..." bun scripts/cleanup-seed-from-prod.ts --dry-run * DATABASE_URL="postgresql://..." bun scripts/cleanup-seed-from-prod.ts --dry-run
*/ */
import { drizzle } from 'drizzle-orm/bun-sql';
import { sql } from 'drizzle-orm';
// ── DB setup ────────────────────────────────────────────────────────────
const DATABASE_URL = process.env.DATABASE_URL; const DATABASE_URL = process.env.DATABASE_URL;
if (!DATABASE_URL) throw new Error('DATABASE_URL not set'); if (!DATABASE_URL) throw new Error('DATABASE_URL not set');
const db = drizzle(DATABASE_URL);
import { SQL as BunSQL } from 'bun:sql';
const client = new BunSQL(DATABASE_URL);
const DRY_RUN = process.argv.includes('--dry-run'); const DRY_RUN = process.argv.includes('--dry-run');
async function query(sql: string, params: any[] = []) { async function count(query: string): Promise<number> {
if (DRY_RUN) { const rows = await db.execute(sql.raw(query));
// For dry run, just count affected rows with a SELECT return Number((rows as any)[0]?.count ?? 0);
return [];
}
return client.unsafe(sql, params);
} }
async function count(sql: string, params: any[] = []): Promise<number> { async function exec(description: string, deleteSql: string) {
const rows = await client.unsafe(sql, params);
return Number(rows[0]?.count ?? 0);
}
async function exec(description: string, sql: string, params: any[] = []) {
// Count first // Count first
const countSql = sql const countSql = deleteSql
.replace(/^DELETE FROM/, 'SELECT COUNT(*) as count FROM') .replace(/^DELETE FROM/, 'SELECT COUNT(*) as count FROM');
.replace(/\s+RETURNING\b.*$/i, ''); const n = await count(countSql);
const n = await count(countSql, params);
if (n === 0) { if (n === 0) {
console.log(` ⏭ ${description}: 0 rows (skipped)`); console.log(` ⏭ ${description}: 0 rows (skipped)`);
@@ -46,7 +39,7 @@ async function exec(description: string, sql: string, params: any[] = []) {
return n; return n;
} }
await client.unsafe(sql, params); await db.execute(sql.raw(deleteSql));
console.log(` ✅ ${description}: ${n} rows deleted`); console.log(` ✅ ${description}: ${n} rows deleted`);
return n; return n;
} }
@@ -227,7 +220,7 @@ async function main() {
console.log('\n🔍 Dry run complete. Re-run without --dry-run to delete.'); console.log('\n🔍 Dry run complete. Re-run without --dry-run to delete.');
} }
await client.close(); process.exit(0);
} }
main().catch((err) => { main().catch((err) => {