Postgres:深度优先递归查询中的嵌套记录

Postgres: Nested records in a Recursive query in depth first manner

我正在开发一个简单的评论系统,用户可以在其中评论其他评论,从而创建层次结构。为了按层次顺序获取评论,我在 Postgres 中使用 Common Table Expression。

以下是使用的字段和查询:

id
user_id
parent_comment_id
message

WITH RECURSIVE CommentCTE AS (
    SELECT id, parent_comment_id, user_id
    FROM comment
    WHERE parent_comment_id is NULL

    UNION ALL

    SELECT child.id, child.parent_comment_id, child.user_id
    FROM comment child
    JOIN CommentCTE
    ON child.parent_comment_id = CommentCTE.id
)
SELECT * FROM CommentCTE

以上查询returns记录广度优先:

id       parent_comment_id       user_id
10              null                30
9               null                30
11               9                  30
14              10                  31
15              10                  31
12              11                  30
13              12                  31

但是是否可以修改它以实现如下所示的内容,即以深度优先的方式一起返回该评论集的记录?重点是通过这种方式获取数据,让前端渲染更流畅

id       parent_comment_id       user_id
9               null                30
11               9                  30
12              11                  30
13              12                  31
10              null                30
14              10                  31
15              10                  31

通常我通过合成一个可以按词法排序的 "Path" 列来解决这个问题,例如0001:0003:0006:00090001:0003:0006 的 child。每个 child 条目都可以通过将路径元素连接到 parent 的路径来创建。您不必return此列给客户端,仅用于排序。

id       parent_comment_id       user_id     sort_key
9               null                30       0009
11               9                  30       0009:0011
12              11                  30       0009:0011:0012
13              12                  31       0009:0011:0012:0013
10              null                30       0010
14              10                  31       0010:0014
15              10                  31       0010:0015

路径元素不必是任何特别的东西,只要它按照您希望 children 在该级别排序的顺序进行词法排序,并且在该级别是唯一的。基于 auto-incrementing ID 就可以了。

严格来说,使用固定长度的路径元素并不是必需的,但更容易推理。

WITH RECURSIVE CommentCTE AS (
SELECT id, parent_comment_id, user_id, 
    lpad(id::text, 4) sort_key
FROM comment
WHERE parent_comment_id is NULL

UNION ALL

SELECT child.id, child.parent_comment_id, child.user_id, 
    concat(CommentCTE.sort_key, ':', lpad(id::text, 4))
FROM comment child
JOIN CommentCTE
ON child.parent_comment_id = CommentCTE.id
)
SELECT * FROM CommentCTE order by sort_key