Drizzle ORM 类型推断问题分析与解决方案

问题概述

Drizzle ORM 是一个现代化的 TypeScript ORM 框架，近期在 0.32.0 版本中出现了一个关键的类型推断问题。许多开发者报告称，在使用 $inferInsert 类型推断时，得到的类型为空对象，而 $inferSelect 则将所有字段都标记为必填，这与预期行为不符。

问题表现

开发者在使用 Drizzle ORM 定义表结构后，期望通过 $inferInsert 获取插入操作的类型定义，但实际得到的却是一个空类型。例如：

const users = pgTable('user', {
  id: serial('id').primaryKey(),
  email: text('email').unique(),
  firstName: text('first_name'),
  // 其他字段...
});

type NewUser = typeof users.$inferInsert; // 结果为 {}

同时，$inferSelect 也存在问题，它会将所有字段都标记为必填，即使某些字段在数据库中是可为空的。

根本原因分析

经过深入调查，这个问题与 TypeScript 的 strictNullChecks 配置密切相关。Drizzle ORM 的类型系统依赖于 TypeScript 的严格空值检查机制来正确推断哪些字段是可选的。

在 Drizzle ORM 内部，类型推断逻辑会检查字段是否为生成字段（generated columns）或是否允许为空。当 strictNullChecks 被禁用时，TypeScript 会将 null 视为可以赋值给任何类型，这导致 Drizzle 的类型推断系统无法正确识别可选字段。

解决方案

1. 启用严格空值检查

最直接的解决方案是在 tsconfig.json 中启用 strictNullChecks：

{
  "compilerOptions": {
    "strictNullChecks": true
  }
}

这能确保 Drizzle ORM 的类型推断系统正常工作，但可能会要求开发者修复代码中其他与空值相关的问题。

2. 临时类型覆盖

如果无法立即启用严格空值检查，可以手动定义类型：

type NewUser = {
  id?: number;
  email: string;
  firstName?: string | null;
  // 其他字段...
};

3. 类型断言

在插入或更新操作中使用类型断言：

db.insert(users).values(userData as any);

虽然这不是类型安全的解决方案，但可以作为临时措施。

最佳实践建议

1. 始终启用严格类型检查：这不仅能解决 Drizzle ORM 的问题，还能提高整体代码质量。

2. 考虑数据库约束：在设计表结构时，明确每个字段的可空性和默认值，这会影响类型推断结果。

3. 逐步迁移：如果现有代码库无法立即启用严格模式，可以考虑逐步迁移，先修复 Drizzle 相关的类型问题。

技术细节

Drizzle ORM 的类型推断系统核心在于 OptionalKeyOnly 类型，它会检查字段是否为生成字段或是否允许为空。在严格模式下，系统能正确识别这些条件，但在非严格模式下，类型系统无法区分 null 和普通类型。

结论

Drizzle ORM 的类型推断问题本质上是由 TypeScript 配置与 ORM 类型系统的交互方式引起的。启用 strictNullChecks 是最推荐的解决方案，它不仅解决了当前问题，还能帮助开发者编写更健壮的代码。对于暂时无法启用严格模式的团队，可以考虑手动类型定义或类型断言作为过渡方案。