如何在NestJS应用中优雅地配置Prisma数据库连接？

在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文件。