Drift数据库迁移：复杂列转换实战指南

概述

在使用Drift数据库时，开发者经常需要进行表结构变更和复杂的数据迁移。本文将通过一个实际案例，详细介绍如何在Drift中执行包含多列转换的复杂迁移操作。

场景分析

假设我们有一个现有表格，包含以下列：

• cA：字符串类型（非空）
• cB：可空字符串类型

现在需要执行以下迁移操作：

1. 新增cC列（字符串类型），其值为：
   • 当cB不为空时，取cB的值
   • 当cB为空时，取cA的值
2. 新增cD列（枚举文本类型），其值基于cA和cB的逻辑计算

解决方案

基础迁移方法

Drift提供了alterTable和TableMigration来实现表结构变更。对于简单的列添加，可以直接使用：

await m.alterTable(
  TableMigration(yourTable)
);

复杂列转换

对于需要基于现有列计算新列值的情况，可以使用columnTransformer参数配合SQL表达式：

await m.alterTable(
  TableMigration(
    yourTable,
    columnTransformer: {
      yourTable.cC: coalesce([yourTable.cB, yourTable.cA])
    },
  )
);

这里使用了SQL的COALESCE函数，它会返回参数列表中第一个非NULL的值。这种方法的优势是：

1. 完全在数据库层面执行，效率高
2. 支持所有SQL支持的表达式和函数
3. 适用于批量数据处理

枚举列处理

对于需要基于复杂逻辑计算的枚举列，可以采用以下策略：

1. 先添加允许NULL的列
2. 执行数据迁移填充初始值
3. 修改列为非空（如果需要）

// 第一步：添加允许NULL的列
await m.alterTable(
  TableMigration(yourTable)
);

// 第二步：批量更新数据
await customStatement(
  'UPDATE your_table SET cD = CASE WHEN cA = ? THEN ? ELSE ? END',
  ['特定值', '枚举值1', '枚举值2']
);

// 第三步：修改列为非空（可选）
await m.alterTable(
  TableMigration(
    yourTable,
    columnTransformer: {
      yourTable.cD: const Constant('默认值')
    },
  )
);

测试验证

为确保迁移正确性，建议编写测试用例：

1. 创建旧版本数据库实例
2. 插入测试数据：
   • cB为NULL的行
   • cB不为NULL的行
3. 执行迁移
4. 验证结果是否符合预期

最佳实践

1. 分步迁移：复杂迁移可以拆分为多个简单步骤
2. 事务处理：确保迁移操作的原子性
3. 性能考虑：大数据表迁移可能需要分批处理
4. 回滚方案：始终准备回滚计划
5. 文档记录：详细记录每次迁移的内容和原因

总结

Drift提供了强大的迁移功能，可以处理各种复杂的数据转换场景。通过合理使用SQL表达式和分步迁移策略，开发者能够安全高效地完成数据库结构变更和数据迁移工作。对于特别复杂的逻辑，可以考虑结合原生SQL和Dart代码来实现最佳效果。