drizzle.provider.ts

import { Injectable, OnModuleDestroy, OnModuleInit } from '@nestjs/common';
import { ConfigService } from '@nestjs/config';
import { drizzle, NodePgDatabase } from 'drizzle-orm/node-postgres';
import { Pool } from 'pg';
import * as schema from 'src/drizzle/drizzle.schema';

@Injectable()
export class Drizzle implements OnModuleInit, OnModuleDestroy {
  private pool: Pool;
  public db: NodePgDatabase<typeof schema>;
  constructor(private readonly config: ConfigService) {}
  onModuleInit() {
    const dbUrl = this.config.get<string>('DATABASE_URL');
    const isProd = this.config.get<string>('NODE_ENV');

    this.pool = new Pool({
      connectionString: dbUrl,
    });

    this.db = drizzle(this.pool, {
      schema,
      logger: !isProd,
    });
  }

  async onModuleDestroy() {
    if (this.pool) {
      await this.pool.end();
    }
  }
}