中间件示例：软删除

以下示例使用 中间件 执行 软删除。软删除是指通过将 deleted 等字段更改为 true 来 标记为已删除 记录，而不是实际从数据库中删除记录。使用软删除的原因包括

• 法规要求你必须保留数据一段时间
• '垃圾箱' / '回收站' 功能，允许用户恢复已删除的内容

警告

注意：此页面演示了中间件的示例用法。我们不打算将此示例作为完全可用的软删除功能，并且它没有涵盖所有边缘情况。例如，此中间件不适用于嵌套写入，因此无法捕获在 update 查询中使用 delete 或 deleteMany 作为选项的情况。

此示例使用以下模式 - 请注意 Post 模型上的 deleted 字段

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

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

model User {
  id        Int     @id @default(autoincrement())
  name      String?
  email     String  @unique
  posts     Post[]
  followers User[]  @relation("UserToUser")
  user      User?   @relation("UserToUser", fields: [userId], references: [id])
  userId    Int?
}

model Post {
  id      Int     @id @default(autoincrement())
  title   String
  content String?
  user    User?   @relation(fields: [userId], references: [id])
  userId  Int?
  tags    Tag[]
  views   Int     @default(0)
  deleted Boolean @default(false)
}

model Category {
  id             Int        @id @default(autoincrement())
  parentCategory Category?  @relation("CategoryToCategory", fields: [categoryId], references: [id])
  category       Category[] @relation("CategoryToCategory")
  categoryId     Int?
}

model Tag {
  tagName String @id // Must be unique
  posts   Post[]
}

步骤 1：存储记录状态

向 Post 模型添加名为 deleted 的字段。你可以根据你的需求选择两种字段类型之一

• Boolean，默认值为 false

  model Post {
    id      Int     @id @default(autoincrement())
    ...
    deleted Boolean @default(false)
  }

• 创建一个可空 DateTime 字段，以便你确切地知道记录何时被标记为已删除 - NULL 表示记录未被删除。在某些情况下，存储记录何时被删除可能是一个法规要求。

  model Post {
    id      Int       @id @default(autoincrement())
    ...
    deleted DateTime?
  }

注意：使用两个独立的字段 (isDeleted 和 deletedDate) 可能会导致这两个字段不同步 - 例如，记录可能被标记为已删除，但没有关联的日期）。

为了简单起见，此示例使用 Boolean 字段类型。

步骤 2：软删除中间件

添加一个执行以下任务的中间件

• 拦截 Post 模型的 delete() 和 deleteMany() 查询
• 将 params.action 分别更改为 update 和 updateMany
• 引入一个 data 参数并设置 { deleted: true }，如果存在其他过滤器参数则保留这些参数

运行以下示例以测试软删除中间件

import { PrismaClient } from '@prisma/client'

const prisma = new PrismaClient({})

async function main() {
  /***********************************/
  /* SOFT DELETE MIDDLEWARE */
  /***********************************/

  prisma.$use(async (params, next) => {
    // Check incoming query type
    if (params.model == 'Post') {
      if (params.action == 'delete') {
        // Delete queries
        // Change action to an update
        params.action = 'update'
        params.args['data'] = { deleted: true }
      }
      if (params.action == 'deleteMany') {
        // Delete many queries
        params.action = 'updateMany'
        if (params.args.data != undefined) {
          params.args.data['deleted'] = true
        } else {
          params.args['data'] = { deleted: true }
        }
      }
    }
    return next(params)
  })

  /***********************************/
  /* TEST */
  /***********************************/

  const titles = [
    { title: 'How to create soft delete middleware' },
    { title: 'How to install Prisma' },
    { title: 'How to update a record' },
  ]

  console.log('\u001b[1;34mSTARTING SOFT DELETE TEST \u001b[0m')
  console.log('\u001b[1;34m#################################### \u001b[0m')

  let i = 0
  let posts = new Array()

  // Create 3 new posts with a randomly assigned title each time
  for (i == 0; i < 3; i++) {
    const createPostOperation = prisma.post.create({
      data: titles[Math.floor(Math.random() * titles.length)],
    })
    posts.push(createPostOperation)
  }

  var postsCreated = await prisma.$transaction(posts)

  console.log(
    'Posts created with IDs: ' +
      '\u001b[1;32m' +
      postsCreated.map((x) => x.id) +
      '\u001b[0m'
  )

  // Delete the first post from the array
  const deletePost = await prisma.post.delete({
    where: {
      id: postsCreated[0].id, // Random ID
    },
  })

  // Delete the 2nd two posts
  const deleteManyPosts = await prisma.post.deleteMany({
    where: {
      id: {
        in: [postsCreated[1].id, postsCreated[2].id],
      },
    },
  })

  const getPosts = await prisma.post.findMany({
    where: {
      id: {
        in: postsCreated.map((x) => x.id),
      },
    },
  })

  console.log()

  console.log(
    'Deleted post with ID: ' + '\u001b[1;32m' + deletePost.id + '\u001b[0m'
  )
  console.log(
    'Deleted posts with IDs: ' +
      '\u001b[1;32m' +
      [postsCreated[1].id + ',' + postsCreated[2].id] +
      '\u001b[0m'
  )
  console.log()
  console.log(
    'Are the posts still available?: ' +
      (getPosts.length == 3
        ? '\u001b[1;32m' + 'Yes!' + '\u001b[0m'
        : '\u001b[1;31m' + 'No!' + '\u001b[0m')
  )
  console.log()
  console.log('\u001b[1;34m#################################### \u001b[0m')
  // 4. Count ALL posts
  const f = await prisma.post.findMany({})
  console.log('Number of posts: ' + '\u001b[1;32m' + f.length + '\u001b[0m')

  // 5. Count DELETED posts
  const r = await prisma.post.findMany({
    where: {
      deleted: true,
    },
  })
  console.log(
    'Number of SOFT deleted posts: ' + '\u001b[1;32m' + r.length + '\u001b[0m'
  )
}

main()

该示例输出以下内容

STARTING SOFT DELETE TEST
####################################
Posts created with IDs: 587,588,589

Deleted post with ID: 587
Deleted posts with IDs: 588,589

Are the posts still available?: Yes!

####################################

提示

注释掉中间件以查看消息更改。

✔ 此软删除方法的优点包括

• 软删除发生在数据访问级别，这意味着你无法删除记录，除非你使用原始 SQL

✘ 此软删除方法的缺点包括

• 除非你显式地按 where: { deleted: false } 过滤，否则仍然可以读取和更新内容 - 在一个拥有大量查询的大型项目中，存在软删除的内容仍然会被显示的风险
• 你仍然可以使用原始 SQL 删除记录

提示

你可以在数据库级别创建规则或触发器 (MySQL 和 PostgreSQL) 来防止记录被删除。

步骤 3：选择性地阻止读取/更新软删除的记录

在步骤 2 中，我们实现了中间件，阻止了 Post 记录被删除。但是，你仍然可以读取和更新已删除的记录。此步骤探讨了两种防止读取和更新已删除记录的方法。

注意：这些选项只是一些想法，有其优缺点，你可以选择完全不同的方法。

选项 1：在自己的应用程序代码中实现过滤器

在此选项中

• Prisma 客户端中间件负责阻止记录被删除
• 你自己的应用程序代码 (可能是 GraphQL API、REST API 或模块) 负责在需要时过滤掉已删除的帖子 ({ where: { deleted: false } }) - 例如，getPost GraphQL 解析器永远不会返回已删除的帖子

✔ 此软删除方法的优点包括

• Prisma 客户端的创建/更新查询没有更改 - 如果你需要，你可以轻松地请求已删除的记录
• 在中间件中修改查询可能会产生一些意想不到的后果，例如更改查询返回类型 (参见选项 2)

✘ 此软删除方法的缺点包括

• 与软删除相关的逻辑维护在两个不同的地方
• 如果你的 API 表面非常大，并且由多个贡献者维护，则可能难以强制执行某些业务规则 (例如，永远不允许更新已删除的记录)

选项 2：使用中间件来确定对软删除记录的读取/更新查询的行为

选项 2 使用 Prisma 客户端中间件来阻止软删除的记录被返回。下表描述了中间件如何影响每个查询

查询	中间件逻辑	返回类型更改
findUnique()	🔧 将查询更改为 findFirst (因为你无法将 deleted: false 过滤器应用于 findUnique()) 🔧 添加 where: { deleted: false } 过滤器以排除软删除的帖子 🔧 从 5.0.0 版本开始，你可以使用 findUnique() 应用 delete: false 过滤器，因为 非唯一字段已公开。	无更改
findMany	🔧 添加 where: { deleted: false } 过滤器以默认情况下排除软删除的帖子 🔧 允许开发人员通过指定 deleted: true 显式地请求 软删除的帖子	无更改
update	🔧 将查询更改为 updateMany (因为你无法将 deleted: false 过滤器应用于 update) 🔧 添加 where: { deleted: false } 过滤器以排除软删除的帖子	{ count: n } 而不是 Post
updateMany	🔧 添加 where: { deleted: false } 过滤器以排除软删除的帖子	无更改

• 是否无法使用 findFirstOrThrow() 或 findUniqueOrThrow() 利用软删除？
  从 5.1.0 版本开始，你可以使用中间件将软删除应用于 findFirstOrThrow() 或 findUniqueOrThrow()。
• 为什么你可以使用 { where: { deleted: true } } 过滤器使用 findMany()，而不能使用 updateMany()？
  此特定示例是为了支持用户可以恢复他们已删除的博客帖子 (这需要软删除帖子的列表) 的场景编写的 - 但用户不应该能够编辑已删除的帖子。
• 我仍然可以将已删除的帖子 connect 或 connectOrCreate 吗？
  在此示例中 - 可以。中间件不会阻止你将现有的软删除帖子连接到用户。

运行以下示例以查看中间件如何影响每个查询

import { PrismaClient, Prisma } from '@prisma/client'

const prisma = new PrismaClient({})

async function main() {
  /***********************************/
  /* SOFT DELETE MIDDLEWARE */
  /***********************************/

  prisma.$use(async (params, next) => {
    if (params.model == 'Post') {
      if (params.action === 'findUnique' || params.action === 'findFirst') {
        // Change to findFirst - you cannot filter
        // by anything except ID / unique with findUnique()
        params.action = 'findFirst'
        // Add 'deleted' filter
        // ID filter maintained
        params.args.where['deleted'] = false
      }
      if (
        params.action === 'findFirstOrThrow' ||
        params.action === 'findUniqueOrThrow'
      ) {
        if (params.args.where) {
          if (params.args.where.deleted == undefined) {
            // Exclude deleted records if they have not been explicitly requested
            params.args.where['deleted'] = false
          }
        } else {
          params.args['where'] = { deleted: false }
        }
      }
      if (params.action === 'findMany') {
        // Find many queries
        if (params.args.where) {
          if (params.args.where.deleted == undefined) {
            params.args.where['deleted'] = false
          }
        } else {
          params.args['where'] = { deleted: false }
        }
      }
    }
    return next(params)
  })

  prisma.$use(async (params, next) => {
    if (params.model == 'Post') {
      if (params.action == 'update') {
        // Change to updateMany - you cannot filter
        // by anything except ID / unique with findUnique()
        params.action = 'updateMany'
        // Add 'deleted' filter
        // ID filter maintained
        params.args.where['deleted'] = false
      }
      if (params.action == 'updateMany') {
        if (params.args.where != undefined) {
          params.args.where['deleted'] = false
        } else {
          params.args['where'] = { deleted: false }
        }
      }
    }
    return next(params)
  })

  prisma.$use(async (params, next) => {
    // Check incoming query type
    if (params.model == 'Post') {
      if (params.action == 'delete') {
        // Delete queries
        // Change action to an update
        params.action = 'update'
        params.args['data'] = { deleted: true }
      }
      if (params.action == 'deleteMany') {
        // Delete many queries
        params.action = 'updateMany'
        if (params.args.data != undefined) {
          params.args.data['deleted'] = true
        } else {
          params.args['data'] = { deleted: true }
        }
      }
    }
    return next(params)
  })

  /***********************************/
  /* TEST */
  /***********************************/

  const titles = [
    { title: 'How to create soft delete middleware' },
    { title: 'How to install Prisma' },
    { title: 'How to update a record' },
  ]

  console.log('\u001b[1;34mSTARTING SOFT DELETE TEST \u001b[0m')
  console.log('\u001b[1;34m#################################### \u001b[0m')

  let i = 0
  let posts = new Array()

  // Create 3 new posts with a randomly assigned title each time
  for (i == 0; i < 3; i++) {
    const createPostOperation = prisma.post.create({
      data: titles[Math.floor(Math.random() * titles.length)],
    })
    posts.push(createPostOperation)
  }

  var postsCreated = await prisma.$transaction(posts)

  console.log(
    'Posts created with IDs: ' +
      '\u001b[1;32m' +
      postsCreated.map((x) => x.id) +
      '\u001b[0m'
  )

  // Delete the first post from the array
  const deletePost = await prisma.post.delete({
    where: {
      id: postsCreated[0].id, // Random ID
    },
  })

  // Delete the 2nd two posts
  const deleteManyPosts = await prisma.post.deleteMany({
    where: {
      id: {
        in: [postsCreated[1].id, postsCreated[2].id],
      },
    },
  })

  const getOnePost = await prisma.post.findUnique({
    where: {
      id: postsCreated[0].id,
    },
  })

  const getOneUniquePostOrThrow = async () =>
    await prisma.post.findUniqueOrThrow({
      where: {
        id: postsCreated[0].id,
      },
    })

  const getOneFirstPostOrThrow = async () =>
    await prisma.post.findFirstOrThrow({
      where: {
        id: postsCreated[0].id,
      },
    })

  const getPosts = await prisma.post.findMany({
    where: {
      id: {
        in: postsCreated.map((x) => x.id),
      },
    },
  })

  const getPostsAnDeletedPosts = await prisma.post.findMany({
    where: {
      id: {
        in: postsCreated.map((x) => x.id),
      },
      deleted: true,
    },
  })

  const updatePost = await prisma.post.update({
    where: {
      id: postsCreated[1].id,
    },
    data: {
      title: 'This is an updated title (update)',
    },
  })

  const updateManyDeletedPosts = await prisma.post.updateMany({
    where: {
      deleted: true,
      id: {
        in: postsCreated.map((x) => x.id),
      },
    },
    data: {
      title: 'This is an updated title (updateMany)',
    },
  })

  console.log()

  console.log(
    'Deleted post (delete) with ID: ' +
      '\u001b[1;32m' +
      deletePost.id +
      '\u001b[0m'
  )
  console.log(
    'Deleted posts (deleteMany) with IDs: ' +
      '\u001b[1;32m' +
      [postsCreated[1].id + ',' + postsCreated[2].id] +
      '\u001b[0m'
  )
  console.log()
  console.log(
    'findUnique: ' +
      (getOnePost?.id != undefined
        ? '\u001b[1;32m' + 'Posts returned!' + '\u001b[0m'
        : '\u001b[1;31m' +
          'Post not returned!' +
          '(Value is: ' +
          JSON.stringify(getOnePost) +
          ')' +
          '\u001b[0m')
  )
  try {
    console.log('findUniqueOrThrow: ')
    await getOneUniquePostOrThrow()
  } catch (error) {
    if (
      error instanceof Prisma.PrismaClientKnownRequestError &&
      error.code == 'P2025'
    )
      console.log(
        '\u001b[1;31m' +
          'PrismaClientKnownRequestError is catched' +
          '(Error name: ' +
          error.name +
          ')' +
          '\u001b[0m'
      )
  }
  try {
    console.log('findFirstOrThrow: ')
    await getOneFirstPostOrThrow()
  } catch (error) {
    if (
      error instanceof Prisma.PrismaClientKnownRequestError &&
      error.code == 'P2025'
    )
      console.log(
        '\u001b[1;31m' +
          'PrismaClientKnownRequestError is catched' +
          '(Error name: ' +
          error.name +
          ')' +
          '\u001b[0m'
      )
  }
  console.log()
  console.log(
    'findMany: ' +
      (getPosts.length == 3
        ? '\u001b[1;32m' + 'Posts returned!' + '\u001b[0m'
        : '\u001b[1;31m' + 'Posts not returned!' + '\u001b[0m')
  )
  console.log(
    'findMany ( delete: true ): ' +
      (getPostsAnDeletedPosts.length == 3
        ? '\u001b[1;32m' + 'Posts returned!' + '\u001b[0m'
        : '\u001b[1;31m' + 'Posts not returned!' + '\u001b[0m')
  )
  console.log()
  console.log(
    'update: ' +
      (updatePost.id != undefined
        ? '\u001b[1;32m' + 'Post updated!' + '\u001b[0m'
        : '\u001b[1;31m' +
          'Post not updated!' +
          '(Value is: ' +
          JSON.stringify(updatePost) +
          ')' +
          '\u001b[0m')
  )
  console.log(
    'updateMany ( delete: true ): ' +
      (updateManyDeletedPosts.count == 3
        ? '\u001b[1;32m' + 'Posts updated!' + '\u001b[0m'
        : '\u001b[1;31m' + 'Posts not updated!' + '\u001b[0m')
  )
  console.log()
  console.log('\u001b[1;34m#################################### \u001b[0m')
  // 4. Count ALL posts
  const f = await prisma.post.findMany({})
  console.log(
    'Number of active posts: ' + '\u001b[1;32m' + f.length + '\u001b[0m'
  )

  // 5. Count DELETED posts
  const r = await prisma.post.findMany({
    where: {
      deleted: true,
    },
  })
  console.log(
    'Number of SOFT deleted posts: ' + '\u001b[1;32m' + r.length + '\u001b[0m'
  )
}

main()

该示例输出以下内容

STARTING SOFT DELETE TEST
####################################
Posts created with IDs: 680,681,682

Deleted post (delete) with ID: 680
Deleted posts (deleteMany) with IDs: 681,682

findUnique: Post not returned!(Value is: [])
findMany: Posts not returned!
findMany ( delete: true ): Posts returned!

update: Post not updated!(Value is: {"count":0})
updateMany ( delete: true ): Posts not updated!

####################################
Number of active posts: 0
Number of SOFT deleted posts: 95

✔ 此方法的优点

• 开发人员可以有意识地选择在 findMany 中包含已删除的记录
• 你不会意外地读取或更新已删除的记录

✖ 此方法的缺点

• 从 API 中无法明显看出您没有获取所有记录，以及 { where: { deleted: false } } 是默认查询的一部分。
• 返回值类型 update 受到影响，因为中间件将查询更改为 updateMany。
• 不处理带有 AND、OR、every 等的复杂查询。
• 在使用来自另一个模型的 include 时，不处理过滤。

常见问题解答

我可以将全局 includeDeleted 添加到 Post 模型吗？

您可能很想通过在 Post 模型中添加 includeDeleted 属性来“破解”您的 API，并使以下查询成为可能

prisma.post.findMany({ where: { includeDeleted: true } })

注意：您仍然需要编写中间件。

我们✘ 不建议这种方法，因为它会用不代表真实数据的字段污染模式。