PostgreSQL中自定义函数（function）返回数据集

1.准备数据

    建表

 create table city (
  cityId  int, 
  cityName varchar(20)
 );

    插入数据

 insert into city values(1,'BeiJing'),(2,'NewYork'),(3,'Hong kong'),(4,'ShaingHai');

2. 函数实例

2.1 setof  表名

    创建函数getCity()，它返回city表中所有的数据，返回值声明为 setof 表名。

create or replace function getCity() returns setof city as 
$$
begin
return query select * from city;
end;
$$
language plpgsql;

    执行函数。

mydb=# select getCity();
     getcity
-----------------
 (1,BeiJing)
 (2,NewYork)
 (3,"Hong kong")
 (4,ShaingHai)
(4 行记录)

    也可以用类似查询表的方式。

mydb=# select * from getCity();
 cityid | cityname
--------+-----------
      1 | BeiJing
      2 | NewYork
      3 | Hong kong
      4 | ShaingHai
(4 行记录)

   PostgreSQL还支持对函数执行结果进行条件判断并过滤。

mydb=# select * from getCity() where cityId > 3;
 cityid | cityname
--------+-----------
      4 | ShaingHai
(1 行记录)

2.2 setof record

    为了使函数更加通用，以解决动态返回数据集的问题。创建函数getRows(text)，将表名city作为函数参数，它返回表中所有的数据，返回值声明为 setof record。

create or replace function getRows(text) returns setof record as
$$
declare
rec record;
begin
for rec in EXECUTE 'select * from ' || $1 loop
return next rec;
end loop;
return;
end
$$
language 'plpgsql';

   执行函数。

mydb=# select * from getRows('city') as city(id int, name varchar(20));
 id |   name
----+-----------
  1 | BeiJing
  2 | NewYork
  3 | Hong kong
  4 | ShaingHai
(4 行记录)