将递归 CTE 与 Ecto 结合使用

Question

我将如何在计划使用 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 的节点。

我可以使用以下递归 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

这应该返回 users.* 对于 id 为 1、2 和 4 的用户。

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

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

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
)

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)