利用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(); // 连接到数据库
    // 将客户端扩展应用到Prisma客户端实例
    Object.assign(this, this.clientExtensions);
  }

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

  /**
   * 定义Prisma客户端扩展
   */
  clientExtensions = this.$extends({
    query: {
      // 针对post模型的所有查询操作进行扩展
      post: {
        // 拦截create操作
        async create({ args, query }) {
          // 1. 执行原始的create查询
          // query(args) 是一个函数，它会执行Prisma客户端的原始查询
          const result = await query(args);

          // 2. 在原始查询成功执行后，插入自定义的后置逻辑
          // 确保只有在数据成功创建后，才执行此处的逻辑
          console.log("Post created successfully. Sending notification...");
          // 假设这里调用一个发送通知的方法，例如：
          // await this.sendNotificationToAdmins(result); 

          // 3. 返回原始查询的结果
          return result;
        },
        // 可以在这里添加对update、delete等其他操作的拦截
        // async update({ args, query }) { ... },
        // async delete({ args, query }) { ... },
      },
      // 可以在这里添加对其他模型的扩展
      // user: { ... }
    },
    // 也可以定义model级别的扩展，例如添加计算字段
    // model: {
    //   post: {
    //     fullName: {
    //       needs: { title: true },
    //       compute(post) {
    //         return `Title: ${post.title}`;
    //       },
    //     },
    //   },
    // },
  });

  // 示例：一个私有的通知方法
  private async sendNotificationToAdmins(post: any): Promise<void> {
    // 实际的通知逻辑，例如通过邮件、短信或消息队列发送通知
    console.log(`Notification sent for new post: "${post.title}" (ID: ${post.uuid})`);
    // await this.notificationService.sendEmail(...);
  }
}

// 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 {
      // 假设 postCategory 已经通过其他方式获取
      const postCategory = { id: 1 }; // 示例数据

      post = await this.prisma.post.create({
        data: {
          uuid: uuidv4(),
          author: createPostDto.author,
          categoryId: postCategory.id,
          title: createPostDto.title,
          content: createPostDto.content,
          createdAt: new Date(),
          updatedAt: new Date(),
        },
      });
      // 注意：这里不需要手动调用sendNotification()，因为它已经在PrismaService的扩展中被触发
      return post;
    } catch (err) {
      this.logger.error(err);
      throw new InternalServerErrorException("Failed to create the post");
    }
  }
}