Как добавить столбец, если он не существует на PostgreSQL?

вопрос простой. Как добавить столбец x в таблице y, но только тогда, когда

89
postgresql postgresql-9.1

9 ответов

вот короткая и сладкая версия, использующая оператор "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;
$$

вы не можете передать их в качестве параметров, вам нужно будет сделать подстановку переменных в строке на стороне клиента, но это автономный запрос, который выдает сообщение только в том случае, если столбец уже существует, добавляет, Если это не так, и будет продолжать терпеть неудачу при других ошибках (например, недопустимый тип данных).

Я не рекомендую делать ни один из этих методов, если это случайные строки, исходящие из внешний источник. Независимо от того, какой метод вы используете (динамические строки на стороне cleint или на стороне сервера, выполняемые как запросы), это будет рецепт катастрофы, поскольку он открывает вас для атак SQL-инъекций.

95

С Postgres 9.6 Это можно сделать с помощью опции if not exists

ALTER TABLE table_name ADD COLUMN IF NOT EXISTS column_name INTEGER;
142
CREATE OR REPLACE function f_add_col(_tbl regclass, _col  text, _type regtype)
  RETURNS bool AS
$func$
BEGIN
   IF EXISTS (SELECT 1 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$  LANGUAGE plpgsql;

звоните:

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

возвращает TRUE на успех, остальное FALSE (столбец уже существует).
Вызывает исключение для недопустимой таблицы или типа name.

почему другая версия?

  • это можно сделать с помощью DO утверждение, но DO операторы не могут ничего возвращать. И если это для повторного использования, я бы создал функцию.

  • я использую идентификатор объекта типы regclass и regtype на _tbl и _type что a) предотвращает впрыску SQL и B) проверяет действительность обоих немедленно (самый дешевый возможный путь). Имя столбца _col все еще должен быть дезинфицирован для EXECUTE С quote_ident(). Больше объяснений в этом связанном ответе:

  • format() требуется Postgres 9.1+. Для пожилых версии объединяются вручную:

    EXECUTE 'ALTER TABLE ' || _tbl || ' ADD COLUMN ' || quote_ident(_col) || ' ' || _type;

  • вы можете указать свое имя таблицы, но вам это не нужно.
    Вы можете дважды процитировать идентификаторы в вызове функции, чтобы сохранить Camel-case и зарезервированные слова (но вы все равно не должны использовать ничего из этого).

  • запрос pg_catalog вместо information_schema. Подробное объяснение:

  • блоки, содержащие EXCEPTION пункт как в настоящее время принимаются ответ существенно медленнее. Это, как правило, проще и быстрее. документация:

Совет: блок, содержащий EXCEPTION предложение значительно больше вход и выход дороже, чем квартал без него. Поэтому не использовать EXCEPTION без надобности.

19

после выбора запроса будет возвращен true/false, используя EXISTS()():
аргумент EXISTS является произвольным оператором SELECT, или подзапрос. Подзапрос вычисляется для определения возвращает все строки. Если он возвращает хотя бы одну строку, результатом EXISTS является "true"; если подзапрос не возвращает строк, результатом EXISTS является "ложь"

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
$$
16

функция ниже проверит столбец, если существует, вернет соответствующее сообщение, иначе он добавит столбец в таблицу. 

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;
$$
1

это в основном решение от sola, но просто немного очищено. Это достаточно отличается от того, что я не просто хотел "улучшить" его решение (плюс, я думаю, что это грубо).

основное отличие заключается в том, что он использует формат EXECUTE. Что я думаю, немного чище, но я считаю, что вы должны быть на PostgresSQL 9.1 или новее.

Это было протестировано на 9.1 и работает. Примечание: это вызовет ошибку, если схема/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)');
1

вы можете сделать это следующим образом.

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

таким образом, он отбросит столбец, если он уже существует. А затем добавьте столбец в определенную таблицу.

1

можно добавить в сценарии миграции функцию вызова и падение, когда это сделано.

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();
0

просто проверьте, вернул ли запрос column_name.

если не выполните что-то вроде этого:

ALTER TABLE x ADD COLUMN y int;

где вы помещаете что-то полезное для " x " и " y " и, конечно, подходящий тип данных, где я использовал int.

-4