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

发布时间：2020-12-13 18:11:06 所属栏目：百科来源：网络整理

导读：我试图将这种格式的嵌套数据从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描述

我试图将这种格式的嵌套数据从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,'another role');

CREATE TYPE role AS (
    "role_name" text
);

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

我的get_people()函数解析得很好,但是存在运行时错误.现在我收到错误：数组值必须以“{”或维度信息开头

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函数,将这样的记录集转换为嵌套数组.

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

我希望这有助于说明为什么我宁愿直接处理PostgreSQL中的嵌套数组.当然,我很乐意听到你的任何建议.

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

考虑使用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'),'name2');

CREATE TABLE roles (role_id integer,role_name text);
INSERT INTO roles (role_id,person_id,role_name) VALUES
 (1,'another role'),(3,2,'role name2!'),(4,'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文本,并将其视为类型角色,而不是.

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

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!