使用Ecto的递归CTE

时间:2016-09-08 05:20:09

标签: postgresql elixir phoenix-framework ecto recursive-cte

如何在计划与Ecto一起运行的查询中使用递归CTE的结果?例如,让我说我有一个表,节点,结构如下:

-- nodes table example --

id  parent_id
1   NULL
2   1
3   1
4   1
5   2
6   2
7   3
8   5

我还有另一个表nodes_users结构如下:

-- nodes_users table example --

node_id   user_id
1         1
2         2
3         3
5         4

现在,我想抓住具有特定节点或其上方节点的所有用户,为了举例,我们选择带有id 8的节点。

我可以使用以下递归查询来执行此操作:

WITH RECURSIVE nodes_tree AS (
    SELECT *
    FROM nodes
    WHERE nodes.id = 8
UNION ALL
    SELECT n.*
    FROM nodes n
    INNER JOIN nodes_tree nt ON nt.parent_id = n.id
)
SELECT u.* FROM users u
INNER JOIN users_nodes un ON un.user_id = u.id
INNER JOIN nodes_tree nt ON nt.id = un.node_id

对于ID为1,2和4的用户,这应该返回用户。*。

我不确定如何使用ecto运行相同的查询,理想情况是以返回可链接输出的方式运行。我知道我可以使用片段宏将原始SQL插入到我的查询中,但是我不确定这将用于何种用途,或者这甚至是最合适的路径。

欢迎提供帮助和/或建议!

1 个答案:

答案 0 :(得分:12)

我能够使用片段完成此任务。这是我使用的代码示例。我可能会将此方法移动到存储过程。

Repo.all(MyProj.User,
  from u in MyProj.User,
  join: un in MyProj.UserNode, on: u.id == un.user_id,
  join: nt in fragment("""
  (
    WITH RECURSIVE node_tree AS (
      SELECT *
      FROM nodes
      WHERE nodes.id = ?
    UNION ALL
      SELECT n.*
      FROM nodes n
      INNER JOIN node_tree nt ON nt.parent_id == n.id
    )
  ) SELECT * FROM node_tree
  """, ^node_id), on: un.node_id == nt.id
)