利用Prisma扩展在NestJS中实现数据库操作后置钩子

// src/prisma/prisma.service.ts
import { INestApplication, Injectable, OnModuleInit } from '@nestjs/common';
import { PrismaClient } from '@prisma/client';

@Injectable()
export class PrismaService extends PrismaClient implements OnModuleInit {
  constructor() {
    super(); // 调用PrismaClient的构造函数
  }

  async onModuleInit(): Promise<void> {
    await this.$connect(); // 连接数据库
    // 将客户端扩展应用到当前PrismaService实例
    Object.assign(this, this.clientExtensions);
  }

  // 可选：在应用关闭时断开数据库连接
  async enableShutdownHooks(app: INestApplication) {
    this.$on('beforeExit', async () => {
      await app.close();
    });
  }

  /**
   * 定义Prisma客户端扩展
   * 这里我们将定义在post模型上创建操作后的钩子
   */
  clientExtensions = this.$extends({
    query: {
      post: {
        async create({ args, query }) {
          // 1. 执行原始的数据库创建操作
          const result = await query(args);

          // 2. 数据库操作成功后，执行自定义的后置逻辑
          // 确保只有在查询成功后才执行此逻辑
          console.log(`新帖子已创建，标题: ${result.title}。正在发送通知...`);
          // 示例：调用一个发送通知的方法
          // await this.sendNotificationToAdmins(result);

          return result; // 返回原始查询结果
        },
        // 你也可以为update、delete等其他操作添加类似的扩展
        // async update({ args, query }) { /* ... */ },
        // async delete({ args, query }) { /* ... */ },
      },
      // 你也可以为其他模型添加扩展
      // user: { /* ... */ },
    },
  });

  // 示例：一个发送通知的方法
  // async sendNotificationToAdmins(post: any) {
  //   // 实际的通知发送逻辑，例如调用第三方服务、发送邮件、推送消息等
  //   console.log(`管理员已收到新帖子 "${post.title}" 的创建通知。`);
  //   // 模拟异步操作
  //   await new Promise(resolve => setTimeout(resolve, 100));
  // }
}

// src/app.module.ts
import { Module } from '@nestjs/common';
import { AppController } from './app.controller';
import { AppService } from './app.service';
import { PrismaService } from './prisma/prisma.service'; // 导入PrismaService

@Module({
  imports: [],
  controllers: [AppController],
  providers: [AppService, PrismaService], // 提供PrismaService
})
export class AppModule {}

// src/post/post.service.ts
import { Injectable, InternalServerErrorException, Logger } from '@nestjs/common';
import { PrismaService } from '../prisma/prisma.service';
import { CreatePostDto } from './dto/create-post.dto';
import { v4 as uuidv4 } from 'uuid';

@Injectable()
export class PostService {
  private readonly logger = new Logger(PostService.name);

  constructor(private readonly prisma: PrismaService) {}

  async createPost(createPostDto: CreatePostDto) {
    let post;
    try {
      // 这里的create操作将触发PrismaService中定义的扩展
      post = await this.prisma.post.create({
        data: {
          uuid: uuidv4(),
          author: createPostDto.author,
          // categoryId: postCategory.id, // 假设postCategory已获取
          title: createPostDto.title,
          content: createPostDto.content,
          createdAt: new Date(),
          updatedAt: new Date(),
        },
      });
      return post;
    } catch (err) {
      this.logger.error(err);
      throw new InternalServerErrorException('Failed to create the post');
    }
  }
}