如何根据postgres中的IF条件删除表格?

时间:2015-04-28 08:10:56

标签: database postgresql

我试图根据条件在启动时删除表:

IF NOT EXISTS (select * from pg_class where relname = 'mytable' and relpersistence = 'u') 
DROP TABLE IF EXISTS mytable

结果:Syntaxerror at 'IF', SQL state: 42601。为什么?如果我不允许使用IF,如何根据条件删除表?

2 个答案:

答案 0 :(得分:5)

IF无法在SQL中使用,这仅适用于PL / pgSQL。

您需要在匿名PL / pgSQL块中使用动态SQL执行此操作。类似的东西:

do
$$
declare
  l_count integer;
begin
  select count(*)
     into l_count
  from pg_class c
    join pg_namespace nsp on c.relnamespace = nsp.oid
  where c.relname = 'mytable' 
    and c.relpersistence = 'u'
    and nsp.nspname = 'public';

  if l_count = 1 then 
    execute 'drop table mytable';
  end if;

end;
$$

您可能应该扩展select语句以加入pg_namespace并在您的where条件中包含模式名称,以确保您不会意外地从错误的模式中删除表。

答案 1 :(得分:2)

a_horse_with_no_name已经接受的答案将有效但如果您希望将来对其他表使用相同的任务,则可以创建自定义函数,因此您应该创建如下函数:

create or replace function drop_table (_tbl varchar) returns void as
$$
begin
if exists(select 1
          from pg_class c
          join pg_namespace nsp on c.relnamespace = nsp.oid
          where c.relname = ''||_tbl||'' 
            and c.relpersistence = 'u'
            and nsp.nspname = 'public') then
  execute format('DROP TABLE %s',_tbl);
  raise notice 'Table %s Deleted',_tbl;
else
  raise notice 'Table %s Not Deleted',_tbl;
end if;
end;
$$
language plpgsql

只需在需要时调用此函数

select drop_table('mytable'); 

select drop_table('mytable_1')