Postgresql函数返回受影响的行数

我创建了一个函数.我定义了返回值void.但我想显示受影响的行数.我能怎么做？

CREATE OR REPLACE FUNCTION update() RETURNS void  AS 
$BODY$
BEGIN
   update test_a set name='cde' where name='abc';
   update test_b set name='mno' where name='klm';
END;
$BODY$  
LANGUAGE plpgsql VOLATILE   
COST 100; 
ALTER FUNCTION update()   
OWNER TO postgres;

解决方法

你应该研究GET DIAGNOSTICS,因为这是一个PLpgsql函数.您可能还会发现UPDATE / INSERTS的Postgres sql扩展RETURNING *很有用.

CREATE OR REPLACE FUNCTION update() RETURNS void  AS 
$BODY$
DECLARE
  a_count integer;
  b_count integer;
BEGIN
   update test_a set name='cde' where name='abc';
   GET DIAGNOSTICS a_count = ROW_COUNT;
   update test_b set name='mno' where name='klm';
   GET DIAGNOSTICS b_count = ROW_COUNT;

   RAISE NOTICE 'The rows affected by A=% and B=%',a_count,b_count ;
END;
$BODY$  
LANGUAGE plpgsql VOLATILE   
COST 100; 
ALTER FUNCTION update()   
OWNER TO postgres;

根据您可能想要实现的内容,可以使用特殊的布尔变量“FOUND”; UPDATE,INSERT和DELETE语句设置FOUND如果至少有一行受影响,则为true;如果没有行受影响,则为false.