PostgreSQL获取操作影响的行数

jopen 10年前

如何来获取普通的操作所影响的行数,PostgreSQL里面有一个内置的变量DIAGNOSTICS与ROW_COUNT可以做到这一点。

一、环境:
DB:9.4beta3

二、准备:

postgres=# create table test(id int);  CREATE TABLE  postgres=# insert into test select generate_series(1,20);  INSERT 0 20  postgres=# select * from test;   id   ----    1    2    3    4    5    6    7    8    9   10   11   12   13   14   15   16   17   18   19   20  (20 rows)
三、创建函数:
CREATE OR REPLACE FUNCTION fun_affect_rows()    RETURNS text AS  $BODY$    declare  v_count int;begin    insert into test values(99),(98);  GET DIAGNOSTICS v_count = ROW_COUNT;  raise notice '本次插入数据量 %', v_count;    delete from test where id < 15;  GET DIAGNOSTICS v_count = ROW_COUNT;  raise notice '本次删除数据量 %', v_count;    update test set id = 100 where id >90;  GET DIAGNOSTICS v_count = ROW_COUNT;  raise notice '本次更新数据量 %', v_count;    return '测试完毕';  end;  $BODY$    LANGUAGE plpgsql VOLATILE    COST 100;  ALTER FUNCTION fun_affect_rows()    OWNER TO postgres;
四、运行情况:
postgres=# select * from fun_affect_rows();  NOTICE:  本次插入数据量 2  NOTICE:  本次删除数据量 14  NOTICE:  本次更新数据量 2   fun_affect_rows   -----------------   测试完毕  (1 row)
如果要返回DML的结果,那就用returning就可以了

五、参考
http://www.postgresql.org/docs/9.0/static/plpgsql-statements.html

来自:http://my.oschina.net/Kenyon/blog/343426