Nodejs Nestjs ORM框架@nestjs/prisma的使用

Nodejs Nestjs ORM框架@nestjs/prisma的使用
[@nestjs](/user/nestjs)/prisma 是一个用于 NestJS 的 Prisma ORM 集成库。Prisma 是一个现代化的 TypeScript/JavaScript ORM，支持多种数据库（如 PostgreSQL、MySQL 和 SQLite）。下面是如何在 NestJS 项目中使用 [@nestjs](/user/nestjs)/prisma 的步骤。

安装依赖

首先，你需要安装必要的依赖：

npm install [@nestjs](/user/nestjs)/prisma prisma @prisma/client

初始化 Prisma

初始化 Prisma 配置文件：

npx prisma init

这将在项目中创建 prisma 文件夹，并生成 schema.prisma 文件，该文件用于定义你的数据模型。

创建 Prisma 模型

编辑 prisma/schema.prisma 文件来定义你的数据模型。例如：

datasource db {
  provider = "postgresql"
  url      = env("DATABASE_URL")
}

generator client {
  provider = "prisma-client-js"
}

model User {
  id    Int      @id @default(autoincrement())
  email String   @unique
  name  String?
}

然后运行：

npx prisma migrate dev --name init
npx prisma generate

这将根据你的数据模型生成迁移脚本，并生成 Prisma Client。

配置 NestJS 使用 Prisma

安装 [@nestjs](/user/nestjs)/prisma 包：

npm install [@nestjs](/user/nestjs)/prisma

在你的 NestJS 应用程序中，你可以通过模块来配置 Prisma。例如，在 app.module.ts 中：

import { Module } from '[@nestjs](/user/nestjs)/common';
import { PrismaModule } from '[@nestjs](/user/nestjs)/prisma';
import { PrismaService } from './prisma.service';

@Module({
  imports: [
    PrismaModule.forRootAsync({
      useFactory: () => ({
        isGlobal: true,
        prismaOptions: {
          datasources: {
            db: {
              url: process.env.DATABASE_URL,
            },
          },
        },
      }),
    }),
  ],
  providers: [PrismaService],
})
export class AppModule {}

创建服务和控制器

接下来，创建一个服务来处理业务逻辑：

import { Injectable, HttpException, HttpStatus } from '[@nestjs](/user/nestjs)/common';
import { PrismaService } from './prisma.service';

@Injectable()
export class UserService {
  constructor(private readonly prisma: PrismaService) {}

  async getUsers() {
    return this.prisma.user.findMany();
  }

  async getUser(id: number) {
    const user = await this.prisma.user.findUnique({ where: { id } });
    if (!user) {
      throw new HttpException('User not found', HttpStatus.NOT_FOUND);
    }
    return user;
  }

  async createUser(data: { email: string; name?: string }) {
    return this.prisma.user.create({ data });
  }

  async updateUser(id: number, data: { email: string; name?: string }) {
    const user = await this.prisma.user.update({
      where: { id },
      data,
    });
    if (!user) {
      throw new HttpException('User not found', HttpStatus.NOT_FOUND);
    }
    return user;
  }

  async deleteUser(id: number) {
    const user = await this.prisma.user.delete({
      where: { id },
    });
    if (!user) {
      throw new HttpException('User not found', HttpStatus.NOT_FOUND);
    }
    return user;
  }
}

然后创建一个控制器来暴露 API：

import { Controller, Get, Post, Body, Param, Delete, Put } from '[@nestjs](/user/nestjs)/common';
import { UserService } from './user.service';

@Controller('users')
export class UserController {
  constructor(private readonly userService: UserService) {}

  @Get()
  findAll() {
    return this.userService.getUsers();
  }

  @Get(':id')
  findOne(@Param('id') id: string) {
    return this.userService.getUser(Number(id));
  }

  @Post()
  create(@Body() createUserDto: { email: string; name?: string }) {
    return this.userService.createUser(createUserDto);
  }

  @Put(':id')
  update(@Param('id') id: string, @Body() updateUserDto: { email: string; name?: string }) {
    return this.userService.updateUser(Number(id), updateUserDto);
  }

  @Delete(':id')
  remove(@Param('id') id: string) {
    return this.userService.deleteUser(Number(id));
  }
}

运行应用

现在你可以运行你的 NestJS 应用程序并测试这些 API：

npm run start

这样，你就完成了在 NestJS 中使用 [@nestjs](/user/nestjs)/prisma 的基本设置。