Drizzle ORM 中数组类型默认值设置问题解析

问题背景

在使用 Drizzle ORM 进行 PostgreSQL 数据库建模时，开发者遇到了一个关于数组类型默认值设置的常见问题。具体表现为当尝试为 integer[] 类型的列设置默认值时，系统报错提示"column is of type integer[] but default expression is of type integer"。

问题复现

在 Drizzle ORM 的 schema 定义中，开发者尝试这样定义一个 users 表：

export const users = pgTable('users', {
  id: serial('id').primaryKey(),
  fullname: text('fullname').notNull(),
  roles: integer('roles').array().default([1]).notNull(),
});

这段代码的意图是为 roles 字段设置一个默认值 [1]，但实际执行时会抛出类型不匹配的错误。

技术分析

PostgreSQL 中的数组类型与普通类型在默认值设置上有显著差异。直接使用 JavaScript 数组 [1] 作为默认值会导致类型系统无法正确识别，因为：

1. PostgreSQL 需要明确的类型转换来区分普通整数和整数数组
2. Drizzle ORM 的类型系统需要明确的 SQL 表达式来生成正确的默认值语法

解决方案

针对这个问题，Drizzle ORM 提供了两种解决方案：

1. 使用 SQL 原生表达式

对于固定值的数组默认值，可以使用 sql 模板标签：

roles: integer('roles').array().default(sql`ARRAY[1]::integer[]`).notNull()

2. 使用动态变量

如果需要使用变量构建数组默认值，可以使用 sql.raw：

const defaultRoles = [1, 2, 3];
roles: integer('roles').array().default(
  sql.raw(`ARRAY[${defaultRoles.join(',')}]::integer[]`)
).notNull()

版本更新

在 Drizzle ORM 的后续版本（drizzle-kit 0.24.0）中，这个问题已经得到修复。更新后，开发者可以直接使用更简洁的语法来设置数组默认值。

最佳实践

1. 对于生产环境，建议升级到最新版本以获得更好的开发体验
2. 在必须使用旧版本时，采用上述 SQL 表达式方案
3. 数组默认值设置时，始终考虑类型明确性，避免隐式转换

总结

Drizzle ORM 作为现代化的 TypeScript ORM 工具，在处理 PostgreSQL 高级特性如数组类型时，既保留了类型安全性，又提供了灵活的解决方案。理解数据库类型系统与 ORM 类型系统之间的映射关系，是高效使用此类工具的关键。