为何在ActiveRecord查询结果上调用.where方法会报语法错误？

问题根源 #

PostgreSQL的查询执行顺序是 WHERE → SELECT，WHERE子句执行时，SELECT里定义的别名（比如plan_name）还未生成，所以直接在where中引用这个别名会触发语法错误。

你之前遍历结果正常，是因为查询已经完整执行，Ruby层面直接读取返回的字段；to_sql输出的SQL能正常执行，是因为那是完整的SELECT语句，数据库执行时会先处理JOIN和CASE逻辑，再返回带plan_name的结果，而非在WHERE阶段引用这个别名。

解决方案 #

方案1：在WHERE子句中重复CASE逻辑

直接把生成plan_name的CASE表达式写到WHERE条件里，绕开别名引用：

orgs_with_plans.where(<<~SQL, 'trial')
  (CASE
    WHEN subscription_data.id IS NULL THEN
      (CASE
        WHEN organizations.trial_end_date > CURRENT_DATE THEN 'trial'
        ELSE 'expired'
      END)
    ELSE subscription_data.plan_name
  END) = ?
SQL

方案2：将查询包装为子查询

把原查询转为子查询，外层查询即可直接引用plan_name别名。修改orgs_with_plans方法：

def orgs_with_plans
  subquery = Subscription.select("DISTINCT ON (organization_id) *")
    .order(:organization_id, created_at: :desc)

  base_query = Organization
    .joins("LEFT JOIN (#{subquery.to_sql}) AS subscription_data ON organizations.id = subscription_data.organization_id")
    .select("
      organizations.*,
      (CASE
        WHEN subscription_data.id IS NULL THEN
          (CASE
            WHEN organizations.trial_end_date > CURRENT_DATE THEN 'trial'
            ELSE 'expired'
          END)
        ELSE subscription_data.plan_name
      END) AS plan_name
    ")

  # 基于子查询构建新的关系，支持引用plan_name
  Organization.from("(#{base_query.to_sql}) AS organizations_with_plans")
end

之后就能正常调用：

orgs_with_plans.where(plan_name: 'trial')

方案3：用Arel构建查询（更优雅，避免字符串拼接）

用ActiveRecord的Arel API构建CASE表达式和JOIN，既保持代码可读性，又能灵活复用逻辑：

def orgs_with_plans
  subquery = Subscription.select(Arel.sql("DISTINCT ON (organization_id) *"))
    .order(:organization_id, created_at: :desc)

  subscription_data = Arel::Table.new('subscription_data')
  organization = Organization.arel_table

  # 构建plan_name的CASE表达式
  plan_name_case = Arel::Nodes::Case.new(subscription_data[:id].eq(nil))
    .when(true)
    .then(
      Arel::Nodes::Case.new(organization[:trial_end_date].gt(Arel.sql('CURRENT_DATE')))
        .when(true).then('trial')
        .else('expired')
    )
    .else(subscription_data[:plan_name])

  Organization
    .joins(Arel::Nodes::OuterJoin.new(
      Arel::Nodes::As.new(subquery.arel, subscription_data),
      organization[:id].eq(subscription_data[:organization_id])
    ))
    .select(organization[Arel.star], plan_name_case.as('plan_name'))
end

如果需要添加WHERE条件，可直接复用plan_name_case：

orgs_with_plans.where(plan_name_case.eq('trial'))

内容的提问来源于stack exchange，提问作者imjared