16 lines
390 B
TypeScript
16 lines
390 B
TypeScript
import { INestApplication, Injectable, OnModuleInit } from '@nestjs/common';
|
|
import { PrismaClient } from '@prisma/client';
|
|
|
|
@Injectable()
|
|
export class PrismaService extends PrismaClient implements OnModuleInit {
|
|
async onModuleInit() {
|
|
await this.$connect();
|
|
}
|
|
|
|
async enableShutdownHooks(app: INestApplication) {
|
|
this.$on('beforeExit', async () => {
|
|
await app.close();
|
|
});
|
|
}
|
|
}
|