postgres.ts

Filename	Percent	Threshold	Total	Covered	Uncovered
src/db/postgres.ts	100.00%	80%	71	71	0

import { eventSchema, migrationsSchema } from 'constants/schema.js';

import { drizzle } from 'drizzle-orm/postgres-js';
import { fileURLToPath } from 'url';
import { migrate } from 'drizzle-orm/postgres-js/migrator';
import path from 'path';
import postgres from 'postgres';
import { schema } from './models.js';

const __dirname = path.dirname(fileURLToPath(import.meta.url));

export class Postgres {
  sql: postgres.Sql<Record<string, unknown>>;
  constructor(
    private connectionUri: string,

    private options: postgres.Options<
      Record<string, postgres.PostgresType<any>> // eslint-disable-line @typescript-eslint/no-explicit-any
    > = {}
  ) {}

  connect() {
    const defaultOptions = {
      onnotice: () => {},
    };
    this.sql = postgres(this.connectionUri, {
      ...defaultOptions,
      ...this.options,
    });
    return drizzle(this.sql, { schema });
  }

  async disconnect() {
    if (this.sql) await this.sql.end();
  }

  /**
   * Dangerously drop and re-create all tables in your PostgreSQL database
   */
  async dropTables() {
    // Per drizzle docs, enforce max connections when running migrations
    const sql = postgres(this.connectionUri, {
      ...this.options,
      max: 1,
      onnotice: () => {},
    });

    // Drop tables schema
    await sql`DROP SCHEMA IF EXISTS ${sql(eventSchema)} CASCADE`;
    // Drop migrations schema
    await sql`DROP SCHEMA IF EXISTS ${sql(migrationsSchema)} CASCADE`;

    const db = drizzle(sql);

    await migrate(db, {
      migrationsFolder: path.join(__dirname, 'migrations'),
    });
    await sql.end();
  }
}

[]

TypeScript Coverage Report generated by type-coverage and typescript-coverage-report at Thu, 04 Apr 2024 19:50:44 GMT

TypeScript coverage report