Drizzle ORM 中表别名使用的最佳实践

在使用 Drizzle ORM 进行数据库查询时，开发者经常会遇到需要多次引用同一个表的情况。本文深入探讨了 Drizzle ORM 中表别名的正确使用方法，特别是针对不同连接类型组合时可能出现的类型推断问题。

问题现象

当开发者尝试在 Drizzle ORM 中为同一个表创建两个别名，并分别使用不同的连接方式（如一个内连接和一个左连接）时，会遇到查询返回类型被推断为 never[] 的问题。例如：

const fromCompany = aliasedTable(company, "fromCompany");
const toCompany = aliasedTable(company, "toCompany");

const data = await db
  .select({
    referral,
    fromCompany,
    toCompany,
  })
  .from(referral)
  .innerJoin(fromCompany, eq(fromCompany.id, referral.fromCompanyId))
  .leftJoin(toCompany, eq(toCompany.id, referral.toCompanyId));

在这种情况下，data 的类型会被错误地推断为 never[]，尽管运行时查询能够正常执行。

根本原因

这个问题源于 aliasedTable 函数的预期用途。实际上，aliasedTable 是专门为关系查询（RQB）API 设计的，而不是用于常规的查询构建器。当在常规查询中使用它时，特别是在混合连接类型的情况下，类型系统无法正确推断结果类型。

解决方案

Drizzle ORM 提供了专门的 alias 函数来处理表别名，这个函数应该用于常规查询构建器中的表别名需求。对于不同的数据库类型，导入路径略有不同：

• PostgreSQL: import { alias } from 'drizzle-orm/pg-core'
• SQLite: import { alias } from 'drizzle-orm/sqlite-core'
• MySQL: import { alias } from 'drizzle-orm/mysql-core'

修正后的代码应该如下：

const fromCompany = alias(company, "fromCompany");
const toCompany = alias(company, "toCompany");

const data = await db
  .select({
    referral,
    fromCompany,
    toCompany,
  })
  .from(referral)
  .innerJoin(fromCompany, eq(fromCompany.id, referral.fromCompanyId))
  .leftJoin(toCompany, eq(toCompany.id, referral.toCompanyId));

连接类型组合的注意事项

虽然使用 alias 函数解决了类型推断问题，但开发者仍需注意不同连接类型组合对查询结果的影响：

1. 双内连接：只有当两个连接条件都满足时才会返回记录
2. 双左连接：即使连接条件不满足也会返回主表记录，相关字段为NULL
3. 混合连接：内连接确保必须存在，左连接允许不存在

最佳实践总结

1. 在常规查询构建器中总是使用 alias 而非 aliasedTable
2. 根据数据库类型从正确的核心模块导入 alias 函数
3. 仔细考虑查询中不同连接类型的组合对结果集的影响
4. 对于复杂的多表查询，建议先测试类型推断是否符合预期

通过遵循这些最佳实践，开发者可以避免类型推断问题，同时构建出既类型安全又符合业务需求的数据库查询。