Ecto 中使用 selected_as 解决 PostgreSQL 分组错误问题

在 Elixir 的 Ecto 库中处理 PostgreSQL 数据库时，开发人员可能会遇到一个特殊的分组错误问题。这个问题通常出现在使用 LTree 路径操作函数（如 subpath）进行分组查询时。

问题现象

当开发人员尝试执行类似以下的 Ecto 查询时：

level = 2
from(t in "table",
  group_by: subpath(t.path, 0, ^level),
  select: %{
    value: fragment("SUM(?)", t.value),
    path: subpath(t.path, 0, ^level)
  }
)

会收到 PostgreSQL 返回的错误信息："column 't.path' must appear in the GROUP BY clause or be used in an aggregate function"。然而，直接在 PostgreSQL 中执行等效的 SQL 查询却能正常工作。

问题根源

深入分析后，我们发现 Ecto 在生成 SQL 时会为 SELECT 和 GROUP BY 子句中的相同表达式使用不同的参数占位符（$1 和 $2）。PostgreSQL 的查询优化器会将这些视为不同的表达式，从而触发分组错误。

解决方案

Ecto 提供了 selected_as 宏来解决这类问题。这个宏允许我们为表达式指定一个别名，确保在查询的不同部分引用的是同一个表达式实例。修正后的查询如下：

from(t in "table",
  group_by: selected_as(:path_alias),
  select: %{
    value: fragment("SUM(?)", t.value),
    path: selected_as(subpath(t.path, 0, ^level), :path_alias)
  }
)

技术原理

selected_as 的工作原理是：

1. 为指定的表达式创建一个内部别名
2. 在查询生成过程中保持这个别名的唯一性
3. 确保 SELECT 和 GROUP BY 子句引用的是完全相同的表达式

这种方法不仅适用于 LTree 操作，也适用于其他需要确保表达式一致性的复杂查询场景。

最佳实践

在使用 Ecto 进行复杂查询时，特别是涉及以下情况时，应考虑使用 selected_as：

• 查询中包含函数调用（如字符串操作、日期处理等）
• 需要对计算字段进行分组
• 查询中多次使用相同的复杂表达式

通过合理使用 selected_as，可以避免许多微妙的 SQL 生成问题，同时保持代码的清晰性和可维护性。