23 lines
674 B
TypeScript
23 lines
674 B
TypeScript
import { Injectable, OnModuleInit, OnModuleDestroy } from '@nestjs/common';
|
|
import { PrismaClient } from '@prisma/client';
|
|
import { PrismaPg } from '@prisma/adapter-pg';
|
|
import { Pool } from 'pg';
|
|
import 'dotenv/config';
|
|
|
|
@Injectable()
|
|
export class PrismaService extends PrismaClient implements OnModuleInit, OnModuleDestroy {
|
|
private pool: Pool;
|
|
|
|
constructor() {
|
|
const pool = new Pool({ connectionString: process.env.DATABASE_URL });
|
|
const adapter = new PrismaPg(pool);
|
|
super({ adapter });
|
|
this.pool = pool;
|
|
}
|
|
|
|
async onModuleInit() { await this.$connect(); }
|
|
async onModuleDestroy() {
|
|
await this.$disconnect();
|
|
await this.pool.end();
|
|
}
|
|
} |