sql - 使用 Postgres JSON 函数从具有与其他表关系的表生成完整的 JSON 对象

Question

我想使用 Postgres 从“user”、“user_role”、“role”和“permission”生成用户 JSON 对象，如下所示:

CREATE TABLE "user" (
        id SERIAL PRIMARY KEY,
        name TEXT,
        location TEXT
);
CREATE TABLE "role" (
        id INT NOT NULL PRIMARY KEY,
        name TEXT
);
CREATE TABLE "user_role" (
        id SERIAL PRIMARY KEY,
        user_id BIGINT,
        role_id INT,
        FOREIGN KEY ("user_id") REFERENCES "user"("id"),
        FOREIGN KEY ("role_id") REFERENCES "role"("id")
);
CREATE TABLE "permission" (
        id INT NOT NULL PRIMARY KEY,
        role_id INT,
        name TEXT,
        FOREIGN KEY ("role_id") REFERENCES "role"("id")
);

INSERT INTO "user" ("name", "location") VALUES ('Hamed', 'Berlin');
INSERT INTO "user" ("name", "location") VALUES ('Zhang', 'Shenzhen');
INSERT INTO "user" ("name", "location") VALUES ('Jake', 'Vancouver');
INSERT INTO "role" ("id", "name") VALUES (1, 'admin');
INSERT INTO "role" ("id", "name") VALUES (2, 'user');
INSERT INTO "permission" ("id", "role_id", "name") VALUES (1, 1, 'add');
INSERT INTO "permission" ("id", "role_id", "name") VALUES (2, 1, 'delete');
INSERT INTO "permission" ("id", "role_id", "name") VALUES (3, 1, 'update');
INSERT INTO "permission" ("id", "role_id", "name") VALUES (4, 2, 'read');
INSERT INTO "user_role" ("user_id", "role_id") VALUES (1, 1);
INSERT INTO "user_role" ("user_id", "role_id") VALUES (1, 2);
INSERT INTO "user_role" ("user_id", "role_id") VALUES (2, 1);

SELECT (
    json_build_object (
        'id', u.id,
        'roles', json_agg(
            r.id
        )
    )
) FROM "user" AS u
LEFT JOIN "user_role" AS r ON r.user_id = u.id
GROUP by u.id;

SELECT (
    json_build_object (
        'id', r.id,
        'permissions', json_agg(
            p.id
        )
    )
) FROM "role" AS r
LEFT JOIN "permission" AS p ON p.role_id = r.id
GROUP by r.id;

输出是:

--------------------------------
1  |  {"id" : 1, "roles" : [1, 2]}
2  |  {"id" : 2, "roles" : [3]}
3  |  {"id" : 3, "roles" : [null]}

--------------------------------
1  |  {"id" : 1, "permissions" : [1, 2, 3]}
2  |  {"id" : 2, "permissions" : [4]}

我有两个问题:

1. 如何摆脱[null]。我们可以用 [] 代替吗？
2. 我想生成这个结果:

--------------------------------
1  |  {"id" : 1, "roles" : [1, 2], "permissions": [1, 2, 3, 4]}
2  |  {"id" : 2, "roles" : [3]   , "permissions": [4]}
3  |  {"id" : 3, "roles" : [null], "permissions": []}

查询应该是什么样子的？您可以找到代码并运行它 here in DB fiddle还。谢谢。

Answer 1

首先，您应该从role 表中获取用户的角色，user_role 只是一个桥接表。查看查询结果:

select 
    u.id as user_id, 
    ur.id as user_role_id, 
    r.id as role_id, 
    p.id as permission_id
from "user" u
left join "user_role" ur on ur.user_id = u.id
left join "role" r on r.id = ur.role_id
left join "permission" p on p.role_id = r.id

 user_id | user_role_id | role_id | permission_id 
---------+--------------+---------+---------------
       1 |            1 |       1 |             3
       1 |            1 |       1 |             2
       1 |            1 |       1 |             1
       1 |            2 |       2 |             4
       2 |            3 |       1 |             3
       2 |            3 |       1 |             2
       2 |            3 |       1 |             1
       3 |              |         |              
(8 rows)

用户 #2 具有角色 #1(不是 #3)。

您可以轻松转换查询以获取聚合数据:

select 
    jsonb_build_object(
        'id', u.id,
        'roles', jsonb_agg(distinct r.id order by r.id),
        'permissions', jsonb_agg(p.id order by p.id))
from "user" u
left join "user_role" ur on ur.user_id = u.id
left join "role" r on r.id = ur.role_id
left join "permission" p on p.role_id = r.id
group by u.id

                   jsonb_build_object                    
---------------------------------------------------------
 {"id": 1, "roles": [1, 2], "permissions": [1, 2, 3, 4]}
 {"id": 2, "roles": [1], "permissions": [1, 2, 3]}
 {"id": 3, "roles": [null], "permissions": [null]}
(3 rows)

没有标准函数可以去除 jsonb 数组的空元素。您可以创建自己的:

create or replace function jsonb_strip_null_elements(jsonb)
returns jsonb language sql immutable as $$
    select coalesce(jsonb_agg(e), '[]')
    from jsonb_array_elements($1) as e
    where e <> 'null'
$$;

Live demo in db-fiddle.