在PostgreSQL中截断模式中的所有表[重复]

Question

我正在尝试使用 PostgreSQL 截断模式中的所有表。它显示此错误：

ERROR:  relation "Building" does not exist
CONTEXT:  SQL statement "TRUNCATE TABLE "Building" CASCADE"
PL/pgSQL function truncate_schema(character varying) line 15 at EXECUTE statement

这是我使用的函数：

CREATE OR REPLACE FUNCTION truncate_schema(schema IN VARCHAR) RETURNS void AS $$
DECLARE
    statements CURSOR FOR
        SELECT table_name FROM information_schema.tables
WHERE table_type = 'BASE TABLE' AND table_schema = schema;
BEGIN
    FOR stmt IN statements LOOP
        EXECUTE 'TRUNCATE TABLE ' || quote_ident(stmt.table_name) || ' CASCADE';
    END LOOP;
END;
$$ LANGUAGE plpgsql;

如何正确地做到这一点？

Answer 1

这可能是因为您没有在 TRUNCATE 语句中包含架构名称，因此它正在寻找 public 架构中的表。

尝试将TRUNCATE 语句更改为如下内容：

EXECUTE 'TRUNCATE TABLE ' || quote_ident(stmt.table_schema) || '.' ||
    quote_ident(stmt.table_name) || ' CASCADE';

另外，关于 CASCADE 需要记住的一点是，它将 TRUNCATE 任何与该表有 外键 关系的表，这可以包括该架构之外的表。

根据 OP 的评论进行编辑：

您还需要将table_schema 添加到statements 后面的查询中，以便在EXECUTE 语句中可用。

Answer 2

这样试试

CREATE OR REPLACE FUNCTION truncate_schema(_schema character varying)
  RETURNS void AS
$BODY$
declare
    selectrow record;
begin
for selectrow in
select 'TRUNCATE TABLE ' || quote_ident(_schema) || '.' ||quote_ident(t.table_name) || ' CASCADE;' as qry 
from (
     SELECT table_name 
     FROM information_schema.tables
     WHERE table_type = 'BASE TABLE' AND table_schema = _schema
     )t
loop
execute selectrow.qry;
end loop;
end;
$BODY$
  LANGUAGE plpgsql