Postgresql自定义函数—表名作为函数参数

• 函数传递表名和另外一个参数示例

CREATE OR REPLACE FUNCTION some_f(_tbl text, col text, OUT result bool)

LANGUAGE plpgsql AS

func

BEGIN

EXECUTE format('SELECT (EXISTS (SELECT FROM %s WHERE b = $1))', _tbl,col) using col INTO result;

END

func;

函数调用：

test=# select some_f('test','test'); --这里只传递了2个参数

some_f

---

t

(1 row)

使用using子句给format函数传递参数($1)，这里是数字'壹'。%s的含义，见第四部分说明

• 表名作为参数，返回表中记录数

CREATE OR REPLACE FUNCTION get_table_info(tablenames text)

RETURNS int AS $$

DECLARE

rowcount int;

BEGIN

EXECUTE format('SELECT count(*) FROM %I', tablenames) INTO rowcount;

return rowcount;

END

$$ LANGUAGE plpgsql;

函数调用：

test=# select get_table_info('test');

get_table_info

---

2

(1 row)

• 表名作为参数，返回表中记录值

CREATE OR REPLACE FUNCTION get_table_info(tablenames text)

RETURNS table(a int,b varchar) AS $$

DECLARE

rowcount int;

BEGIN

return query EXECUTE format('SELECT * FROM %I', tablenames);

END

$$ LANGUAGE plpgsql;

函数调用：

test=# select get_table_info('test');

get_table_info

---

(10,test)

(1,testtest)

这种调用方式，返回表的数据，作为元组形式

或者

test=# select * from get_table_info('test');

a | b

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

10 | test

1 | testtest

(2 rows)

这种调用方式，返回表的数据同正常查询结果的返回

• 几个注意问题

1. Out参数，在调用时，不需要在调用时传递
2. 当需要返回table类型时，可以使用return query或者return next
3. Format函数参数的说明：

%I is equivalent to quote_ident, and %L is equivalent to quote_nullable. The format function can be used in conjunction with the USING clause:

EXECUTE format('UPDATE tbl SET %I = 1 WHERE key = 2', colname) USING newvalue, keyvalue;