跳到主要内容

使用扩展-收缩模式迁移数据

10 分钟

简介​

在生产环境中更改数据库 schema 时，确保数据一致性并避免停机至关重要。本指南向您展示如何使用扩展-收缩模式安全地在列之间迁移数据。我们将通过一个实际示例，替换布尔字段为枚举字段，同时保留现有数据。

先决条件​

在开始本指南之前，请确保您已具备：

  • 已安装 Node.js（版本 18 或更高）
  • 一个已存在 schema 的 Prisma ORM 项目
  • 一个受支持的数据库（PostgreSQL、MySQL、SQLite、SQL Server 等）
  • 可访问开发和生产数据库
  • 对 Git 分支有基本了解
  • 对 TypeScript 有基本熟悉

1. 设置您的环境​

1.1. 审查初始 schema​

从包含 Post 模型的初始 schema 开始

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

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

model Post {
  id        Int     @id @default(autoincrement())
  title     String
  content   String?
  published Boolean @default(false)
}

1.2. 创建开发分支​

为您的更改创建一个新分支

git checkout -b create-status-field

2. 扩展 schema​

2.1. 添加新列​

更新您的 schema 以添加新的 Status 枚举和字段

model Post {
  id        Int      @id @default(autoincrement())
  title     String
  content   String?
  published Boolean? @default(false)
  status    Status   @default(Unknown)
}

enum Status {
  Unknown
  Draft
  InProgress
  InReview
  Published
}

2.2. 创建迁移​

生成迁移

npx prisma migrate dev --name add-status-column

3. 迁移数据​

3.1. 创建迁移脚本​

为数据迁移创建一个新的 TypeScript 文件

import { PrismaClient } from '@prisma/client'

const prisma = new PrismaClient()

async function main() {
  await prisma.$transaction(async (tx) => {
    const posts = await tx.post.findMany()
    for (const post of posts) {
      await tx.post.update({
        where: { id: post.id },
        data: {
          status: post.published ? 'Published' : 'Unknown',
        },
      })
    }
  })
}

main()
  .catch(async (e) => {
    console.error(e)
    process.exit(1)
  })
  .finally(async () => await prisma.$disconnect())

3.2. 设置迁移脚本​

将迁移脚本添加到您的 package.json

{
  "scripts": {
    "data-migration:add-status-column": "tsx ./prisma/migrations/<migration-timestamp>/data-migration.ts"
  }
}

3.3. 执行迁移​

  1. 更新您的 DATABASE_URL 以指向生产数据库
  2. 运行迁移脚本
npm run data-migration:add-status-column

4. 收缩 schema​

4.1. 创建清理分支​

创建一个新分支以删除旧列

git checkout -b drop-published-column

4.2. 删除旧列​

更新您的 schema 以删除 published 字段

model Post {
  id      Int      @id @default(autoincrement())
  title   String
  content String?
  status  Status   @default(Unknown)
}

enum Status {
  Draft
  InProgress
  InReview
  Published
}

4.3. 生成清理迁移​

创建并运行最终迁移

npx prisma migrate dev --name drop-published-column

5. 部署到生产环境​

5.1. 设置部署​

将以下命令添加到您的 CI/CD 管道

npx prisma migrate deploy

5.2. 监控部署​

部署后，密切关注日志中的任何错误并监控应用程序的行为。

故障排除​

常见问题和解决方案​

  1. 因缺少默认值导致迁移失败

    • 确保您已添加适当的默认值
    • 检查所有现有记录是否可以迁移

  2. 数据丢失预防

    • 在运行迁移之前，请务必备份您的数据库
    • 首先在生产数据的副本上测试迁移

  3. 事务回滚

    • 如果数据迁移失败，事务将自动回滚
    • 修复任何错误并重试迁移

下一步​

现在您已经完成了第一次扩展-收缩迁移，您可以：

获取更多信息


与 Prisma 保持联系

通过以下方式与 Prisma 保持联系，继续您的 Prisma 之旅： 我们的活跃社区。保持信息更新，参与其中，并与其他开发者协作

我们真诚地珍视您的参与，并期待您成为我们社区的一员！

© . All rights reserved.