基本结构
create or replace function somefunc()
returns varchar as
$$
declare
name varchar := 'wangzhen';
begin
return name;
end
$$ language plpgsql ;
$$作用
- 函数代码在function中实际上为一个字符串,代码1和代码2等价,但代码中字符串的单引号需要写两个进行转义。
-- 代码1
create or replace function somefunc()
returns varchar as
$$
declare
name varchar := 'wangzhen';
begin
return name;
end
$$ language plpgsql ;
--代码2
create or replace function somefunc()
returns varchar as
'
declare
name varchar := ''wangzhen'';
begin
return name;
end
' language plpgsql ;
- “美元符引用”书写字符串常量,使单引号、反斜线、$符等按照字面值进行解释,不需要写两个或在4个或着更多进行转义,代码3和代码4等价。
--代码3
create or replace function somefunc()
returns varchar as
$$
declare
name varchar := 'wangzhen''blog';
begin
return name;
end
$$ language plpgsql ;
--代码4
create or replace function somefunc()
returns varchar as
$$
declare
name varchar := $tag$wangzhen'blog$tag$;
begin
return name;
end
$$ language plpgsql ;
- $中间可以包含可选的标签,但是标签要成对出现,且大小写敏感。代码5和代码6和代码7等价。
--代码5
create or replace function somefunc()
returns varchar as
$body$
declare
name varchar := $tag$wangzhen'blog$tag$;
begin
return name;
end
$body$ language plpgsql ;
--代码6
create or replace function somefunc()
returns varchar as
$func$
declare
name varchar := $$wangzhen'blog$$;
begin
return name;
end
$func$ language plpgsql ;
--代码7
create or replace function somefunc()
returns varchar as
$$
declare
name varchar := 'wangzhen''blog';
begin
return name;
end
$$ language plpgsql ;
代码块
- 块名可选,代码8和9等价
--代码8
create or replace function somefunc()
returns varchar as
$$
<<outblock>>
declare
name varchar := 'wangzhen';
begin
return outblock.name;
end;
$$ language plpgsql;
--代码9
create or replace function somefunc()
returns varchar as
$$
declare
name varchar := 'wangzhen';
begin
return name;
end;
$$ language plpgsql;
- 块可嵌套,内层同名变量覆盖外层同名变量,可通过指定块名引用。
--代码10
create or replace function somefunc()
returns varchar as
$$
<<outblock>>
declare
name varchar := 'wangzhen';
begin
<<innerblock>>
declare
name varchar := 'xiaozhang';
begin
outblock.name = innerblock.name;
end;
return outblock.name;
end;
$$ language plpgsql;
abase=# select somefunc();
somefunc
-----------
xiaozhang
- 函数体是一个隐藏的块,块名是函数名。
--代码11
create or replace function somefunc(name varchar)
returns varchar as
$$
<<outblock>>
declare
name varchar := somefunc.name;
begin
return outblock.name;
end;
$$ language plpgsql;
abase=# select somefunc('xiaoli');
somefunc
----------
xiaoli
(1 row)
命名参数
- 在函数声明时直接命名参数
--代码12
create or replace function somefunc(newname varchar)
returns varchar as
$$
declare
name varchar := newname;
begin
return name;
end;
$$ language plpgsql;
- 使用别名和美元符号
--代码13
create or replace function somefunc(varchar,varchar)
returns varchar as
$$
declare
newname alias for $1;
name varchar := newname;
name2 varchar := $2;
begin
return name;
end;
$$ language plpgsql;
执行SQL
- 没有返回结果的SQL可以直接执行
--代码14
create or replace function somefunc2()
returns void as
$$
begin
create table test1 (n_id integer,c_mc varchar(300));
insert into test1 (n_id,c_mc) values (1,'wangzhen');
end;
$$ language plpgsql;
- 不需要返回结果的select可以通过perform执行。perform替换select
--代码15
create or replace function somefunc3()
returns void as
$$
begin
perform * from test1;
end;
$$ language plpgsql;
--代码16
create or replace function somefunc3()
returns void as
$$
begin
perform somefunc();
--perform 怎么处理with和其他复杂查询?
end;
$$ language plpgsql;
- 执行动态拼接SQL
--代码17
create or replace function somefunc3()
returns void as
$$
declare
var_sql varchar := 'insert into test1 values (0,'||quote_literal('admin')||')';
begin
execute var_sql;
execute format('insert into %I values (%L,%L)','test1',2,'xiaoniu');
end;
$$ language plpgsql;
返回结果
返回结果分为返回标量(单行)和返回结果集合(多行)两种情况。
- 返回标量
-- 代码18
-- 返回单行int
create or replace function somefunc4()
returns int as
$$
begin
return 100;
end
$$ language plpgsql;
-- 代码19
-- 通过out参数返回
create or replace function somefunc4(out v1 integer)
as
$$
begin
v1 := 100;
end
$$ language plpgsql;
-- 代码20
-- 返回单行record 复合类型
create or replace function somefunc4(id integer,mc varchar)
returns record as
$$
declare
r record;
begin
r := row(id,mc);
return r;
end
$$ language plpgsql;
- SETOF 作用
返回结果集合(多行)时,需要使用SETOF指定
--代码21
--返回单列多行
create or replace function somefunc5()
returns setof int as
$$
declare
id int;
begin
for id in select n_id from test1
loop
return next id;
end loop;
return;
end
$$ language plpgsql;
结果:
abase=# select somefunc5();
somefunc5
-----------
1
1
0
0
2
(5 rows)
- 返回结果集
-- 代码22
-- 通过out参数返回
-- returns setof record 可以省略
create or replace function somefunc6(out id integer,out name varchar)
returns setof record as
$$
declare
r record;
begin
for r in select n_id,c_mc from test1
loop
id := r.n_id;
name := r.c_mc;
return next;
end loop;
end
$$ language plpgsql;
结果:
abase=# select * from somefunc6() ;
id | name
----+-----------
1 | wangzhen
1 | xiaozhang
0 | admin
0 | admin
2 | xiaoniu
(5 rows)
-- 代码23
-- 通过return next 返回自定义类型
create type test_rs as (id int,mc varchar);
create or replace function somefunc7()
returns setof test_rs as
$$
declare
r test_rs%rowtype;
begin
for r in select n_id,c_mc from test1
loop
return next r;
end loop;
return;
end
$$ language plpgsql;
-- 代码24
-- 通过return query 返回自定义类型
create or replace function somefunc7()
returns setof test_rs as
$$
declare
r test_rs%rowtype;
begin
return query select n_id,c_mc from test1;
return;
end
$$ language plpgsql;
-- 代码25
-- 通过return next返回表类型
create or replace function somefunc7()
returns setof test1 as
$$
declare
r test1%rowtype;
begin
for r in select n_id,c_mc from test1
loop
return next r;
end loop;
return;
end
$$ language plpgsql;
-- 代码26
-- 通过拼接sql return next 返回自定义类型
create or replace function somefunc7()
returns setof test_rs as
$$
declare
r test_rs%rowtype;
begin
for r in execute 'select n_id,c_mc from test1'
loop
return next r;
end loop;
return;
end
$$ language plpgsql;
-- 代码26
-- 通过拼接sql return query 返回自定义类型
create or replace function somefunc7()
returns setof test_rs as
$$
declare
r test_rs%rowtype;
begin
return query execute 'select n_id,c_mc from test1';
return;
end
$$ language plpgsql;
- 返回结果集时PostgreSQL 9.5.4,需要使用自定义类型或表类型制定返回类型,不能用record匿名类型。
循环判断
- LOOP循环
create or replace function testloop1()
returns void as
$$
declare
count int :=0;
begin
loop
exit when count >=100;
count := count + 1;
raise notice 'count is %',count;
end loop;
end
$$ language plpgsql ;
- WHILE循环
create or replace function testloop2()
returns void as
$$
declare
count int :=0;
begin
while count <100 loop
count := count + 1;
raise notice 'count is %',count;
end loop;
end
$$ language plpgsql ;
- FOR循环
create or replace function testloop3()
returns void as
$$
begin
for i in 1..100 loop
raise notice 'count is %',i;
end loop;
end
$$ language plpgsql;
create or replace function testloop4()
returns void as
$$
begin
for i in reverse 100..1 by 2 loop
raise notice 'count is %',i;
end loop;
end
$$ language plpgsql;
- FOREACH循环
create or replace function testloop5()
returns void as
$$
declare
v_arr int[]:= array[1,2,3,4,5];
i int;
begin
foreach i in array v_arr loop
raise notice 'count is %',i;
end loop;
end
$$ language plpgsql;