postgresql - 从 PL/pgSQL 函数返回嵌套复合类型

Question

我正在尝试将这种格式的嵌套数据从 PostgreSQL 返回到 PHP 关联数组中。

[
  'person_id': 1,
  'name': 'My Name',
  'roles': [
      [ 'role_id': 1, 'role_name': 'Name' ],
      [ 'role_id': 2, 'role_name': 'Another role name' ]
  ]
]

似乎可以使用 composite types . This answer描述如何从函数返回复合类型，但不处理复合类型数组。我在使用数组时遇到了一些问题。

这是我的表和类型:

CREATE TEMP TABLE people (person_id integer, name text);
INSERT INTO "people" ("person_id", "name") VALUES
    (1, 'name!');

CREATE TEMP TABLE roles (role_id integer, person_id integer, role_name text);
INSERT INTO "roles" ("role_id", "person_id", "role_name") VALUES
    (1, 1,  'role name!'),
    (2, 1,  'another role');

CREATE TYPE role AS (
    "role_name" text
);

CREATE TYPE person AS (
    "person_id" int,
    "name" text,
    "roles" role[]
);

我的 get_people() 函数解析正常，但存在运行时错误。现在我收到错误:array value must start with "{"or dimension information

CREATE OR REPLACE FUNCTION get_people()
    RETURNS person[] AS $$
DECLARE myroles role[];
DECLARE myperson people%ROWTYPE;
DECLARE result person[];
BEGIN
    FOR myperson IN
        SELECT *
        FROM "people"
    LOOP
        SELECT "role_name" INTO myroles
        FROM "roles"
        WHERE "person_id" = myperson.person_id;

        result := array_append(
            result,
            (myperson.person_id, myperson.name, myroles::role[])::person
        );
    END LOOP;

    RETURN result;
END; $$ LANGUAGE plpgsql;

UPDATE 回复 Erwin Brandstetter 在他的回答结尾处的问题:
是的，我可以返回复合类型的 SETOF。我发现 SET 比数组更容易处理，因为 SELECT 查询返回 SET。我宁愿返回嵌套数组的原因是因为我认为将嵌套数据表示为一组行有点尴尬。这是一个例子:

 person_id | person_name | role_name | role_id
-----------+-------------+-----------+-----------
         1 | Dilby       | Some role |    1978
         1 | Dilby       | Role 2    |       2
         2 | Dobie       | NULL      |    NULL

在此示例中，人员 1 有 2 个角色，人员 2 没有。我正在为我的另一个 PL/pgSQL 函数使用这样的结构。我写了一个脆弱的 PHP 函数，将这样的记录集转换为嵌套数组。

这种表示法工作正常，但我担心向该结构中添加更多嵌套字段。如果每个人也有一组作业怎么办？状态？等我的转换功能将不得不变得更加复杂。数据的表示也将很复杂。如果一个人有 n 个角色，m 个职位，和 o 个状态，则该人填写 max(n, m, o) 行，包含 person_id、person_name 以及他们在额外行中无用地复制的任何其他数据。我一点也不担心性能，但我想以最简单的方式做到这一点。当然..也许这是最简单的方法!

我希望这有助于说明为什么我宁愿直接处理 PostgreSQL 中的嵌套数组。当然，我很想听听您的任何建议。

对于使用 PHP 处理 PostgreSQL 复合类型的任何人，我发现这个库对于在 PHP 中解析 PostgreSQL 的 array_agg() 输出非常有用:https://github.com/nehxby/db_type .此外，这个项目看起来很有趣:https://github.com/chanmix51/Pomm

Answer 1

考虑这个(改进和修复的)测试用例，使用 PostgreSQL 9.1.4 测试:

CREATE SCHEMA x;
SET search_path = x, pg_temp;

CREATE TABLE people (person_id integer primary key, name text);
INSERT INTO people (person_id, name) VALUES
 (1, 'name1')
,(2, 'name2');

CREATE TABLE roles (role_id integer, person_id integer, role_name text);
INSERT INTO roles (role_id, person_id, role_name) VALUES
 (1, 1, 'role name!')
,(2, 1, 'another role')
,(3, 2, 'role name2!')
,(4, 2, 'another role2');

CREATE TYPE role AS (
  role_id   int
 ,role_name text
);

CREATE TYPE person AS (
  person_id int
 ,name text
 ,roles role[]
);

功能:

CREATE OR REPLACE FUNCTION get_people()
    RETURNS person[]  LANGUAGE sql AS
$func$
SELECT ARRAY (
   SELECT (p.person_id, p.name
          ,array_agg((r.role_id, r.role_name)::role))::person
   FROM    people p
   JOIN    roles r USING (person_id)
   GROUP   BY p.person_id
   ORDER   BY p.person_id
   )
$func$;

调用:

SELECT get_people();

清理:

DROP SCHEMA x CASCADE;

核心特征是:

• 一个大大简化的函数，只包含一个普通的 SQL 查询。
• 您的主要错误是您从表 roles 中获取了 role_name text 并将其视为类型 role，但事实并非如此。<

我会让代码说明一切。要解释的太多了，我现在没有更多时间了。

这是非常高级的东西，我不确定您是否真的需要返回这个嵌套类型。也许有更简单的方法，比如非嵌套复杂类型的 SET？