GreenPlum/PostgreSQL 创建函数返回数据集

Posted 2022-12-13 DATA数据猿

GreenPlum/PostgreSQL 创建函数返回数据集

一、创建表

create table dataset(
	id int,
	name varchar(25)
);

insert into dataset values(1,'A'),(2,'B'),(3,'C'),(4,'D');

二、创建函数 setof 表名

2.1 查询并返回指定表数据

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

2.2 执行函数，返回结果

select get_dataset()

2.3 对函数执行结果进行条件判断并过滤

select * from get_dataset() where id = 1;

三、创建函数 setof record

3.1 在执行函数时，通过输入表名称，查看表中数据，并依次返回每条表记录。

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

3.2 执行函数时，需要指定表类型

select * from get_datarows('dataset') as dataset(id int,name varchar(25));