计算PostgreSQL中批处理查询影响的行数答案

【问题标题】：Calculate number of rows affected by batch query in PostgreSQL计算PostgreSQL中批处理查询影响的行数
【发布时间】：2012-04-27 20:08:00
【问题描述】：

首先，是的，我已经阅读了 DO 声明的文档 :) http://www.postgresql.org/docs/9.1/static/sql-do.html

所以我的问题：

我需要执行一些包含 UPDATE 语句的动态代码块并计算所有受影响的行数。我正在使用Ado.Net 提供者。

在 Oracle 中，解决方案有 4 个步骤：

在命令中添加 InputOutput 参数“N”
将BEGIN ... END;添加到命令
在每条语句后添加 :N := :N + sql%rowcount。
完成了！执行后，我们可以从命令中读取N个参数。

我怎样才能用 PostgreSQL 做到这一点？我正在使用 npgsql 提供程序，但如果有帮助可以迁移到 devard。

【问题讨论】：

标签： postgresql plpgsql dynamic-sql

【解决方案1】：

DO 语句块非常适合执行动态 SQL。它们不利于返回值。为此使用plpgsql function。

你需要的关键语句是：

GET DIAGNOSTICS integer_var = ROW_COUNT;

Details in the manual.

示例代码：

CREATE OR REPLACE FUNCTION f_upd_some()
  RETURNS integer AS
$func$
DECLARE
   ct int;
   i  int;
BEGIN
   EXECUTE 'UPDATE tbl1 ...';       -- something dynamic here
   GET DIAGNOSTICS ct = ROW_COUNT;  -- initialize with 1st count

   UPDATE tbl2 ...;                 -- nothing dynamic here 
   GET DIAGNOSTICS i = ROW_COUNT;
   ct := ct + i;                    -- add up

   RETURN ct;
END
$func$  LANGUAGE plpgsql;

呼叫：

SELECT * FROM f_upd_some();

【讨论】：

【解决方案2】：

我的解决方案很简单。在 Oracle 中，我需要使用变量来计算更新行的总和，因为 command.ExecuteNonQuery() 仅返回受批处理中最后一次 UPDATE 影响的行数。

但是，npgsql 返回由所有 UPDATE 查询更新的所有行的总和。所以我只需要调用command.ExecuteNonQuery() 就可以得到不带任何变量的结果。比使用 Oracle 容易得多。

【讨论】：