使用递归CTE与Ecto

Question

我该如何在使用Ecto运行的查询中使用递归CTE的结果？例如，假设我有一个名为nodes的表，结构如下：

-- 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的节点。

我可以使用以下递归的postgresql查询来实现：

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的用户的 users.*。

我不确定如何使用 Ecto 运行同样的查询，最好以可链接的输出方式返回。我知道可以使用 fragment 宏将原始 SQL 插入到查询中，但我不确定在此用例中应该放置它，或者它是否是最合适的路线。

帮助和/或建议将不胜感激！

- Saba

---

Ecto 3.2 引入了 Ecto DSL 支持。 - narrowtux

Answer 1

我使用了一个片段来完成这个任务。这是我使用的代码示例。我可能会将这种方法迁移到存储过程中。

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
)

- Saba

Answer 2

时至今日，距离该问题被提出已经近7年了，Ecto的文档中增加了一个关于如何使用递归CTE的章节。以下是它的工作原理。

首先，构建查询的基本情况和递归步骤。然后，将其联合并传递给Ecto.Query.with_cte。您还需要设置recursive_ctes(true)：

base_case =
  from(row in "nodes", where: row.id == ^node_id)

recursive_step =
  from(n in "nodes", join: nt in "node_tree", on: nt.parent_id == n.id)

node_tree = base_case |> union(^recursive_step)

MyProj.User
|> recursive_ctes(true)
|> with_cte("node_tree", as: ^node_tree)
|> join(:inner, [u], un in MyProj.UserNode, on: u.id == un.user_id)
|> join(:inner, [u, un], nt in "node"tree", on: un.node_id == nt.id)

- Frerich Raabe