PostgreSQL - 经理层次结构的递归 CTE 自加入



我在PostgreSQL中将递归查询放在一起以显示员工之间的管理层次结构时遇到了麻烦。为了实现这一点,我需要将表自身自行连接,直到每个员工都到达其层次结构的末尾。我的数据在给定日期如下所示:

+------------+-------------+-----------------+---------------------+
|    date    | employee_id | terminated_flag | manager_employee_id |
+------------+-------------+-----------------+---------------------+
| 2019-01-31 |           3 |               0 |                   2 |
| 2019-01-31 |           2 |               1 |                   1 |
| 2019-01-31 |           1 |               0 |                     |
+------------+-------------+-----------------+---------------------+

理想情况下,我想创建一个 JSONB 列,其中包含给定员工的完整层次结构和经理详细信息。我知道我可以通过递归附加到现有的 JSONB 列,但达到这一点一直是困难。所需的输出如下所示(删除列以方便阅读):

+------------+-------------+-----------------------------------------------------------+
|    date    | employee_id |                     manager_hierarchy                     |
+------------+-------------+-----------------------------------------------------------+
| 2019-01-31 |           3 | {{"level":1,"id":2,"term":1},{"level":2,"id":1,"term":0}} |
| 2019-01-31 |           2 | {{"level":1,"id":1,"term":0}}                             |
| 2019-01-31 |           1 |                                                           |
+------------+-------------+-----------------------------------------------------------+

在我的数据集中,从任何给定的员工到首席执行官之间可能有 N 个级别,因此一旦每个员工都到达 CEO,递归就需要结束,而 CEO 的 manager_employee_id 值为 NULL。

这可能吗?感谢您的帮助!

我会将其视为递归 CTE 以获取层次结构,然后聚合以创建 jsonb 值:

with recursive t as (
      select v.*
      from (values (3, 2, 0), (2, 1, 1), (1, null, 0)) v(employee_id, manager_employee_id, terminated_flag)
     ),
     cte as (
      select distinct employee_id, manager_employee_id, terminated_flag, 1 as lev
      from t
      union all
      select cte.employee_id, t.manager_employee_id, t.terminated_flag, lev + 1
      from cte join
           t
           on cte.manager_employee_id = t.employee_id
     )
select employee_id, jsonb_agg(jsonb_build_object('level', lev, 'id', manager_employee_id, 'terminated_flag', terminated_flag))
from cte
group by employee_id;

这是一个数据库<>小提琴。

最新更新