3

我是在 postgresql 和一般情况下编写存储函数的新手。我正在尝试使用输入参数编写 onw 并返回存储在临时表中的一组结果。我在我的函数中执行以下操作。1) 获取所有消费者的列表并将他们的 id 存储在临时表中。2) 遍历特定表并从上述列表中检索与每个值对应的值并存储在临时表中。3）返回临时表。

这是我自己尝试编写的函数，

create or replace function getPumps(status varchar) returns setof record as $$    (setof record?) 
DECLARE 
cons_id integer[]; 
i integer; 
temp table tmp_table;--Point B 
BEGIN 
select consumer_id into cons_id  from db_consumer_pump_details; 
 FOR i in select * from cons_id LOOP 
    select objectid,pump_id,pump_serial_id,repdate,pumpmake,db_consumer_pump_details.status,db_consumer.consumer_name,db_consumer.wenexa_id,db_consumer.rr_no into tmp_table  from db_consumer_pump_details inner join db_consumer on db_consumer.consumer_id=db_consumer_pump_details.consumer_id 
where db_consumer_pump_details.consumer_id=i and db_consumer_pump_details.status=$1--Point A 
order by db_consumer_pump_details.consumer_id,pump_id,createddate desc limit 2 
END LOOP; 
return tmp_table   
END; 
$$ 
LANGUAGE plpgsql;

但是我不确定我的方法以及我是否在上面代码中标记的 A 点和 B 点。并且在尝试创建临时表时遇到大量错误。

编辑：使该功能正常工作，但是当我尝试运行该功能时出现以下错误。

   ERROR:  array value must start with "{" or dimension information

这是我修改后的功能。

 create temp table tmp_table(objectid integer,pump_id integer,pump_serial_id varchar(50),repdate timestamp with time zone,pumpmake varchar(50),status varchar(2),consumer_name varchar(50),wenexa_id varchar(50),rr_no varchar(25));

  select consumer_id into cons_id  from db_consumer_pump_details;
   FOR i in select * from cons_id LOOP
insert into tmp_table 
select objectid,pump_id,pump_serial_id,repdate,pumpmake,db_consumer_pump_details.status,db_consumer.consumer_name,db_consumer.wenexa_id,db_consumer.rr_no   from db_consumer_pump_details inner join db_consumer on db_consumer.consumer_id=db_consumer_pump_details.consumer_id where db_consumer_pump_details.consumer_id=i and db_consumer_pump_details.status=$1
order by db_consumer_pump_details.consumer_id,pump_id,createddate desc limit 2;
 END LOOP;
 return query (select * from tmp_table);
 drop table tmp_table;
  END;
  $$
  LANGUAGE plpgsql;
4

2 回答 2

5

AFAIK 不能将表声明为 postgres 中的变量。你可以做的是在你的函数体中创建一个并彻底使用它（甚至在函数之外）。但请注意，直到会话或提交结束时才会删除临时表。

要走的路是使用RETURN NEXT 或 RETURN QUERY

至于函数结果类型，我总是发现 RETURNS TABLE 更具可读性。

编辑： 您的 cons_id 数组是不必要的，只需迭代 select 返回的值。您还可以在单​​个函数中有多个返回查询语句，以将查询结果附加到函数返回的结果中。

在你的情况下：

CREATE OR REPLACE FUNCTION getPumps(status varchar) 
RETURNS TABLE (objectid INTEGER,pump_id INTEGER,pump_serial_id INTEGER....)   
AS 
$$
BEGIN 
    FOR i in SELECT consumer_id FROM db_consumer_pump_details LOOP

    RETURN QUERY(
        SELECT objectid,pump_id,pump_serial_id,repdate,pumpmake,db_consumer_pump_details.status,db_consumer.consumer_name,db_consumer.wenexa_id,db_consumer.rr_no FROM db_consumer_pump_details INNER JOIN db_consumer ON db_consumer.consumer_id=db_consumer_pump_details.consumer_id 
        WHERE db_consumer_pump_details.consumer_id=i AND db_consumer_pump_details.status=$1
        ORDER BY db_consumer_pump_details.consumer_id,pump_id,createddate DESC LIMIT 2 
    );
    END LOOP;
END;
$$

编辑2：

你可能想看看这个groupwise-k-maximum 问题的解决方案，因为这正是你在这里处理的。

于 2011-11-28T13:22:33.840 回答
0

只返回一个表（或查询）可能更容易

CREATE FUNCTION extended_sales(p_itemno int)
RETURNS TABLE(quantity int, total numeric) AS $$
BEGIN
  RETURN QUERY SELECT quantity, quantity * price FROM sales
             WHERE itemno = p_itemno;
END;
$$ LANGUAGE plpgsql;

（从postgresql 文档复制）

于 2011-11-28T13:22:13.923 回答