从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' ]
  ]
]

似乎可以使用复合类型.这个答案描述了如何从函数返回复合类型,但它不处理复合类型的数组.我在使用数组时遇到了一些麻烦.

这是我的表和类型:

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

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

这种表示很好,但我担心在这个结构中添加更多的嵌套字段.如果每个人也有一组工作怎么办？状态？我的转换功能必须变得更加复杂.数据的表示也将是复杂的.如果一个人有ñ角色,米作业和Ø状态,那人填补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查询.
• 你的关键错误是你role_name text从表中取出roles并将其视为不是的类型role.

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

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