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

布满荆棘的人生 2022-04-24 06:26 1622阅读 0赞

## 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 行记录)

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

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';

执行函数。