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

时间:2018-04-30 10:37:04

标签: postgresql common-table-expression recursive-query

我正在开发一个简单的评论系统,用户可以在其中评论其他评论,从而创建层次结构。为了按层次结构顺序获取注释,我在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

上述查询以广度优先的方式返回记录:

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

1 个答案:

答案 0 :(得分:3)

通常我通过合成一个“路径”列来解决这个问题,该列可以在词汇上排序,例如, 0001:0003:0006:00090001:0003:0006的孩子。可以通过将path元素连接到父路径来创建每个子条目。您不必将此列返回给客户端,只需将其用于排序。

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

如果路径元素按照您希望该级别的子级排序的顺序进行词法排序,并且在该级别上是唯一的,则路径元素不必具有任何特殊性。基于自动递增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