Добавление нового значения в существующий тип ENUM

У меня есть столбец таблицы, который использует enumтип. Я хочу обновить этот enumтип, чтобы иметь дополнительное возможное значение. Я не хочу удалять любые существующие значения, просто добавьте новое значение. Какой самый простой способ сделать это?

ПРИМЕЧАНИЕ: если вы используете PostgreSQL 9.1 или более поздней версии и можете вносить изменения вне транзакции, см. Этот ответ для более простого подхода.

У меня была такая же проблема несколько дней назад и я нашел этот пост. Так что мой ответ может быть полезен для тех, кто ищет решение :)

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

-- 1. rename the enum type you want to change
alter type some_enum_type rename to _some_enum_type;
-- 2. create new type
create type some_enum_type as enum ('old', 'values', 'and', 'new', 'ones');
-- 3. rename column(s) which uses our enum type
alter table some_table rename column some_column to _some_column;
-- 4. add new column of new type
alter table some_table add some_column some_enum_type not null default 'new';
-- 5. copy values to the new column
update some_table set some_column = _some_column::text::some_enum_type;
-- 6. remove old column and type
alter table some_table drop column _some_column;
drop type _some_enum_type;

3-6 следует повторить, если имеется более 1 столбца.

PostgreSQL 9.1 представляет возможность для типов ALTER Enum:

ALTER TYPE enum_type ADD VALUE 'new_value'; -- appends to list
ALTER TYPE enum_type ADD VALUE 'new_value' BEFORE 'old_value';
ALTER TYPE enum_type ADD VALUE 'new_value' AFTER 'old_value';

Возможное решение заключается в следующем; предварительным условием является отсутствие конфликтов в используемых значениях перечисления. (например, при удалении значения enum убедитесь, что это значение больше не используется.)

-- rename the old enum
alter type my_enum rename to my_enum__;
-- create the new enum
create type my_enum as enum ('value1', 'value2', 'value3');

-- alter all you enum columns
alter table my_table
  alter column my_column type my_enum using my_column::text::my_enum;

-- drop the old enum
drop type my_enum__;

Также в этом случае порядок столбцов не будет изменен.

Если вы попадаете в ситуацию, когда вам нужно добавить enumзначения в транзакцию, например, выполнив ее при переходе с пролета на ALTER TYPEутверждение, вы получите ошибку ERROR: ALTER TYPE ... ADD cannot run inside a transaction block(см. Выпуск № 350 ), вы можете добавить такие значения pg_enumнепосредственно в качестве обходного пути ( type_egais_unitsэто имя цели enum):

INSERT INTO pg_enum (enumtypid, enumlabel, enumsortorder)
    SELECT 'type_egais_units'::regtype::oid, 'NEW_ENUM_VALUE', ( SELECT MAX(enumsortorder) + 1 FROM pg_enum WHERE enumtypid = 'type_egais_units'::regtype )

Дополнение @Dariusz 1

Для Rails 4.2.1 есть этот раздел документации:

== Транзакционные миграции

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

class ChangeEnum < ActiveRecord::Migration
  disable_ddl_transaction!

  def up
    execute "ALTER TYPE model_size ADD VALUE 'new_value'"
  end
end

Из Postgres 9.1 Документация :

ALTER TYPE name ADD VALUE new_enum_value [ { BEFORE | AFTER } existing_enum_value ]

Пример:

ALTER TYPE user_status ADD VALUE 'PROVISIONAL' AFTER 'NORMAL'

Отказ от ответственности: я не пробовал это решение, поэтому оно может не работать ;-)

Вы должны смотреть на pg_enum. Если вы хотите изменить только метку существующего ENUM, простое ОБНОВЛЕНИЕ сделает это.

Чтобы добавить новые значения ENUM:

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

Иллюстрация
У вас есть следующий набор меток:

ENUM ('enum1', 'enum2', 'enum3')

и вы хотите получить:

ENUM ('enum1', 'enum1b', 'enum2', 'enum3')

затем:

INSERT INTO pg_enum (OID, 'newenum3');
UPDATE TABLE SET enumvalue TO 'newenum3' WHERE enumvalue='enum3';
UPDATE TABLE SET enumvalue TO 'enum3' WHERE enumvalue='enum2';

затем:

UPDATE TABLE pg_enum SET name='enum1b' WHERE name='enum2' AND enumtypid=OID;

И так далее...

Кажется, я не могу оставить комментарий, поэтому просто скажу, что обновление pg_enum работает в Postgres 8.4. Для настройки наших перечислений я добавил новые значения в существующие типы перечислений с помощью:

INSERT INTO pg_enum (enumtypid, enumlabel)
  SELECT typelem, 'NEWENUM' FROM pg_type WHERE
    typname = '_ENUMNAME_WITH_LEADING_UNDERSCORE';

Это немного страшно, но имеет смысл, учитывая то, как Postgres на самом деле хранит свои данные.

Работает обновление pg_enum, как и трюк с промежуточным столбцом, выделенный выше. Можно также использовать ИСПОЛЬЗОВАНИЕ магии, чтобы напрямую изменить тип столбца:

CREATE TYPE test AS enum('a', 'b');
CREATE TABLE foo (bar test);
INSERT INTO foo VALUES ('a'), ('b');

ALTER TABLE foo ALTER COLUMN bar TYPE varchar;

DROP TYPE test;
CREATE TYPE test as enum('a', 'b', 'c');

ALTER TABLE foo ALTER COLUMN bar TYPE test
USING CASE
WHEN bar = ANY (enum_range(null::test)::varchar[])
THEN bar::test
WHEN bar = ANY ('{convert, these, values}'::varchar[])
THEN 'c'::test
ELSE NULL
END;

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

Также обратите внимание, что PG9.1 вводит оператор ALTER TYPE, который будет работать с перечислениями:

http://developer.postgresql.org/pgdocs/postgres/release-9-1-alpha.html

Самое простое: избавиться от перечислений. Их нелегко изменить, и поэтому их следует использовать очень редко.

Не удается добавить комментарий в соответствующее место, но ALTER TABLE foo ALTER COLUMN bar TYPE new_enum_type USING bar::text::new_enum_typeпо умолчанию для столбца не удалось. Мне пришлось:

ALTER table ALTER COLUMN bar DROP DEFAULT;

и тогда это сработало.

на всякий случай, если вы используете Rails и у вас есть несколько операторов, вам нужно будет выполнить одно за другим, например:

execute "ALTER TYPE XXX ADD VALUE IF NOT EXISTS 'YYY';"
execute "ALTER TYPE XXX ADD VALUE IF NOT EXISTS 'ZZZ';"

Вот более общее, но довольно быстрое решение, которое, помимо изменения самого типа, обновляет все столбцы в базе данных, используя его. Этот метод можно применять, даже если новая версия ENUM отличается более чем на одну метку или пропускает некоторые из оригинальных. Код ниже заменяется my_schema.my_type AS ENUM ('a', 'b', 'c')на ENUM ('a', 'b', 'd', 'e'):

CREATE OR REPLACE FUNCTION tmp() RETURNS BOOLEAN AS
$BODY$

DECLARE
    item RECORD;

BEGIN

    -- 1. create new type in replacement to my_type
    CREATE TYPE my_schema.my_type_NEW
        AS ENUM ('a', 'b', 'd', 'e');

    -- 2. select all columns in the db that have type my_type
    FOR item IN
        SELECT table_schema, table_name, column_name, udt_schema, udt_name
            FROM information_schema.columns
            WHERE
                udt_schema   = 'my_schema'
            AND udt_name     = 'my_type'
    LOOP
        -- 3. Change the type of every column using my_type to my_type_NEW
        EXECUTE
            ' ALTER TABLE ' || item.table_schema || '.' || item.table_name
         || ' ALTER COLUMN ' || item.column_name
         || ' TYPE my_schema.my_type_NEW'
         || ' USING ' || item.column_name || '::text::my_schema.my_type_NEW;';
    END LOOP;

    -- 4. Delete an old version of the type
    DROP TYPE my_schema.my_type;

    -- 5. Remove _NEW suffix from the new type
    ALTER TYPE my_schema.my_type_NEW
        RENAME TO my_type;

    RETURN true;

END
$BODY$
LANGUAGE 'plpgsql';

SELECT * FROM tmp();
DROP FUNCTION tmp();

Весь процесс будет выполняться довольно быстро, потому что, если порядок меток сохраняется, фактическое изменение данных не произойдет. Я применил метод к 5 таблицам, используя по my_type50 000–70 000 строк в каждой, и весь процесс занял всего 10 секунд.

Конечно, функция будет возвращать исключение в случае, если где-то в данных используются метки, отсутствующие в новой версии ENUM, но в такой ситуации что-то должно быть сделано в любом случае заранее.

Для тех, кто ищет решение для транзакций, следующее работает.

Вместо ENUMa DOMAINследует использовать тип TEXTс ограничением, проверяющим, что значение находится в указанном списке допустимых значений (как предлагается в некоторых комментариях). Единственная проблема заключается в том, что никакое ограничение не может быть добавлено (и, следовательно, ни изменено) к домену, если он используется каким-либо составным типом (документы просто говорят, что это «должно в конечном итоге быть улучшено»). Однако такое ограничение можно обойти, используя ограничение, вызывающее функцию, следующим образом.

START TRANSACTION;

CREATE FUNCTION test_is_allowed_label(lbl TEXT) RETURNS BOOL AS $function$
    SELECT lbl IN ('one', 'two', 'three');
$function$ LANGUAGE SQL IMMUTABLE;

CREATE DOMAIN test_domain AS TEXT CONSTRAINT val_check CHECK (test_is_allowed_label(value));

CREATE TYPE test_composite AS (num INT, word test_domain);

CREATE TABLE test_table (val test_composite);
INSERT INTO test_table (val) VALUES ((1, 'one')::test_composite), ((3, 'three')::test_composite);
-- INSERT INTO test_table (val) VALUES ((4, 'four')::test_composite); -- restricted by the CHECK constraint

CREATE VIEW test_view AS SELECT * FROM test_table; -- just to show that the views using the type work as expected

CREATE OR REPLACE FUNCTION test_is_allowed_label(lbl TEXT) RETURNS BOOL AS $function$
    SELECT lbl IN ('one', 'two', 'three', 'four');
$function$ LANGUAGE SQL IMMUTABLE;

INSERT INTO test_table (val) VALUES ((4, 'four')::test_composite); -- allowed by the new effective definition of the constraint

SELECT * FROM test_view;

CREATE OR REPLACE FUNCTION test_is_allowed_label(lbl TEXT) RETURNS BOOL AS $function$
    SELECT lbl IN ('one', 'two', 'three');
$function$ LANGUAGE SQL IMMUTABLE;

-- INSERT INTO test_table (val) VALUES ((4, 'four')::test_composite); -- restricted by the CHECK constraint, again

SELECT * FROM test_view; -- note the view lists the restricted value 'four' as no checks are made on existing data

DROP VIEW test_view;
DROP TABLE test_table;
DROP TYPE test_composite;
DROP DOMAIN test_domain;
DROP FUNCTION test_is_allowed_label(TEXT);

COMMIT;

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

Единственным недостатком является то, что никакие проверки не выполняются на существующих данных, когда некоторые разрешенные значения удалены (что может быть приемлемо, особенно для этого вопроса). (Вызов ALTER DOMAIN test_domain VALIDATE CONSTRAINT val_checkзавершается той же ошибкой, что и добавление нового ограничения в домен, используемый составным типом, к сожалению.)

Обратите внимание, что небольшая модификация, такая как CHECK (value = ANY(get_allowed_values())), где get_allowed_values()функция вернула список разрешенных значений, не сработает - что довольно странно, поэтому я надеюсь, что предложенное выше решение работает надежно (пока у меня работает ...). (работает, на самом деле - это была моя ошибка)

Как обсуждалось выше, ALTERкоманда не может быть записана внутри транзакции. Предлагаемый способ - вставить в таблицу pg_enum напрямую, с помощью retrieving the typelem from pg_type tableи calculating the next enumsortorder number;

Ниже приведен код, который я использую. (Проверяет, существует ли дублирующее значение перед вставкой (ограничение между enumtypid и enumlabel name)

INSERT INTO pg_enum (enumtypid, enumlabel, enumsortorder)
    SELECT typelem,
    'NEW_ENUM_VALUE',
    (SELECT MAX(enumsortorder) + 1 
        FROM pg_enum e
        JOIN pg_type p
        ON p.typelem = e.enumtypid
        WHERE p.typname = '_mytypename'
    )
    FROM pg_type p
    WHERE p.typname = '_mytypename'
    AND NOT EXISTS (
        SELECT * FROM 
        pg_enum e
        JOIN pg_type p
        ON p.typelem = e.enumtypid
        WHERE e.enumlabel = 'NEW_ENUM_VALUE'
        AND p.typname = '_mytypename'
    )

Обратите внимание, что вашему имени типа предшествует знак подчеркивания в таблице pg_type. Кроме того, имя типа должно быть строчными в предложении where.

Теперь это можно безопасно записать в сценарий переноса БД.

Я не знаю, есть ли другой вариант, но мы можем сбросить значение, используя:

select oid from pg_type where typname = 'fase';'
select * from pg_enum where enumtypid = 24773;'
select * from pg_enum where enumtypid = 24773 and enumsortorder = 6;
delete from pg_enum where enumtypid = 24773 and enumsortorder = 6;

При использовании Navicat вы можете перейти к типам (в представлении -> другие -> типы) - получить представление дизайна типа - и нажать кнопку «добавить метку».