1. 函数参数的别名: 传递给函数的参数都是用$1、$2这样的标识符来表示的。为了增加可读性,我们可以为其声明别名。之后别名和数字标识符均可指向该参数值,见如下示例: 1). 在函数声明的同时给出参数变量名。 CREATE FUNCTION sales_tax(subtotal real) RETURNS real AS $$ BEGIN RETURN subtotal * 0.06; END; $$ LANGUAGE plpgsql; 2). 在声明段中为参数变量定义别名。 CREATE FUNCTION sales_tax(REAL) RETURNS real AS $$ DECLARE subtotal ALIAS FOR $1; BEGIN RETURN subtotal * 0.06; END; $$ LANGUAGE plpgsql; 3). 对于输出参数而言,我们仍然可以遵守1)和2)中的规则。 CREATE FUNCTION sales_tax(subtotal real, OUT tax real) AS $$ BEGIN tax := subtotal * 0.06; END; $$ LANGUAGE plpgsql; 4). 如果PL/pgSQL函数的返回类型为多态类型(anyelement或anyarray),那么函数就会创建一个特殊的参数:$0。我们仍然可以为该变量设置别名。 CREATE FUNCTION add_three_values(v1 anyelement, v2 anyelement, v3 anyelement) RETURNS anyelement AS $$ DECLARE result ALIAS FOR $0; BEGIN result := v1 + v2 + v3; RETURN result; END; $$ LANGUAGE plpgsql;
3. 行类型: 见如下形式的变量声明: name table_name%ROWTYPE; name composite_type_name;
table_name%ROWTYPE表示指定表的行类型,我们在创建一个表的时候,PostgreSQL也会随之创建出一个与之相应的复合类型,该类
型名等同于表名,因此,我们可以通过以上两种方式来声明行类型的变量。由此方式声明的变量,可以保存SELECT返回结果中的一行。如果要访问变量中的某
个域字段,可以使用点表示法,如rowvar.field,但是行类型的变量只能访问自定义字段,无法访问系统提供的隐含字段,如OID等。对于函数的参
数,我们只能使用复合类型标识变量的数据类型。最后需要说明的是,推荐使用%ROWTYPE的声明方式,这样可以具有更好的可移植性,因为在Oracle
的PL/SQL中也存在相同的概念,其声明方式也为%ROWTYPE。见如下示例: CREATE FUNCTION merge_fields(t_row table1) RETURNS text AS $$ DECLARE t2_row table2%ROWTYPE; BEGIN SELECT * INTO t2_row FROM table2 WHERE id = 1 limit 1; RETURN t_row.f1 || t2_row.f3 || t_row.f5 || t2_row.f7; END; $$ LANGUAGE plpgsql;
4. 记录类型: 见如下形式的变量声明: name RECORD; 记录变量类似于行类型变量,但是它们没有预定义的结构,只能通过SELECT或FOR命令来获取实际的行结构,因此记录变量在被初始化之前无法访问,否则将引发运行时错误。 注:RECORD不是真正的数据类型,只是一个占位符。
2. SELECT INTO: 通过该语句可以为记录变量或行类型变量进行赋值,其表现形式为:SELECT INTO target select_expressions FROM ...,该赋值方式一次只能赋值一个变量。表达式中的target可以表示为是一个记录变量、行变量,或者是一组用逗号分隔的简单变量和记录/行字段的列表。select_expressions以及剩余部分和普通SQL一样。
如果将一行或者一个变量列表用做目标,那么选出的数值必需精确匹配目标的结构,否则就会产生运行时错误。如果目标是一个记录变量,那么它自动将自己构造
成命令结果列的行类型。如果命令返回零行,目标被赋予空值。如果命令返回多行,那么将只有第一行被赋予目标,其它行将被忽略。在执行SELECT
INTO语句之后,可以通过检查内置变量FOUND来判断本次赋值是否成功,如: SELECT INTO myrec * FROM emp WHERE empname = myname; IF NOT FOUND THEN RAISE EXCEPTION 'employee % not found', myname; END IF; 要测试一个记录/行结果是否为空,可以使用IS NULL条件进行判断,但是对于返回多条记录的情况则无法判断,如: DECLARE users_rec RECORD; BEGIN SELECT INTO users_rec * FROM users WHERE user_id = 3; IF users_rec.homepage IS NULL THEN RETURN 'http://'; END IF; END;
2. 条件: 在PL/pgSQL中有以下三种形式的条件语句。 1). IF-THEN IF boolean-expression THEN statements END IF; 2). IF-THEN-ELSE IF boolean-expression THEN statements ELSE statements END IF; 3). IF-THEN-ELSIF-ELSE IF boolean-expression THEN statements ELSIF boolean-expression THEN statements ELSIF boolean-expression THEN statements ELSE statements END IF; 关于条件语句,这里就不在做过多的赘述了。
3. 循环: 1). LOOP LOOP statements END LOOP [ label ]; LOOP定义一个无条件的循环,直到由EXIT或者RETURN语句终止。可选的label可以由EXIT和CONTINUE语句使用,用于在嵌套循环中声明应该应用于哪一层循环。 2). EXIT EXIT [ label ] [ WHEN expression ]; 如果没有给出label,就退出最内层的循环,然后执行跟在END LOOP后面的语句。如果给出label,它必须是当前或更高层的嵌套循环块或语句块的标签。之后该命名块或循环就会终止,而控制则直接转到对应循环/块的END语句后面的语句上。 如果声明了WHEN,EXIT命令只有在expression为真时才被执行,否则将直接执行EXIT后面的语句。见如下示例: LOOP -- do something EXIT WHEN count > 0; END LOOP; 3). CONTINUE CONTINUE [ label ] [ WHEN expression ];
如果没有给出label,CONTINUE就会跳到最内层循环的开始处,重新进行判断,以决定是否继续执行循环内的语句。如果指定label,则跳到该
label所在的循环开始处。如果声明了WHEN,CONTINUE命令只有在expression为真时才被执行,否则将直接执行CONTINUE后面
的语句。见如下示例: LOOP -- do something EXIT WHEN count > 100; CONTINUE WHEN count < 50; END LOOP; 4). WHILE [ < WHILE expression LOOP statements END LOOP [ label ]; 只要条件表达式为真,其块内的语句就会被循环执行。条件是在每次进入循环体时进行判断的。见如下示例: WHILE amount_owed > 0 AND gift_certificate_balance > 0 LOOP --do something END LOOP; 5). FOR [ < FOR name IN [ REVERSE ] expression .. expression LOOP statements END LOOP [ label ]; 变量name自动被定义为integer类型,其作用域仅为FOR循环的块内。表示范围上下界的两个表达式只在进入循环时计算一次。每次迭代name值自增1,但如果声明了REVERSE,name变量在每次迭代中将自减1,见如下示例: FOR i IN 1..10 LOOP --do something RAISE NOTICE 'i IS %', i; END LOOP;
FOR i IN REVERSE 10..1 LOOP --do something END LOOP;
4. 遍历命令结果: [ < FOR record_or_row IN query LOOP statements END LOOP [ label ]; 这是另外一种形式的FOR循环,在该循环中可以遍历命令的结果并操作相应的数据,见如下示例: FOR rec IN SELECT * FROM some_table LOOP PERFORM. some_func(rec.one_col); END LOOP; PL/pgSQL还提供了另外一种遍历命令结果的方式,和上面的方式相比,唯一的差别是该方式将SELECT语句存于字符串文本中,然后再交由EXECUTE命令动态的执行。和前一种方式相比,该方式的灵活性更高,但是效率较低。 [ < FOR record_or_row IN EXECUTE text_expression LOOP statements END LOOP [ label ];
5. 异常捕获: 在PL/pgSQL函数中,如果没有异常捕获,函数会在发生错误时直接退出,与其相关的事物也会随之回滚。我们可以通过使用带有EXCEPTION子句的BEGIN块来捕获异常并使其从中恢复。见如下声明形式: [ < [ DECLARE declarations ] BEGIN statements EXCEPTION WHEN condition [ OR condition ... ] THEN handler_statements WHEN condition [ OR condition ... ] THEN handler_statements END;
如果没有错误发生,只有BEGIN块中的statements会被正常执行,然而一旦这些语句中有任意一条发生错误,其后的语句都将被跳过,直接跳转到
EXCEPTION块的开始处。此时系统将搜索异常条件列表,寻找匹配该异常的第一个条件,如果找到匹配,则执行相应的
handler_statements,之后再执行END的下一条语句。如果没有找到匹配,该错误就会被继续向外抛出,其结果与没有EXCEPTION子
句完全等同。如果此时handler_statements中的语句发生新错误,它将不能被该EXCEPTION子句捕获,而是继续向外传播,交由其外层
的EXCEPTION子句捕获并处理。见如下示例: INSERT INTO mytab(firstname, lastname) VALUES('Tom', 'Jones'); BEGIN UPDATE mytab SET firstname = 'Joe' WHERE lastname = 'Jones'; x := x + 1; y := x / 0; EXCEPTION WHEN division_by_zero THEN RAISE NOTICE 'caught division_by_zero'; RETURN x; END;
当以上函数执行到y := x /
0语句时,将会引发一个异常错误,代码将跳转到EXCEPTION块的开始处,之后系统会寻找匹配的异常捕捉条件,此时division_by_zero
完全匹配,这样该条件内的代码将会被继续执行。需要说明的是,RETURN语句中返回的x值为x := x +
1执行后的新值,但是在除零之前的update语句将会被回滚,BEGIN之前的insert语句将仍然生效。
六、游标:
1. 声明游标变量: 在PL/pgSQL中对游标的访问都是通过游标变量实现的,其数据类型为refcursor。 创建游标变量的方法有以下两种: 1). 和声明其他类型的变量一样,直接声明一个游标类型的变量即可。 2). 使用游标专有的声明语法,如: name CURSOR [ ( arguments ) ] FOR query; 其中arguments为一组逗号分隔的name datatype列表,见如下示例: curs1 refcursor; curs2 CURSOR FOR SELECT * FROM tenk1; curs3 CURSOR (key integer) IS SELECT * FROM tenk1 WHERE unique1 = key; 在上面三个例子中,只有第一个是未绑定游标,剩下两个游标均已被绑定。
2. 打开游标: 游标在使用之前必须先被打开,在PL/pgSQL中有三种形式的OPEN语句,其中两种用于未绑定的游标变量,另外一种用于绑定的游标变量。 1). OPEN FOR: 其声明形式为: OPEN unbound_cursor FOR query; 该形式只能用于未绑定的游标变量,其查询语句必须是SELECT,或其他返回记录行的语句,如EXPLAIN。在PostgreSQL中,该查询和普通的SQL命令平等对待,即先替换变量名,同时也将该查询的执行计划缓存起来,以供后用。见如下示例: OPEN curs1 FOR SELECT * FROM foo WHERE key = mykey; 2). OPEN FOR EXECUTE 其声明形式为: OPEN unbound_cursor FOREXECUTE query-string; 和上面的形式一样,该形式也仅适用于未绑定的游标变量。EXECUTE将动态执行其后以文本形式表示的查询字符串。 OPEN curs1 FOR EXECUTE 'SELECT * FROM ' || quote_ident($1); 3). 打开一个绑定的游标 其声明形式为: OPEN bound_cursor [ ( argument_values ) ]; 该形式仅适用于绑定的游标变量,只有当该变量在声明时包含接收参数,才能以传递参数的形式打开该游标,这些参数将被实际代入到游标声明的查询语句中,见如下示例: OPEN curs2; OPEN curs3(42);
3. 使用游标: 游标一旦打开,就可以按照以下方式进行读取。然而需要说明的是,游标的打开和读取必须在同一个事物内,因为在PostgreSQL中,如果事物结束,事物内打开的游标将会被隐含的关闭。 1). FETCH 其声明形式为: FETCH cursor INTO target; FETCH命令从游标中读取下一行记录的数据到目标中,其中目标可以是行变量、记录变量,或者是一组逗号分隔的普通变量的列表,读取成功与否,可通过PL/pgSQL内置变量FOUND来判断,其规则等同于SELECT INTO。见如下示例: FETCH curs1 INTO rowvar; --rowvar为行变量 FETCH curs2 INTO foo, bar, baz; 2). CLOSE 其声明形式为: CLOSE cursor; 关闭当前已经打开的游标,以释放其占有的系统资源,见如下示例: CLOSE curs1;