PL/pgSQL编写postgresql函数（二）--基本语句

目录

基本语句

1 赋值

赋值运算符 := 或 =

2 单一行结果返回

SELECT...INTO 语法赋值

更新操作结果返回

3 多行结果返回

方式一：使用表充当容器

方式二：使用自定义TYPE充当容器

方式三：return query进行结果整体写入容器

方式四：return setof record返回

4 无结果执行

5 执行动态命令

2.1 语法

2.2示例准备

2.3 例子

6 打印输出语法

---

基本语句

1 赋值

赋值运算符 := 或 =

语法：variable { := | = } expression;

例子：

create or replace function set_value() returns void as 

$$

declare

v_id int;

v_name varchar;

 

    begin

    v_id := 100;

raise notice 'v_id=%',v_id;

v_name = '瀚高' ;

raise notice 'v_name=%',v_name;

    end 

$$

language plpgsql

打印输出结果：

NOTICE:  00000: v_id=100

NOTICE:  00000: v_name=瀚高

2 单一行结果返回

SELECT...INTO 语法赋值

语法:   SELECT select_expressisons  INTO target  FROM ...;

SELECT  INTO target  select_expressisons FROM ...;

SELECT  select_expressisons FROM ...  INTO target ;

其中，变量target可以为基本数据变量、行变量与记录变量。

例子：

create or replace function fn_record(v_id int) returns text as 

$$

declare

v_record  RECORD;   

 

    begin

    select   id,name,age  into v_record from t_test where id=v_id ;

raise notice 'v_record=%',v_record;   --1号位

select  into v_record id,name,age   from t_test where id=v_id ;

raise notice 'v_record=%',v_record;   --2号位

select   id,name,code from t_test where id=v_id into v_record;

raise notice 'v_record=%',v_record;   --3号位

return v_record.id ||v_record.name;

    end 

$$

language plpgsql

打印输出结果：

NOTICE:  00000: v_record=(1,张三,18)   --1号位

NOTICE:  00000: v_record=(1,张三,18)   --2号位

NOTICE:  00000: v_record=(1,张三,10010-11)  --3号位

更新操作结果返回

INSERT ... RETURNING expressions INTO [STRICT] target;

UPDATE ... RETURNING expressions INTO [STRICT] target;

DELETE ... RETURNING expressions INTO [STRICT] target;

例子：

create or replace function fn_into() returns void as 

$$

declare

v_record  RECORD;   

    begin

     insert into t_test(name,age,code) values ('瀚高1',15,'10010-01')

      returning name,age,code into  v_record;

raise notice 'insert--v_record=%',v_record;

 update  t_test set name='瀚高2' where name='张三'

  returning name,age,code into  v_record;

raise notice 'update--v_record=%',v_record;

 delete from t_test where name='李四'

  returning name,age,code into  v_record;

raise notice 'delete--v_record=%',v_record;

    end 

$$

language plpgsql

 

   打印执行结果：

NOTICE:  00000: insert--v_record=(瀚高1,15,10010-01)    插入的数据

NOTICE:  00000: update--v_record=(瀚高2,18,10010-11)   更新后的数据

NOTICE:  00000: delete--v_record=(李四,20,10010-12)     被删除的数据

 

3 多行结果返回

关键字  setof

例子：

create table t_test(id serial,name varchar,age int,code text);

insert into t_test(name,age,code) values('张三',18,'10010-11'),

('李四',20,'10010-12'),

('王五',22,'10010-13');

 

方式一：使用表充当容器

create table v_user as select * from t_test limit 0;       --表类型容器

create or replace function fn_get_values() returns setof v_user as 

$$  

declare

var_user v_user%rowtype;

    begin

    for var_user in select * from t_test where id != 1 loop

     return next var_user;

end loop;

return ;

    end 

$$

language plpgsql

方式二：使用自定义TYPE充当容器

create type v_users as (id int,name varchar,code text);   --自定义类型容器

create or replace function fn_get_value() returns setof v_users as 

$$  

declare

var_user v_users%rowtype;

    begin

    for var_user in select id,name,code from t_test where id != 1 loop

     return next var_user;

end loop;

return ;

    end 

$$

language plpgsql

方式三：return query进行结果整体写入容器

create unlogged table v_user as select * from t_test limit 0;  --表类型容器

create type v_user as (id int,name varchar,code text);      --自定义类型容器

create or replace function fn_manyrow() returns setof v_user as

$$

begin

return query  select * from t_test where id != 1;

end 

$$

language plpgsql

 

方式四：return setof record返回

例子

create or replace function fn_manyrow1()

RETURNS SETOF RECORD as $$

declare

    r record;

begin

    return query(select id,name,age from t_test);

end;

$$language plpgsql;

调用

select * from fn_manyrow1() as t(id int,name varchar ,age int);

调用函数执行结果

test_db=# select * from fn_manyrow1() as t(id int,name varchar ,age int);

 id | name  | age

----+-------+-----

  3 | 王五  |  22

  4 | 瀚高1 |  15

  1 | 瀚高2 |  18

(3 行记录)

 

     注意使用record返回时函数调用，不在是之前的select * from XXX,而是 select * from XXX as(col1 datatype1,col2 datatype2)这种。我们之前讲过，record没有固定结构，所以我们需要在查询时指定record结果以一定格式展示。

要求：as 后字段类型顺序与record中存放的结果数据类型顺序一致。

4 无结果执行

如果不需要接收执行结果，使用PERFORM 替换select。否则有执行结果返回会报错。

例子：

create or replace function fn_perform() returns void as 

$$   

    begin

    select current_time;

    end 

$$

language plpgsql

报错如下：原因是select current_time; 有返回值，但没有使用变量接收。

test_db=# select fn_perform();

ERROR:  42601: query has no destination for result data

提示:  If you want to discard the results of a SELECT, use PERFORM instead.

背景:  PL/pgSQL function fn_perform() line 4 at SQL statement

上述例子修改如下：使用perform 替换select

create or replace function fn_perform() returns void as 

$$

    begin

    perform current_time;

    end 

$$

language plpgsql

 

 

5 执行动态命令

2.1 语法

EXECUTE command-string [ INTO [STRICT] target ] [ USING expression [, ... ] ];

关键字：EXECUTE

command-string 执行命令串 （SQL语句也是一条命令）

USING 表达式提供插入到该命令中的值

 

2.2示例准备

create table t_user(

id varchar default sys_guid() ,name varchar,

age int,code text)

partition by list(substring(code,1,2));

create table t_user_10 partition of t_user for values in ('10');

create table t_user_11 partition of t_user for values in ('11');

create table t_user_12 partition of t_user for values in ('12');

alter table  t_user_10 add constraint pk_t_user_10_id primary key(id);

alter table  t_user_11 add constraint pk_t_user_11_id primary key(id);

alter table  t_user_12 add constraint pk_t_user_12_id primary key(id);

insert into t_user(name,age,code) values('张三',18,'10010-11');

insert into t_user(name,age,code) values('李四',18,'11010-11');

insert into t_user(name,age,code) values('王五',18,'12010-11');

 

2.3 例子

create or replace function fn_execute(partition_code varchar,uid varchar) returns text as 

$$

    declare

var_name text;

var_sql text;

    begin

raise notice '=====%','select *  from t_user_'||partition_code||' where id = '''||uid||''';';   

    var_sql := 'select name  from t_user_'||partition_code||' where id = '''||uid||''';';

EXECUTE var_sql into var_name;

       return var_name;

    end 

$$

language plpgsql

     

  在上面例子准备中我们知道有一张用户基表，它有三张分区子表，分区条件是code字段值得前两位。

需求：根据分区标识值及数据主键快速定位某条数据。

如右函数：

       输入参数为分区数值parttion_code和数据主键值uid。由于我们创建分区表时表名是有规律的，每个分区子表表名就是基表表名加上了分区值，所以编写一个统一函数进行动态SQL执行。

       注意：使用EXECUTE时，如果要与INTO写值连用，语法为 EXECUTE sql_string INTO var_value

6 打印输出语法

语法：   RAISE level 'format' [, expression [, ...]];

RAISE  是报告信息和抛出错误的语句

level   表示消息级别（debug、log、info、notice、warning、exception）

‘format’  消息内容使用’’引起来，format是消息内容，其中使用%作为占位符（如果要在format中表示%本身，需写成%%）。

[, expression [, ...]]  为%占位符的变量，依次用“,”排列。

例子：

 

例子：

RAISE NOTICE 'Calling cs_create_job(%)',v_job_idr;  

--v_job_id变量的值将替换format中的%。

 

raise log   '这是个日志消息';    --输出在日志文件中

 

raise info  '这个是一个信息';    --以下几个信息打印在控制台

 

raise notice '这是个提示消息';

 

raise warning '这是个警告消息';

 

raise exception '这是个异常消息';