Skip to content

Commit

Permalink
chore: attempt 3
Browse files Browse the repository at this point in the history
  • Loading branch information
conradbekondo committed Jan 22, 2025
1 parent aac21a6 commit 4b00f44
Showing 1 changed file with 16 additions and 13 deletions.
29 changes: 16 additions & 13 deletions scripts/seed.ts
Original file line number Diff line number Diff line change
@@ -1,23 +1,26 @@
import 'dotenv/config';
import { drizzle } from 'drizzle-orm/neon-serverless';
import { drizzle } from 'drizzle-orm/neon-serverless';
import { PgTransaction } from 'drizzle-orm/pg-core';
import * as categories from '../db/seed/categories';
import * as users from '../db/seed/users';
import * as wallets from '../db/seed/wallets';
import * as categories from '../db/seed/categories';
import * as users from '../db/seed/users';
import * as wallets from '../db/seed/wallets';
import { DefaultWriter } from '../db/log-writer';
import { DefaultLogger } from 'drizzle-orm/logger';
import { Pool } from '@neondatabase/serverless';

type Seeder = { name: string, seed: (t: PgTransaction<any>) => Promise<void> }
const seeders: Seeder[] = [users, categories, wallets];

const logger = process.env['NODE_ENV'] === 'development' ? new DefaultLogger({ writer: new DefaultWriter() }) : false
console.log('db url = ', process.env['DATABASE_URL'])
const db = drizzle({ connection: String(process.env['DATABASE_URL']), logger })

db.transaction(async t => {
for await (const { seed, name } of seeders) {
console.log(`Seeding "${name} ⚙️`);
await seed(t as unknown as PgTransaction<any>);
console.log(`Seeded "${name}" ✅`);
}
}).catch(console.error);
const db = drizzle(new Pool({ connectionString: String(process.env['DATABASE_URL']) }), { logger });
db.$client.connect().then(() =>
db.transaction(async t => {
for await (const { seed, name } of seeders) {
console.log(`Seeding "${name} ⚙️`);
await seed(t as unknown as PgTransaction<any>);
console.log(`Seeded "${name}" ✅`);
}
}))
.then(() => db.$client.end())
.catch(console.error);

0 comments on commit 4b00f44

Please sign in to comment.