计算受 DELETE 影响的行数

我用这个代码来验证DELETE句子，但我相信你知道更好的方法：

CREATE OR REPLACE FUNCTION my_schema.sp_delete_row_table(table_name character varying
                                                       , id_column character varying
                                                       , id_value integer)
  RETURNS integer AS
$BODY$
  DECLARE
    BEFORE_ROWS integer;
    AFTER_ROWS integer;
  BEGIN
    EXECUTE 'SELECT count(*) FROM ' || TABLE_NAME INTO BEFORE_ROWS;
    EXECUTE 'DELETE FROM ' || TABLE_NAME || ' WHERE ' || ID_COLUMN || ' = ' || (ID_VALUE)::varchar;
    EXECUTE 'SELECT count(*) FROM ' || TABLE_NAME INTO AFTER_ROWS;

    IF BEFORE_ROWS - AFTER_ROWS = 1 THEN
      RETURN 1;
    ELSE
      RETURN 2;
    END IF;
  EXCEPTION WHEN OTHERS THEN
    RETURN 0;
  END;
$BODY$
  LANGUAGE plpgsql VOLATILE;

如何改进这段代码？我需要它在 Postgres 8.4、9.1 和 9.2 中工作。

其实，你cannot use FOUND with EXECUTE. 手册:

特别注意的是EXECUTE改变输出GET DIAGNOSTICS, 但没有改变FOUND.

还有一些其他方面可能需要改进。首先，您的原始版本对 SQL 注入是开放的。我建议：

CREATE OR REPLACE FUNCTION my_schema.sp_delete_row_table(table_name regclass
                                                       , id_column  text
                                                       , id_value   int
                                                       , OUT del_ct int) AS
$func$
BEGIN
   EXECUTE format ('DELETE FROM %s WHERE %I = $1', table_name, id_column);
   USING id_value;                     -- assuming integer columns

   GET DIAGNOSTICS del_ct = ROW_COUNT; -- directly assign OUT parameter

EXCEPTION WHEN OTHERS THEN
   del_ct := 0;
END
$func$  LANGUAGE plpgsql;

format()需要 Postgres 9.1 或更高版本。您可以将其替换为字符串连接，但请务必使用正确的转义列名quote_ident()!
其余的也适用于 8.4。

密切相关的答案：

• 动态 SQL (EXECUTE) 作为 IF 语句的条件
• 表名作为 PostgreSQL 函数参数