将 SQL 语句转换为 rails 命令

Question

我有一种情况，我只需要从特定的活动记录查询响应中获取几条记录。

@annotations = Annotations.select('*', ROW_NUMBER() OVER (PARTITION BY column_a) ORDER BY column_b)

上面是@annotations 是活动记录响应的查询，我想在其上应用以下逻辑。有没有更好的方法来用rails方式编写以下逻辑？

with some_table as
(
 select *, row_number() over (partition by column_a order by column_b) rn
 from the_table
)
select * from some_table
where (column_a = 'ABC' and rn <= 10) or (column_b <> 'AAA')

Answer 1

ActiveRecord 在其高级 API 中不提供 CTE；但是，我们可以用一点 Arel 将其作为 FROM 子句中的子查询

annotations_table = Annotation.arel_table 
sub_query = annotations_table.project(
  Arel.star,
  Arel.sql('row_number() over (partition by column_a order by column_b)').as(:rn)
)
query = Arel::Nodes::As.new(sub_query, Arel.sql(annotations_table.name))

Annotation.from(query).where(
  annotations_table[:column_a].eq('ABC').and(
    annotations_table[:rn].lt(10)
  ).or(annotations_table[:column_b].not_eq('AAA'))
)

结果将是使用您的 CTE 和您描述的过滤器的 Annotation 对象的集合。

SQL：

select annotations.* 
from (
 select *, row_number() over (partition by column_a order by column_b) AS rn
 from annotations
) AS annotations
where (annotations.column_a = 'ABC' and annotations.rn <= 10) or (annotations.column_b <> 'AAA')

注意事项：

• 通过一些额外的工作，我们可以使它成为 CTE，但在这种情况下似乎不需要它

• 我们也可以使用一堆技巧将这个 row_number() over (partition by column_a order by column_b) 转换为 Arel，但这似乎与问题无关。