Drizzle ORM 中 $count 方法的表名引用问题解析

在使用 Drizzle ORM 进行数据库查询时，开发者可能会遇到一个关于表名引用的常见陷阱。本文将通过一个典型场景，深入分析问题本质并提供解决方案。

问题现象

当开发者尝试使用 Drizzle ORM 0.35.1 版本的 $count 方法构建关联查询时，可能会遇到如下错误：

Unknown column 'customers.customer_id' in 'where clause'

具体表现为，当按照官方示例编写如下代码时：

await ctx.db.query.customers.findMany({
  extras: {
    tasksCount: ctx.db
      .$count(
        tasks,
        eq(tasks.customerId, customers.id),
      )
      .as("tasksCount"),
  },
});

生成的 SQL 子查询会错误地引用表名：

(select count(*) from "tasks" where "customers"."customer_id" = "customers"."id")

问题根源

这个问题的核心在于 Drizzle ORM 的列引用机制。在构建 where 条件时，ORM 没有正确处理表名前缀，导致生成的 SQL 语句中表名引用不正确。

具体来说：

1. tasks.customerId 本应生成 tasks.customer_id
2. customers.id 本应生成 customers.id
3. 但实际生成的 where 条件却变成了 customers.customer_id = customers.id

解决方案

目前有两种可行的解决方法：

方法一：使用原始 SQL 表达式

tasksCount: ctx.db
  .$count(
    tasks,
    sql`\`tasks\`.\`customer_id\` = \`customers\`.\`id\``,
  )
  .as("tasksCount"),

这种方法直接指定完整的表名和列名引用，确保生成的 SQL 语句正确。

方法二：使用 select 方法替代

对于更复杂的关联查询场景，官方建议使用 select 方法并显式指定 join 条件：

await ctx.db.select({
  customers: customers,
  tasksCount: count(tasks.id).as("tasksCount")
})
.from(customers)
.leftJoin(tasks, eq(tasks.customerId, customers.id))
.groupBy(customers.id);

最佳实践建议

1. 在构建复杂关联查询时，优先考虑使用 select 方法而非 $count
2. 如果必须使用 $count，确保检查生成的 SQL 语句是否正确引用了表名
3. 对于跨表条件，考虑使用原始 SQL 表达式确保引用正确
4. 在开发过程中，启用 SQL 日志输出以验证生成的查询语句

总结

Drizzle ORM 的 $count 方法在表名引用上存在一定的局限性，开发者需要特别注意跨表条件的构建方式。通过理解 ORM 的生成机制并采用适当的解决方法，可以避免这类问题并构建出正确的关联查询。