使用 Ruby on Rails 进行 Postgres 公共表表达式查询

2024-03-28

我试图找到在 Rails 应用程序中使用通用表表达式进行 Postgres 查询的最佳方法,因为我知道 ActiveRecord 显然不支持 CTE。

我有一张桌子叫user_activity_transitions其中包含一系列正在启动和停止的用户活动记录(每行表示状态更改:例如启动或停止)。

One user_activity_id可能有很多情侣开始停下来,他们分布在不同的两排。 如果活动当前正在进行并且尚未停止,则也可能只有“已开始”。这sort_key第一个状态从 0 开始,每次状态更改时增加 10。

id      to_state     sort_key     user_activity_id    created_at
1       started      0            18                  2014-11-15 16:56:00
2       stopped      10           18                  2014-11-15 16:57:00
3       started      20           18                  2014-11-15 16:58:00
4       stopped      30           18                  2014-11-15 16:59:00
5       started      40           18                  2014-11-15 17:00:00

我想要的是以下输出,将开始-停止的对组合在一起,以便能够计算持续时间等。

user_activity_id     started_created_at      stopped_created_at
18                   2014-11-15 16:56:00     2014-11-15 16:57:00
18                   2014-11-15 16:58:00     2014-11-15 16:59:00
18                   2014-11-15 17:00:00     null

表的实现方式使得运行该查询变得更加困难,但对于未来的更改(例如新的中间状态)更加灵活,因此不会进行修改。

我的 Postgres 查询(以及 Rails 中的相关代码):

query = <<-SQL
    with started as (
    select 
        id,
        sort_key,
        user_activity_id,
        created_at as started_created_at
    from
        user_activity_transitions
    where  
        sort_key % 4 = 0
    ), stopped as (
    select 
        id,
        sort_key-10 as sort_key2,
        user_activity_id,
        created_at as stopped_created_at
    from
    user_activity_transitions
    where
        sort_key % 4 = 2
    )
    select
        started.user_activity_id AS user_activity_id,
        started.started_created_at AS started_created_at,
        stopped.stopped_created_at AS stopped_created_at
    FROM
        started
    left join stopped on stopped.sort_key2 = started.sort_key
    and stopped.user_activity_id = started.user_activity_id
SQL

results = ActiveRecord::Base.connection.execute(query)

它的作用是根据排序键的模数检查“欺骗”SQL 连接 2 个连续的行。

查询工作正常。但是使用这个原始的 AR 调用让我很恼火,尤其是因为connection.execute回报相当混乱。我基本上需要循环遍历结果并将其放入正确的哈希中。

2个问题:

  1. 有没有办法摆脱 CTE 并使用运行相同的查询 轨道魔法?
  2. 如果没有,是否有更好的方法来获得我想要的漂亮哈希结果?

请记住,我对 Rails 很陌生,而且不是查询专家,因此可能会有明显的改进......

多谢!


虽然 Rails 不直接支持 CTE,但您可以模拟单个 CTE 并且仍然利用 ActiveRecord。使用 CTE 代替from子查询。

Thing
  .from(
    # Using a subquery in place of a single CTE
    Thing
      .select(
        '*',
        %{row_number() over(
            partition by
              this, that
            order by
              created_at desc
          ) as rank
        }
      )
    :things
  )
  .where(rank: 1)

这并不完全相同,但相当于......

with ranked_things as (
  select
    *,
    row_number() over(
      partition by
        this, that
      order by
        created_at desc
    ) as rank
)
select *
from ranked_things
where rank = 1
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用 Ruby on Rails 进行 Postgres 公共表表达式查询 的相关文章

随机推荐