NestJS与Prisma：实现数据库操作后的钩子与副作用处理

import { Injectable, OnModuleInit, InternalServerErrorException, Logger } from '@nestjs/common';
import { PrismaClient } from '@prisma/client';

@Injectable()
export class PrismaService extends PrismaClient implements OnModuleInit {
  private readonly logger = new Logger(PrismaService.name);

  // 定义客户端扩展
  private clientExtensions = this.$extends({
    query: {
      post: {
        /**
         * 拦截 'post' 模型的 'create' 操作
         * @param {object} args - 原始查询的参数
         * @param {Function} query - 用于执行原始查询的函数
         * @returns {Promise<any>} 原始查询的结果
         */
        async create({ args, query }) {
          let result;
          try {
            // 1. 执行原始的数据库创建操作
            result = await query(args);

            // 2. 数据库操作成功后，执行自定义的副作用逻辑
            // 例如：发送通知、更新缓存、触发其他服务等
            console.log(`新文章创建成功，ID: ${result.id}。正在发送通知...`);
            // 模拟发送通知方法
            await PrismaService.sendNotificationToAdmins(result); 

          } catch (error) {
            this.logger.error(`创建文章失败或后置处理异常: ${error.message}`);
            // 可以选择重新抛出异常，或者进行其他错误处理
            throw new InternalServerErrorException("创建文章失败");
          }

          // 3. 返回原始查询的结果
          return result;
        },
        // 可以在这里添加其他操作的拦截，例如 update, delete
        async update({ args, query }) {
          const result = await query(args);
          console.log(`文章更新成功，ID: ${result.id}。执行更新后逻辑...`);
          // await PrismaService.sendUpdateNotification(result);
          return result;
        },
        async delete({ args, query }) {
          const result = await query(args);
          console.log(`文章删除成功，ID: ${args.where.id}。执行删除后逻辑...`);
          // await PrismaService.logDeletion(args.where.id);
          return result;
        }
      },
      // 可以在这里为其他模型定义扩展
      // user: { /* ... */ }
    },
    // 也可以添加其他类型的扩展，如 model, client
  });

  async onModuleInit(): Promise<void> {
    // 连接到Prisma数据库
    await this.$connect();
    this.logger.log('Prisma Client 已连接.');

    // 将扩展后的客户端实例赋值给当前服务实例，
    // 使得在其他服务中注入 PrismaService 时，使用的是带有扩展功能的客户端
    Object.assign(this, this.clientExtensions);
  }

  // 示例：模拟发送通知的方法
  private static async sendNotificationToAdmins(post: any): Promise<void> {
    // 实际应用中，这里会调用邮件服务、消息队列或第三方API
    return new Promise(resolve => {
      setTimeout(() => {
        console.log(`[通知服务] 已向管理员发送关于文章 "${post.title}" (ID: ${post.id}) 的创建通知。`);
        resolve();
      }, 500); // 模拟异步操作
    });
  }

  // 在应用程序关闭时断开Prisma连接
  async onModuleDestroy(): Promise<void> {
    await this.$disconnect();
    this.logger.log('Prisma Client 已断开连接.');
  }
}

import { Injectable } from '@nestjs/common';
import { PrismaService } from './prisma.service'; // 假设prisma.service.ts在同一目录
import { CreatePostDto } from './dto/create-post.dto'; // 假设有这个DTO

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

  async createPost(createPostDto: CreatePostDto) {
    // 调用 prisma.post.create() 将自动触发 PrismaService 中定义的扩展逻辑
    const newPost = await this.prisma.post.create({
      data: {
        uuid: createPostDto.uuid, // 假设uuid由外部生成
        author: createPostDto.author,
        categoryId: createPostDto.categoryId,
        title: createPostDto.title,
        content: createPostDto.content,
        createdAt: new Date(),
        updatedAt: new Date(),
      },
    });
    return newPost;
  }

  // 其他CRUD操作...
}