在NestJS应用中灵活配置Prisma数据库连接
本文将演示如何利用@nestjs/config包优雅地配置NestJS应用中的Prisma数据库连接，提升配置的灵活性和可维护性。
首先，确保已安装@nestjs/config和Prisma依赖包。 以下步骤将指导您完成配置：
第一步：创建数据库配置文件。在项目根目录下创建config文件夹，并在其中新建database.config.ts文件。使用@nestjs/config的registerAs装饰器定义数据库配置：
import { registerAs } from '@nestjs/config';
import { parse } from 'url';
export default registerAs('database', () => ({
host: process.env.DB_HOST,
port: parseInt(process.env.DB_PORT, 10),
username: process.env.DB_USERNAME,
password: process.env.DB_PASSWORD,
database: process.env.DB_DATABASE,
// ... other Prisma connection options
}));
此配置从环境变量读取数据库连接信息。 运行应用前，请确保已设置DB_HOST、DB_PORT、DB_USERNAME、DB_PASSWORD和DB_DATABASE等环境变量。
第二步：在根模块（通常为app.module.ts）中导入并配置ConfigModule。使用load选项加载第一步创建的database.config.ts文件，并注入ConfigService和PrismaService：
import { Module } from '@nestjs/common';
import { ConfigModule, ConfigService } from '@nestjs/config';
import databaseConfig from './config/database.config';
import { PrismaService } from './prisma.service';
@Module({
imports: [
ConfigModule.forRoot({
load: [databaseConfig],
}),
],
providers: [ConfigService, PrismaService],
})
export class AppModule {}
第三步：创建prisma.service.ts服务文件。在此文件中，使用ConfigService获取数据库配置，并实例化PrismaClient创建Prisma客户端。 为了确保资源正确释放，实现onModuleDestroy方法关闭数据库连接：
import { Injectable, OnModuleDestroy } from '@nestjs/common';
import { PrismaClient } from '@prisma/client';
import { ConfigService } from '@nestjs/config';
@Injectable()
export class PrismaService extends PrismaClient implements OnModuleDestroy {
constructor(private configService: ConfigService) {
super(configService.get('database'));
}
async onModuleDestroy() {
await this.close();
}
}
最后，在其他模块或服务中注入PrismaService即可方便地使用Prisma客户端进行数据库操作。这种方式使数据库配置更清晰，易于维护和扩展。 记得在运行应用前配置好环境变量或.env文件。
