185

问题很简单。如何将列添加x到表中y,但仅当x列不存在时?我在这里找到了唯一的解决方案,如何检查列是否存在。

SELECT column_name 
FROM information_schema.columns 
WHERE table_name='x' and column_name='y';
4

11 回答 11

521

对于Postgres 9.6,这可以使用选项来完成if not exists

ALTER TABLE table_name ADD COLUMN IF NOT EXISTS column_name INTEGER;
于 2016-08-02T13:33:07.760 回答
147

这是使用“DO”语句的简短版本:

DO $$ 
    BEGIN
        BEGIN
            ALTER TABLE <table_name> ADD COLUMN <column_name> <column_type>;
        EXCEPTION
            WHEN duplicate_column THEN RAISE NOTICE 'column <column_name> already exists in <table_name>.';
        END;
    END;
$$

您不能将这些作为参数传递,您需要在客户端的字符串中进行变量替换,但这是一个自包含查询,仅在列已存在时发出消息,如果不存在则添加,并且将继续因其他错误(如无效数据类型)而失败。

如果这些方法是来自外部来源的随机字符串,我不建议使用任何这些方法。无论您使用哪种方法(作为查询执行的客户端或服务器端动态字符串),这都会导致灾难,因为它会使您面临 SQL 注入攻击。

于 2012-09-26T18:59:08.023 回答
26

添加了Postgres 9.6 ALTER TABLE tbl ADD COLUMN IF NOT EXISTS column_name
所以这现在大多已经过时了。您可能会在旧版本中使用它,或者在变体中使用它来检查的不仅仅是列名。


CREATE OR REPLACE function f_add_col(_tbl regclass, _col  text, _type regtype)
  RETURNS bool
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF EXISTS (SELECT FROM pg_attribute
              WHERE  attrelid = _tbl
              AND    attname  = _col
              AND    NOT attisdropped) THEN
      RETURN false;
   ELSE
      EXECUTE format('ALTER TABLE %s ADD COLUMN %I %s', _tbl, _col, _type);
      RETURN true;
   END IF;
END
$func$;

称呼:

SELECT f_add_col('public.kat', 'pfad1', 'int');

成功返回true,否则false(列已存在)。
引发无效表或类型名称的异常。

为什么是另一个版本?

  • 这可以通过DO语句来完成,但DO语句不能返回任何内容。如果它是为了重复使用,我会创建一个函数。

  • 我使用对象标识符类型 regclassregtype用于a) 防止 SQL 注入_tbl_typeb) 立即检查两者的有效性(最便宜的方法)。列名_col仍需使用 进行EXECUTE清理quote_ident()。看:

  • format()需要 Postgres 9.1+。对于旧版本手动连接:

      EXECUTE 'ALTER TABLE ' || _tbl || ' ADD COLUMN ' || quote_ident(_col) || ' ' || _type;
    
  • 您可以对表名进行模式限定,但您不必这样做。
    您可以在函数调用中对标识符进行双引号以保留驼峰式和保留字(但无论如何您都不应该使用其中的任何一个)。

  • 我查询pg_catalog而不是information_schema. 详细解释:

  • 包含EXCEPTION子句的块要慢得多。
    这更简单,更快捷。手册:

提示

包含EXCEPTION子句的块的进入和退出比没有子句的块要昂贵得多。因此,不要在EXCEPTION没有必要的情况下使用。

于 2012-09-26T19:41:49.213 回答
20

以下选择查询将返回true/false,使用EXISTS()函数。

EXISTS()
EXISTS 的参数是任意的 SELECT 语句或子查询。评估子查询以确定它是否返回任何行。如果返回至少一行,则 EXISTS 的结果为“true”;如果子查询不返回任何行,则 EXISTS 的结果为“false”

SELECT EXISTS(SELECT  column_name 
                FROM  information_schema.columns 
               WHERE  table_schema = 'public' 
                 AND  table_name = 'x' 
                 AND  column_name = 'y'); 

并使用以下动态 SQL 语句来更改您的表

DO
$$
BEGIN
IF NOT EXISTS (SELECT column_name 
                 FROM  information_schema.columns 
                WHERE  table_schema = 'public' 
                  AND  table_name = 'x' 
                  AND  column_name = 'y') THEN
ALTER TABLE x ADD COLUMN y int DEFAULT NULL;
ELSE
RAISE NOTICE 'Already exists';
END IF;
END
$$
于 2014-12-03T08:05:14.813 回答
5

对于那些使用 Postgre 9.5+(我相信你们中的大多数人)的人来说,有一个非常简单和干净的解决方案

ALTER TABLE if exists <tablename> add if not exists <columnname> <columntype>
于 2020-02-20T01:41:23.240 回答
1

下面的函数将检查该列是否存在返回适当的消息,否则它将将该列添加到表中。

create or replace function addcol(schemaname varchar, tablename varchar, colname varchar, coltype varchar)
returns varchar 
language 'plpgsql'
as 
$$
declare 
    col_name varchar ;
begin 
      execute 'select column_name from information_schema.columns  where  table_schema = ' ||
      quote_literal(schemaname)||' and table_name='|| quote_literal(tablename) || '   and    column_name= '|| quote_literal(colname)    
      into   col_name ;   

      raise info  ' the val : % ', col_name;
      if(col_name is null ) then 
          col_name := colname;
          execute 'alter table ' ||schemaname|| '.'|| tablename || ' add column '|| colname || '  ' || coltype; 
      else
           col_name := colname ||' Already exist';
      end if;
return col_name;
end;
$$
于 2012-09-26T10:28:45.677 回答
1

这基本上是sola的解决方案,但只是清理了一下。不同的是,我不只是想“改进”他的解决方案(另外,我觉得这有点粗鲁)。

主要区别在于它使用 EXECUTE 格式。我认为这有点干净,但我相信这意味着您必须使用 PostgresSQL 9.1 或更高版本。

这已经在 9.1 上进行了测试并且可以工作。注意:如果 schema/table_name/或 data_type 无效,它将引发错误。这可以“修复”,但在许多情况下可能是正确的行为。

CREATE OR REPLACE FUNCTION add_column(schema_name TEXT, table_name TEXT, 
column_name TEXT, data_type TEXT)
RETURNS BOOLEAN
AS
$BODY$
DECLARE
  _tmp text;
BEGIN

  EXECUTE format('SELECT COLUMN_NAME FROM information_schema.columns WHERE 
    table_schema=%L
    AND table_name=%L
    AND column_name=%L', schema_name, table_name, column_name)
  INTO _tmp;

  IF _tmp IS NOT NULL THEN
    RAISE NOTICE 'Column % already exists in %.%', column_name, schema_name, table_name;
    RETURN FALSE;
  END IF;

  EXECUTE format('ALTER TABLE %I.%I ADD COLUMN %I %s;', schema_name, table_name, column_name, data_type);

  RAISE NOTICE 'Column % added to %.%', column_name, schema_name, table_name;

  RETURN TRUE;
END;
$BODY$
LANGUAGE 'plpgsql';

用法:

select add_column('public', 'foo', 'bar', 'varchar(30)');
于 2012-09-26T14:23:31.383 回答
0

可以添加到迁移脚本调用函数并在完成时删除。

create or replace function patch_column() returns void as
$$
begin
    if exists (
        select * from information_schema.columns
            where table_name='my_table'
            and column_name='missing_col'
     )
    then
        raise notice 'missing_col already exists';
    else
        alter table my_table
            add column missing_col varchar;
    end if;
end;
$$ language plpgsql;

select patch_column();

drop function if exists patch_column();
于 2014-03-12T13:57:14.647 回答
0

就我而言,由于它是如何创建的,我们的迁移脚本很难跨越不同的模式。

为了解决这个问题,我们使用了一个刚刚捕获并忽略错误的异常。这也有一个很好的副作用,即更容易查看。

但是,请注意其他解决方案有自己的优势,可能超过此解决方案:

DO $$
BEGIN
  BEGIN
    ALTER TABLE IF EXISTS bobby_tables RENAME COLUMN "dckx" TO "xkcd";
  EXCEPTION
    WHEN undefined_column THEN RAISE NOTICE 'Column was already renamed';
  END;
END $$;
于 2019-08-19T22:24:48.273 回答
-2

您可以通过以下方式做到这一点。

ALTER TABLE tableName drop column if exists columnName; 
ALTER TABLE tableName ADD COLUMN columnName character varying(8);

因此,如果该列已存在,它将删除该列。然后将该列添加到特定表中。

于 2016-03-29T04:51:47.810 回答
-4

只需检查查询是否返回了 column_name。

如果没有,请执行以下操作:

ALTER TABLE x ADD COLUMN y int;

在哪里放置对“x”和“y”有用的东西,当然还有我使用 int 的合适数据类型。

于 2012-09-26T08:23:34.483 回答