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;

更新在回答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

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

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

我希望这有助于说明为什么我宁愿直接处理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查询。
您的关键错误是您从表role_name text中取出roles并将其视为role类型，而不是{{1}}。

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

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

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

1 个答案: