首先是SQL的例子: create or replace function set_relation_done(integer, integer) returns void AS $$ update relations set done=TRUE where userid=$1 and friendid=$2; $$ language SQL; 这是一个没有返回值的函数,第一行和第三行是很格式化的东西,中间的才是正文,实现update的功能,可以看出,它和直接的sql语句没有太大的区别。
create or replace function get_relation_is_done(integer, integer, OUT done boolean) AS $$ select done from relations where userid=$1 and friendid=$2; $$ language SQL;
drop type tag_count_t CASCADE; create type tag_count_t AS (tag text, count integer); create or replace function get_my_tags(integer, integer) returns setof tag_count_t AS $$ select tag, use_count from friend_tags where friendid=$1 and use_count>0 order by use_count DESC LIMIT $2; $$ language SQL; 也可以构造有返回值的函数,如果是简单类型的返回值,可以很方便的放在参数的OUT修饰符后面(当然也可以returns),如果是较复杂的setof等,就要放在returns后面了。这里,不需要显式的return语句,最后一条有返回值的sql语句的结果将被作为结果返回。这时,需要保证返回值的存在和类型匹配。
然后是plpgsql语言的例子: create or replace function insert_friend_tags(id integer, to_tags text) returns void AS $$ declare to_tags_a text[]; begin to_tags_a := string_to_array(to_tags, ','); for i in array_lower(to_tags_a, 1)..array_upper(to_tags_a, 1) loop insert into friend_tags (friendid, tag) values(id, to_tags_a[i]::text); end loop; end; $$ language plpgsql;
create or replace function get_unrelation_friend_ids(integer, text) returns setof tag_ids_t as $$ declare friend_ids integer[]; num integer; un_friend_ids tag_ids_t; begin friend_ids := string_to_array($2, ','); for i in array_lower(friend_ids, 1)..array_upper(friend_ids,1) loop perform friendid from relations where userid=$1 and friendid=friend_ids[i]::integer; if found then continue; else select into un_friend_ids friend_ids[i]::integer; return next un_friend_ids; end if; end loop; return; end; $$ language plpgsql; 有返回值的时候,需要return next和return配合使用。if found可以用来检验上一条语句的执行状态,不仅针对select、perform,甚至update、insert等都可以。